A curated list of neural network pruning and related resources. Inspired by awesome-deep-vision, awesome-adversarial-machine-learning, awesome-deep-learning-papers and Awesome-NAS.
Please feel free to pull requests or open an issue to add papers.
Type | F |
W |
S |
Other |
---|---|---|---|---|
Explanation | Filter pruning | Weight pruning | Special Networks | other types |
A Survey of Structured Pruning (arXiv version and IEEE T-PAMI version)
Please cite our paper if it's helpful:
@article{he2024structured,
author={He, Yang and Xiao, Lingao},
journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},
title={Structured Pruning for Deep Convolutional Neural Networks: A Survey},
year={2024},
volume={46},
number={5},
pages={2900-2919},
doi={10.1109/TPAMI.2023.3334614}}
The related papers are categorized as below:
Title | Venue | Type | Code |
---|---|---|---|
Dynamic Network Surgery for Efficient DNNs | NeurIPS | W |
Caffe(Author) |
Learning the Number of Neurons in Deep Networks | NeurIPS | F |
- |
Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Coding | ICLR (Best) | W |
Caffe(Author) |
Title | Venue | Type | Code |
---|---|---|---|
Learning both Weights and Connections for Efficient Neural Networks | NeurIPS | W |
PyTorch(3rd) |
Awesome-model-compression-and-acceleration