Learning both Weights and Connections for Efficient Neural Networks
Open accessAbstract
Neural networks are both computationally intensive and memory intensive, making them difficult to deploy on embedded systems. Also, conventional networks fix the architecture before training starts; as a result, training cannot improve the architecture. To address these limitations, we describe a method to reduce the storage and computation required by neural networks by an order of magnitude without affecting their accuracy by learning only the important connections. Our method prunes redundant connections using a three-step method. First, we train the network to learn which connections are important. Next, we prune the unimportant connections. Finally, we retrain the network to fine tune the weights of the remaining connections. On the ImageNet dataset, our method reduced the number of parameters of AlexNet by a factor of 9x, from 61 million to 6.7 million, without incurring accuracy loss. Similar experiments with VGG-16 found that the number of parameters can be reduced by 13x, from 138 million to 10.3 million, again with no loss of accuracy.
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
Claims other papers make about this one
Each is checked against this paper’s own text: does it say what the citing paper says it does?
Floating point add process takes three orders of magnitudes of add operation.
“It is evident that floating point add process takes three orders of magnitudes of add operation [19].”
From Optimized Convolutional Neural Network at the IoT edge for image detection using pruning and quantization (2024) · cites this paper as [19]
Researchers have proposed techniques for reducing transformers' size, latency and energy consumption for efficient inference for a wide range of applications.
“These challenges have motivated researchers to propose techniques for reducing transformers' size, latency and energy consumption for efficient inference for a wide range of applications.”
From A Survey of Techniques for Optimizing Transformer Inference (2023) · cites this paper as [40]
The methods for reducing transformers' size, latency and energy consumption include pruning, quantization, knowledge distillation and neural architecture search.
“The methods include pruning [40, 41], Quantization [42-44], Knowledge Distillation [45] and Neural Architecture Search [46].”
From A Survey of Techniques for Optimizing Transformer Inference (2023) · cites this paper as [40]
FedPAQ provides near-optimal theoretical guarantees for strongly convex and non-convex loss functions and empirically demonstrates communication-computation tradeoff.
“FedPAQ [204] simultaneously incorporates these features and provides near-optimal theoretical guarantees for both strongly convex and non-convex loss functions, while empirically demonstrating the communication-computation tradeoff.”
From Convergence of Edge Computing and Deep Learning: A Comprehensive Survey (2020) · cites this paper as [204]
Lin et al. [53] observe that 99.9% of the gradient exchange in distributed SGD (DSGD) are redundant.
“Lin et al. [53] observe that 99.9% of the gradient exchange in distributed SGD (DSGD) are redundant, which demonstrates the power of gradient compression.”
From Edge Intelligence: Paving the Last Mile of Artificial Intelligence With Edge Computing (2019) · cites this paper as [53]
Deep gradient compression (DGC) compresses the gradient by 270-600× for a wide range of CNNs and RNNs.
“Based on this observation, Lin et al. propose deep gradient compression (DGC), which compresses the gradient by 270-600× for a wide range of CNNs and RNNs.”
From Edge Intelligence: Paving the Last Mile of Artificial Intelligence With Edge Computing (2019) · cites this paper as [53]
Sparse architectures produced by pruning are difficult to train from the start, reaching lower accuracy than the original networks.
“However, contemporary experience is that the sparse architectures produced by pruning are difficult to train from the start, which would similarly improve training performance.”
From The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks (2018) · cites this paper as Han et al., 2015
Cites
Compressing Neural Networks with the Hashing Trick
Chen, W., Wilson, J. T., Tyree, S., Weinberger, K. Q., & Chen, Y. (2015) · arXiv · doi:10.48550/arxiv.1504.04788 · Cited by 974 · see citing works
Flagged first, then by citations
Cited by
Sorted by citations
Recent advances in convolutional neural networks
Gu, J., Wang, Z., Kuen, J., Ma, L., Shahroudy, A., Shuai, B., Liu, T., Wang, X., Wang, G., Cai, J., et al. (2018) · Pattern Recognition · doi:10.1016/j.patcog.2017.10.013 · Cited by 6,576 · see citing works
Efficient Processing of Deep Neural Networks: A Tutorial and Survey
Sze, V., Chen, Y.-H., Yang, T.-J., & Emer, J. S. (2017) · Proceedings of the IEEE · doi:10.1109/jproc.2017.2761740 · Cited by 4,215 · see citing works
XNOR-Net: ImageNet Classification Using Binary Convolutional Neural Networks
Rastegari, M., Ordonez, V., Redmon, J., & Farhadi, A. (2016) · Lecture Notes in Computer Science · doi:10.1007/978-3-319-46493-0_32 · Cited by 4,408 · see citing works
The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks
Frankle, J., & Carbin, M. (2018) · arXiv · doi:10.48550/arxiv.1803.03635 · Cited by 1,529 · see citing works
Convergence of Edge Computing and Deep Learning: A Comprehensive Survey
Wang, X., Han, Y., Leung, V. C. M., Niyato, D., Yan, X., & Chen, X. (2020) · IEEE Communications Surveys & Tutorials · doi:10.1109/comst.2020.2970550 · Cited by 1,498 · see citing works
AMC: AutoML for Model Compression and Acceleration on Mobile Devices
He, Y., Lin, J., Liu, Z., Wang, H., Li, L.-J., & Han, S. (2018) · Lecture Notes in Computer Science · doi:10.1007/978-3-030-01234-2_48 · Cited by 1,400 · see citing works
Self-citationRecent advances in deep learning for object detection
Wu, X., Sahoo, D., & Hoi, S. C. H. (2020) · Neurocomputing · doi:10.1016/j.neucom.2020.01.085 · Cited by 1,047 · see citing works
Designing Energy-Efficient Convolutional Neural Networks Using Energy-Aware Pruning
Yang, T.-J., Chen, Y.-H., & Sze, V. (2017) · 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) · doi:10.1109/cvpr.2017.643 · Cited by 732 · see citing works
ESE: Efficient Speech Recognition Engine with Sparse LSTM on FPGA
Han, S., Kang, J., Mao, H., Hu, Y., Li, X., Li, Y., Xie, D., Luo, H., Yao, S., Wang, Y., et al. (2017) · Proceedings of the 2017 ACM/SIGDA International Symposium on Field-Programmable Gate Arrays · doi:10.1145/3020078.3021745 · Cited by 591 · see citing works
Self-citationScaling for edge inference of deep neural networks
Xu, X., Ding, Y., Hu, S. X., Niemier, M., Cong, J., Hu, Y., & Shi, Y. (2018) · Nature Electronics · doi:10.1038/s41928-018-0059-3 · Cited by 462 · see citing works
FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness
Dao, T., Fu, D. Y., Ermon, S., Rudra, A., & Ré, C. (2022) · arXiv · doi:10.48550/arxiv.2205.14135 · Cited by 1,214 · see citing works
“[36] Song Han, Jeff Pool, John Tran, and William J Dally.”
BackgroundMethodEfficient Acceleration of Deep Learning Inference on Resource-Constrained Edge Devices: A Review
Shuvo, M. M. H., Islam, S. K., Cheng, J., & Morshed, B. I. (2023) · Proceedings of the IEEE · doi:10.1109/jproc.2022.3226481 · Cited by 436 · see citing works
Bai, X., Wang, X., Liu, X., Liu, Q., Song, J., Sebe, N., & Kim, B. (2021) · Pattern Recognition · doi:10.1016/j.patcog.2021.108102 · Cited by 313 · see citing works
Li, Y., Li, K., Liu, X., Wang, Y., & Zhang, L. (2021) · Applied Energy · doi:10.1016/j.apenergy.2020.116410 · Cited by 247 · see citing works
Yang, Z., Moczulski, M., Denil, M., De Freitas, N., Song, L., & Wang, Z. (2015) · 2015 IEEE International Conference on Computer Vision (ICCV) · doi:10.1109/iccv.2015.173 · Cited by 215 · see citing works
Patch Slimming for Efficient Vision Transformers
Tang, Y., Han, K., Wang, Y., Xu, C., Guo, J., Xu, C., & Tao, D. (2022) · 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) · doi:10.1109/cvpr52688.2022.01185 · Cited by 201 · see citing works
Filter-based deep-compression with global average pooling for convolutional networks
Hsiao, T.-Y., Chang, Y.-C., Chou, H.-H., & Chiu, C.-T. (2019) · Journal of Systems Architecture · doi:10.1016/j.sysarc.2019.02.008 · Cited by 190 · see citing works
OpenEI: An Open Framework for Edge Intelligence
Zhang, X., Wang, Y., Lu, S., Liu, L., xu, L., & Shi, W. (2019) · 2019 IEEE 39th International Conference on Distributed Computing Systems (ICDCS) · doi:10.1109/icdcs.2019.00182 · Cited by 139 · see citing works
“[24] pruned redundant connections using a three-step method: First, the network learns which connections are important, and then they prune the unimportant connections.”
MethodMathur, A., Lane, N. D., Bhattacharya, S., Boran, A., Forlivesi, C., & Kawsar, F. (2017) · Proceedings of the 15th Annual International Conference on Mobile Systems, Applications, and Services · doi:10.1145/3081333.3081359 · Cited by 128 · see citing works
PQK: Model Compression via Pruning, Quantization, and Knowledge Distillation
Kim, J., Chang, S., & Kwak, N. (2021) · Interspeech 2021 · doi:10.21437/interspeech.2021-248 · Cited by 85 · see citing works
Network Sketching: Exploiting Binary Structure in Deep CNNs
Guo, Y., Yao, A., Zhao, H., & Chen, Y. (2017) · 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) · doi:10.1109/cvpr.2017.430 · Cited by 81 · see citing works
Georgiev, P., Bhattacharya, S., Lane, N. D., & Mascolo, C. (2017) · Proceedings of the ACM on Interactive Mobile Wearable and Ubiquitous Technologies · doi:10.1145/3131895 · Cited by 67 · see citing works
Analog/Mixed-Signal Hardware Error Modeling for Deep Learning Inference
Rekhi, A. S., Zimmer, B., Nedovic, N., Liu, N., Venkatesan, R., Wang, M., Khailany, B., Dally, W. J., & Gray, C. T. (2019) · Proceedings of the 56th Annual Design Automation Conference 2019 · doi:10.1145/3316781.3317770 · Cited by 67 · see citing works
Self-citationVisual Scene Understanding for Autonomous Driving Using Semantic Segmentation
Hofmarcher, M., Unterthiner, T., Arjona-Medina, J., Klambauer, G., Hochreiter, S., & Nessler, B. (2019) · Lecture Notes in Computer Science · doi:10.1007/978-3-030-28954-6_15 · Cited by 51 · see citing works
Neural Networks Meet Physical Networks: Distributed Inference Between Edge Devices and the Cloud
Chinchali, S. P., Cidon, E., Pergament, E., Chu, T., & Katti, S. (2018) · Proceedings of the 17th ACM Workshop on Hot Topics in Networks · doi:10.1145/3286062.3286070 · Cited by 46 · see citing works
Random sketch learning for deep neural networks in edge computing
Li, B., Chen, P., Liu, H., Guo, W., Cao, X., Du, J., Zhao, C., & Zhang, J. (2021) · Nature Computational Science · doi:10.1038/s43588-021-00039-6 · Cited by 40 · see citing works
Naveen, S., & Kounte, M. R. (2024) · Multimedia Tools and Applications · doi:10.1007/s11042-024-20523-1 · Cited by 33 · see citing works
A Deep Look into Logarithmic Quantization of Model Parameters in Neural Networks
Cai, J., Takemoto, M., & Nakajo, H. (2018) · Proceedings of the 10th International Conference on Advances in Information Technology · doi:10.1145/3291280.3291800 · Cited by 31 · see citing works
Collaborative cloud-edge computation for personalized driving behavior modeling
Zhang, X., Qiao, M., Liu, L., Xu, Y., & Shi, W. (2019) · Proceedings of the 4th ACM/IEEE Symposium on Edge Computing · doi:10.1145/3318216.3363310 · Cited by 30 · see citing works
CoNNa–Hardware accelerator for compressed convolutional neural networks
Struharik, R. J. R., Vukobratović, B. Z., Erdeljan, A. M., & Rakanović, D. M. (2020) · Microprocessors and Microsystems · doi:10.1016/j.micpro.2020.102991 · Cited by 18 · see citing works
Cloud-Edge Collaborative Defect Detection Based on Efficient Yolo Networks and Incremental Learning
Lei, Z., Zhang, Y., Wang, J., & Zhou, M. (2024) · Sensors · doi:10.3390/s24185921 · Cited by 16 · see citing works
TextPruner: A Model Pruning Toolkit for Pre-Trained Language Models
Yang, Z., Cui, Y., & Chen, Z. (2022) · Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics: System Demonstrations · doi:10.18653/v1/2022.acl-demo.4 · Cited by 14 · see citing works
An Efficient Object Detection Algorithm Based on Compressed Networks
Li, J., Peng, K., & Chang, C.-C. (2018) · Symmetry · doi:10.3390/sym10070235 · Cited by 6 · see citing works
Huang, S., Chen, S., Peng, H., Manu, D., Kong, Z., Yuan, G., Yang, L., Wang, S., Liu, H., & Ding, C. (2021) · Proceedings of the 2021 Great Lakes Symposium on VLSI · doi:10.1145/3453688.3461740 · Cited by 4 · see citing works
SMoA: Improving Multi-agent Large Language Models with Sparse Mixture-of-Agents
Li, D., Tan, Z., Qian, P., Li, Y., Chaudhary, K. S., Hu, L., & Shen, J. (2024) · arXiv · doi:10.48550/arxiv.2411.03284
“To overcome the limitations of MoA and enhance the utility of multi-agent LLMs, we propose Sparse Mixture-of-Agents (SMoA), inspired by the design of sparse mixture-of-experts (SMoE) (Han et al., 2015; Shazeer et al., 2016).”
MethodInfluential
- Authors
- Han, S., Pool, J., Tran, J., & Dally, W. J.
- Affiliations
- Stanford University (US)
- Nvidia (United Kingdom) (GB)
- Venue
- arXiv
- Publisher
- arXiv
- Published
- 2015
- Volume
- 28
- Pages
- 1135-1143
- ISSN
- 2331-8422 (ISSN-L)
- arXiv
- 1506.02626
- OpenAlex
- W1845051632
- Cited by
- 1,481
- References
- 1
- Citations by year
- 2026: 15 · 2025: 76 · 2024: 125 · 2023: 174 · 2022: 218 · 2021: 254 · 2020: 228 · 2019: 176 · 2018: 130 · 2017: 52 · 2016: 29 · 2015: 3 · 2012: 1
- Type
- posted content
- Language
- en
- Open access
- green
- arxiv.org
- Locations
- arXiv (Cornell University) (submittedVersion)
- arXiv (Cornell University)
- neural information processing systems
- Topics
- Physical Sciences › Computer Science › Computer Vision and Pattern Recognition › Advanced Neural Network Applications
- Physical Sciences › Computer Science › Artificial Intelligence › Machine Learning and Data Classification
- Physical Sciences › Computer Science › Artificial Intelligence › Domain Adaptation and Few-Shot Learning
- Keywords
- neural network pruning, connection pruning, weight learning, efficient neural networks, parameter reduction, AlexNet, VGG16, ImageNet, fine-tuning
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review
