Learning to Prune Deep Neural Networks via Layer-wise Optimal Brain Surgeon
Not on the map. Why?
Abstract (arXiv)
How to develop slim and accurate deep neural networks has become crucial for real- world applications, especially for those employed in embedded systems. Though previous work along this research line has shown some promising results, most existing methods either fail to significantly compress a well-trained deep network or require a heavy retraining process for the pruned deep network to re-boost its prediction performance. In this paper, we propose a new layer-wise pruning method for deep neural networks. In our proposed method, parameters of each individual layer are pruned independently based on second order derivatives of a layer-wise error function with respect to the corresponding parameters. We prove that the final prediction performance drop after pruning is bounded by a linear combination of the reconstructed errors caused at each layer. Therefore, there is a guarantee that one only needs to perform a light retraining process on the pruned network to resume its original prediction performance. We conduct extensive experiments on benchmark datasets to demonstrate the effectiveness of our pruning method compared with several state-of-the-art baseline methods.
Key facts. Cited by 219 papers in the map (data through 7 Oct 2026). First posted on arXiv 22 May 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 48; 2022: 41; 2023: 46; 2024: 33; 2025: 37; 2026: 14.
219 papers.
Sort:
By area:
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2021A Survey of Quantization Methods for Efficient Neural Network Inferencecited by 798
- 2021BRECQ: Pushing the Limit of Post-Training Quantization by Block Reconstructioncited by 464
- 2022Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruningcited by 317
- 2023DepGraph: Towards Any Structural Pruningcited by 202
- 2023Structural Pruning for Diffusion Modelscited by 193
- 2021Efficient Deep Learning: A Survey on Making Deep Learning Models Smaller, Faster, and Bettercited by 153
- 2021Learned Token Pruning for Transformerscited by 138
- 2021Group Fisher Pruning for Practical Network Compressioncited by 130
- 2024Mobile Edge Intelligence for Large Language Models: A Contemporary Surveycited by 125
- 2022Monarch: Expressive Structured Matrices for Efficient and Accurate Trainingcited by 100
- 2022Unified Visual Transformer Compressioncited by 79
38 more
- 2022Efficient Methods for Natural Language Processing: A Surveycited by 78
- 2021Can Subnetwork Structure be the Key to Out-of-Distribution Generalization?cited by 75
- 2021Pixelated Butterfly: Simple and Efficient Sparse training for Neural Network Modelscited by 75
- 2023Simple Hardware-Efficient Long Convolutions for Sequence Modelingcited by 62
- 2021AC/DC: Alternating Compressed/DeCompressed Training of Deep Neural Networkscited by 58
- 2021M-FAC: Efficient Matrix-Free Approximations of Second-Order Informationcited by 54
- 2021Federated Dynamic Sparse Training: Computing Less, Communicating Less, Yet Learning Bettercited by 54
- 2025GPTAQ: Efficient Finetuning-Free Quantization for Asymmetric Calibrationcited by 54
- 2023LoRAPrune: Structured Pruning Meets Low-Rank Parameter-Efficient Fine-Tuningcited by 53
- 2021Analytic Insights into Structure and Rank of Neural Network Hessian Mapscited by 52
- 2022SQuant: On-the-Fly Data-Free Quantization via Diagonal Hessian Approximationcited by 51
- 2021Dynamic Sparse Training for Deep Reinforcement Learningcited by 49
- 2024Addressing Loss of Plasticity and Catastrophic Forgetting in Continual Learningcited by 49
- 2024MoDeGPT: Modular Decomposition for Large Language Model Compressioncited by 45
- 2021Hessian-Aware Pruning and Optimal Neural Implantcited by 43
- 2024DiP-GO: A Diffusion Pruner via Few-step Gradient Optimizationcited by 43
- 2021Neural Mean Discrepancy for Efficient Out-of-Distribution Detectioncited by 42
- 2023The Emergence of Essential Sparsity in Large Pre-trained Models: The Weights that Mattercited by 42
- 2022The Combinatorial Brain Surgeon: Pruning Weights That Cancel One Another in Neural Networkscited by 41
- 2023Fast as CHITA: Neural Network Pruning with Combinatorial Optimizationcited by 40
- 2023SEENN: Towards Temporal Spiking Early-Exit Neural Networkscited by 39
- 2021MQBench: Towards Reproducible and Deployable Model Quantization Benchmarkcited by 38
- 2021Edge-Cloud Polarization and Collaboration: A Comprehensive Survey for AIcited by 38
- 2024Lightweight Deep Learning for Resource-Constrained Environments: A Surveycited by 38
- 2023LoRAShear: Efficient Large Language Model Structured Pruning and Knowledge Recoverycited by 37
- 2021Achieving on-Mobile Real-Time Super-Resolution with Neural Architecture and Pruning Searchcited by 34
- 2021Applications and Techniques for Fast Machine Learning in Sciencecited by 32
- 2023FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Corescited by 32
- 2022Pruning's Effect on Generalization Through the Lens of Training and Regularizationcited by 31
- 2022RLx2: Training a Sparse Deep Reinforcement Learning Model from Scratchcited by 30
- 2023NTK-SAP: Improving neural network pruning by aligning training dynamicscited by 30
- 2023QuantEase: Optimization-based Quantization for Language Modelscited by 28
- 2023Sparsity May Cry: Let Us Fail (Current) Sparse Neural Networks Together!cited by 27
- 2024Differentiable Weightless Neural Networkscited by 27
- 2022Why Random Pruning Is All We Need to Start Sparsecited by 26
- 2024DISP-LLM: Dimension-Independent Structural Pruning for Large Language Modelscited by 24
- 2024SGLP: A Similarity Guided Fast Layer Partition Pruning for Compressing Large Deep Modelscited by 24
- 2023Visual Prompting Upgrades Neural Network Sparsification: A Data-Model Perspectivecited by 23
50 most cited of 219 All 219 (JSON)
- 2026Trustworthy Data- and ML-Ops for Intelligent Transportation Systems and Logistics1 Oct 2026
- 2026Redundancy Meets Synergy: Dependency-aware Expert Selection for MoE via Submodular Optimization30 Sep 2026
- 2026PruneForget: Joint Unlearning and Pruning of Vision Models26 Sep 2026
- 2026Effective pruning of task-trained recurrent neural networks using noisy fluctuations and connection rescaling5 Aug 2026
- 2026Importance-Aware OBS Pruning for Diffusion Models22 Jul 2026
- 2026Hybrid Compression: Integrating Pruning and Quantization for Optimized Neural Networks22 Jun 2026
- 2026EinSort: Sorting is All We Need for Tensorizing LLM7 Jun 2026
- 2026Structural Pruning of Large Vision Language Models: A Comprehensive Study on Pruning Dynamics, Recovery, and Data Efficiency27 Apr 2026
- 2026Optimal Brain Decomposition for Accurate LLM Low-Rank Approximation1 Apr 2026
- 2026PruneFuse: Efficient Data Selection via Weight Pruning and Network Fusion27 Mar 2026
- 2026Locate-then-Sparsify: Attribution Guided Sparse Strategy for Visual Hallucination Mitigation17 Mar 2026
- 2026FastLightGen: Fast and Light Video Generation with Fewer Steps and Parameters2 Mar 2026
38 more
- 2026FlattenGPT: Depth Compression for Transformer with Layer Flattening9 Feb 2026
- 2026Towards Understanding Best Practices for Quantization of Vision-Language Models21 Jan 2026
- 2025Metanetworks as Regulatory Operators: Learning to Edit for Requirement Compliance17 Dec 2025
- 2025ZK-APEX: Zero-Knowledge Approximate Personalized Unlearning with Executable Proofs9 Dec 2025
- 2025Theoretical Compression Bounds for Wide Multilayer Perceptrons6 Dec 2025
- 2025Think Before You Prune: Self-Reflective Structured Pruning for Reasoning Language Models1 Dec 2025
- 2025Kernelized Sparse Fine-Tuning with Bi-level Parameter Competition for Vision Models28 Oct 2025
- 2025VL-SAE: Interpreting and Enhancing Vision-Language Alignment with a Unified Concept Set24 Oct 2025
- 2025Compress to Impress: Efficient LLM Adaptation Using a Single Gradient Step on 100 Samples23 Oct 2025
- 2025From Local to Global: Revisiting Structured Pruning Paradigms for Large Language Models20 Oct 2025
- 2025SQS: Bayesian DNN Compression through Sparse Quantized Sub-distributions10 Oct 2025
- 2025OBS-Diff: Accurate Pruning For Diffusion Models in One-Shot8 Oct 2025
- 2025HEAPr: Hessian-based Efficient Atomic Expert Pruning in Output Space26 Sep 2025
- 2025Sensitivity-Aware Post-Training Quantization for Deep Neural Networks6 Sep 2025
- 2025FAIR-Pruner: A Flexible Framework for Automatic Layer-Wise Pruning via Tolerance of Difference4 Aug 2025
- 2025Investigating Structural Pruning and Recovery Techniques for Compressing Multimodal Large Language Models: An Empirical Study28 Jul 2025
- 2025MLoRQ: Bridging Low-Rank and Quantization for Transformer Compression13 Jul 2025
- 2025DOTResize: Reducing LLM Width via Discrete Optimal Transport-based Neuron Merging6 Jul 2025
- 2025Towards Universal & Efficient Model Compression via Exponential Torque Pruning27 Jun 2025
- 2025Sparse-Reg: Improving Sample Complexity in Offline Reinforcement Learning using Sparsity20 Jun 2025
- 2025ALTER: All-in-One Layer Pruning and Temporal Expert Routing for Efficient Diffusion Generation27 May 2025
- 2025μ-MoE: Test-Time Pruning as Micro-Grained Mixture-of-Experts24 May 2025
- 2025TeleSparse: Practical Privacy-Preserving Verification of Deep Neural Networks27 Apr 2025
- 2025CUT: Pruning Pre-Trained Multi-Task Models into Compact Models for Edge Devices14 Apr 2025
- 2025AB-Cache: Training-Free Acceleration of Diffusion Models via Adams-Bashforth Cached Feature Reuse13 Apr 2025
- 2025The Effects of Grouped Structural Global Pruning of Vision Transformers on Domain Generalisation5 Apr 2025
- 2025GPTAQ: Efficient Finetuning-Free Quantization for Asymmetric Calibration3 Apr 2025
- 2025STADE: Standard Deviation as a Pruning Metric28 Mar 2025
- 2025An Efficient Training Algorithm for Models with Block-wise Sparsity27 Mar 2025
- 2025Safe Vision-Language Models via Unsafe Weights Manipulation14 Mar 2025
- 2025Optimal Brain Apoptosis25 Feb 2025
- 2025On-device edge learning for IoT data streams: a survey25 Feb 2025
- 2025Systematic Weight Evaluation for Pruning Large Language Models: Enhancing Performance and Sustainability24 Feb 2025
- 2025PLayer-FL: A Principled Approach to Personalized Layer-wise Cross-Silo Federated Learning12 Feb 2025
- 2025Compressing Model with Few Class-Imbalance Samples: An Out-of-Distribution Expedition9 Feb 2025
- 2025Training-Free Restoration of Pruned Neural Networks6 Feb 2025
- 2025Hardware-Aware DNN Compression for Homogeneous Edge Devices25 Jan 2025
- 2025Optimize Incompatible Parameters through Compatibility-aware Knowledge Integration10 Jan 2025
50 newest of 219 All 219 (JSON)
- 2023Structural Pruning for Diffusion Modelscited by 193
- 2024DiP-GO: A Diffusion Pruner via Few-step Gradient Optimizationcited by 43
- 2025OBS-Diff: Accurate Pruning For Diffusion Models in One-Shotcited by 19
- 2025AB-Cache: Training-Free Acceleration of Diffusion Models via Adams-Bashforth Cached Feature Reusecited by 11
- 2023F3-Pruning: A Training-Free and Generalized Pruning Strategy towards Faster and Finer Text-to-Video Synthesiscited by 9
- 2025ALTER: All-in-One Layer Pruning and Temporal Expert Routing for Efficient Diffusion Generationcited by 4
- 2026Importance-Aware OBS Pruning for Diffusion Modelscited by 0
- 2026Importance-Aware OBS Pruning for Diffusion Models22 Jul 2026
- 2025OBS-Diff: Accurate Pruning For Diffusion Models in One-Shot8 Oct 2025
- 2025ALTER: All-in-One Layer Pruning and Temporal Expert Routing for Efficient Diffusion Generation27 May 2025
- 2025AB-Cache: Training-Free Acceleration of Diffusion Models via Adams-Bashforth Cached Feature Reuse13 Apr 2025
- 2024DiP-GO: A Diffusion Pruner via Few-step Gradient Optimization22 Oct 2024
- 2023F3-Pruning: A Training-Free and Generalized Pruning Strategy towards Faster and Finer Text-to-Video Synthesis6 Dec 2023
- 2023Structural Pruning for Diffusion Models18 May 2023
- 2024MULTIFLOW: Shifting Towards Task-Agnostic Vision-Language Pruningcited by 13
- 2026Towards Understanding Best Practices for Quantization of Vision-Language Modelscited by 2
- 2026Locate-then-Sparsify: Attribution Guided Sparse Strategy for Visual Hallucination Mitigationcited by 2
- 2026Structural Pruning of Large Vision Language Models: A Comprehensive Study on Pruning Dynamics, Recovery, and Data Efficiencycited by 2
- 2025Safe Vision-Language Models via Unsafe Weights Manipulationcited by 1
- 2025Investigating Structural Pruning and Recovery Techniques for Compressing Multimodal Large Language Models: An Empirical Studycited by 1
- 2026Structural Pruning of Large Vision Language Models: A Comprehensive Study on Pruning Dynamics, Recovery, and Data Efficiency27 Apr 2026
- 2026Locate-then-Sparsify: Attribution Guided Sparse Strategy for Visual Hallucination Mitigation17 Mar 2026
- 2026Towards Understanding Best Practices for Quantization of Vision-Language Models21 Jan 2026
- 2025Investigating Structural Pruning and Recovery Techniques for Compressing Multimodal Large Language Models: An Empirical Study28 Jul 2025
- 2025Safe Vision-Language Models via Unsafe Weights Manipulation14 Mar 2025
- 2024MULTIFLOW: Shifting Towards Task-Agnostic Vision-Language Pruning8 Apr 2024
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2021A Survey of Quantization Methods for Efficient Neural Network Inferencecited by 798
- 2021BRECQ: Pushing the Limit of Post-Training Quantization by Block Reconstructioncited by 464
- 2022Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruningcited by 317
- 2023DepGraph: Towards Any Structural Pruningcited by 202
- 2021Efficient Deep Learning: A Survey on Making Deep Learning Models Smaller, Faster, and Bettercited by 153
- 2021Group Fisher Pruning for Practical Network Compressioncited by 130
- 2024Mobile Edge Intelligence for Large Language Models: A Contemporary Surveycited by 125
- 2022Monarch: Expressive Structured Matrices for Efficient and Accurate Trainingcited by 100
- 2022Unified Visual Transformer Compressioncited by 79
- 2022Efficient Methods for Natural Language Processing: A Surveycited by 78
- 2021Can Subnetwork Structure be the Key to Out-of-Distribution Generalization?cited by 75
12 most cited of 195 All 195 (JSON)
- 2026Trustworthy Data- and ML-Ops for Intelligent Transportation Systems and Logistics1 Oct 2026
- 2026Effective pruning of task-trained recurrent neural networks using noisy fluctuations and connection rescaling5 Aug 2026
- 2026Hybrid Compression: Integrating Pruning and Quantization for Optimized Neural Networks22 Jun 2026
- 2026EinSort: Sorting is All We Need for Tensorizing LLM7 Jun 2026
- 2026Optimal Brain Decomposition for Accurate LLM Low-Rank Approximation1 Apr 2026
- 2026PruneFuse: Efficient Data Selection via Weight Pruning and Network Fusion27 Mar 2026
- 2026FastLightGen: Fast and Light Video Generation with Fewer Steps and Parameters2 Mar 2026
- 2026FlattenGPT: Depth Compression for Transformer with Layer Flattening9 Feb 2026
- 2025Metanetworks as Regulatory Operators: Learning to Edit for Requirement Compliance17 Dec 2025
- 2025Theoretical Compression Bounds for Wide Multilayer Perceptrons6 Dec 2025
- 2025Kernelized Sparse Fine-Tuning with Bi-level Parameter Competition for Vision Models28 Oct 2025
- 2025Compress to Impress: Efficient LLM Adaptation Using a Single Gradient Step on 100 Samples23 Oct 2025
12 newest of 195 All 195 (JSON)