The LLM Surgeon
Not on the map. Why?
Abstract (arXiv)
State-of-the-art language models are becoming increasingly large in an effort to achieve the highest performance on large corpora of available textual data. However, the sheer size of the Transformer architectures makes it difficult to deploy models within computational, environmental or device-specific constraints. We explore data-driven compression of existing pretrained models as an alternative to training smaller models from scratch. To do so, we scale Kronecker-factored curvature approximations of the target loss landscape to large language models. In doing so, we can compute both the dynamic allocation of structures that can be removed as well as updates of remaining weights that account for the removal. We provide a general framework for unstructured, semi-structured and structured pruning and improve upon weight updates to capture more correlations between weights, while remaining computationally efficient. Experimentally, our method can prune rows and columns from a range of OPT models and Llamav2-7B by 20%-30%, with a negligible loss in performance, and achieve state-of-the-art results in unstructured and semi-structured pruning of large language models.
Key facts. Cited by 36 papers in the map (data through 7 Oct 2026). Builds on 20 of its 29 references that are in the map. First posted on arXiv 28 Dec 2023.
Builds on
20 of 29 references.
References · 20
- 2017Attention Is All You Needcited by 59,932
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,464
- 2016Pointer Sentinel Mixture Modelscited by 2,992
- 2019HuggingFace's Transformers: State-of-the-art Natural Language Processingcited by 2,201
- 2023SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shotcited by 1,090
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 776
- 2023A Simple and Effective Pruning Approach for Large Language Modelscited by 776
- 2017Learning Sparse Neural Networks through L0 Regularizationcited by 571
- 2022Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruningcited by 317
- 2019Limitations of the Empirical Fisher Approximation for Natural Gradient Descentcited by 213
8 more
- 2021Learning N:M Fine-grained Structured Sparse Neural Networks From Scratchcited by 177
- 2017Practical Gauss-Newton Optimisation for Deep Learningcited by 139
- 2022The Optimal BERT Surgeon: Scalable and Accurate Second-Order Pruning for Large Language Modelscited by 115
- 2019EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasiscited by 98
- 2023Kronecker-Factored Approximate Curvature for Modern Neural Network Architecturescited by 49
- 2022Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximationscited by 42
- 2023Learning Layer-wise Equivariances Automatically using Gradientscited by 23
- 2022Efficient Approximations of the Fisher Matrix in Neural Networks using Kronecker Product Singular Value Decompositioncited by 9
Built on by
Citing papers per year: 2024: 7; 2025: 14; 2026: 15.
36 papers.
Sort:
By area:
- 2024Model Compression and Efficient Inference for Large Language Models: A Surveycited by 62
- 2024Pruner-Zero: Evolving Symbolic Pruning Metric from scratch for Large Language Modelscited by 40
- 2024DISP-LLM: Dimension-Independent Structural Pruning for Large Language Modelscited by 24
- 2024A deeper look at depth pruning of LLMscited by 23
- 20252SSP: A Two-Stage Framework for Structured Pruning of LLMscited by 12
- 2024HW-GPT-Bench: Hardware-Aware Architecture Benchmark for Language Modelscited by 10
- 2025ToMoE: Converting Dense Large Language Models to Mixture-of-Experts through Dynamic Structural Pruningcited by 10
- 2025EfficientLLM: Scalable Pruning-Aware Pretraining for Architecture-Agnostic Edge Language Modelscited by 10
- 2025ReplaceMe: Network Simplification via Depth Pruning and Transformer Block Linearizationcited by 10
- 2025Position: Curvature Matrices Should Be Democratized via Linear Operatorscited by 9
- 2025Olica: Efficient Structured Pruning of Large Language Models without Retrainingcited by 6
- 2025Efficient LLMs with AMP: Attention Heads and MLP Pruningcited by 5
24 more
- 2025SparseSSM: Efficient Selective Structured State Space Models Can Be Pruned in One-Shotcited by 5
- 2025laplax -- Laplace Approximations with JAXcited by 5
- 2025Pivoting Factorization: A Compact Meta Low-Rank Representation of Sparsity for Efficient Inference in Large Language Modelscited by 4
- 2025Skrr: Skip and Re-use Text Encoder Layers for Memory Efficient Text-to-Image Generationcited by 4
- 2026Optimal Brain Decomposition for Accurate LLM Low-Rank Approximationcited by 4
- 2024STAT: Shrinking Transformers After Trainingcited by 2
- 2025BLaST: High Performance Inference and Pretraining using BLock Sparse Transformerscited by 2
- 2026Reasoning-preserved Efficient Distillation of Large Language Models via Activation-aware Initializationcited by 2
- 2026Understanding Calibration and Truncation Error Propagation in Training-Free Low-Rank Compression for LLMscited by 2
- 2025Elastic ViTs from Pretrained Models without Retrainingcited by 1
- 2024OpenBA-V2: Reaching 77.3% High Compression Ratio with Fast Multi-Stage Pruningcited by 0
- 2025Accelerating Attention with Basis Decompositioncited by 0
- 2026Compressing LLMs with MoP: Mixture of Prunerscited by 0
- 2026FlattenGPT: Depth Compression for Transformer with Layer Flatteningcited by 0
- 2026DOT-MoE: Differentiable Optimal Transport for MoEficationcited by 0
- 2026SVD-Surgeon: Optimal Singular-Value Surgery for Large Language Model Compressioncited by 0
- 2026Compressing What Matters: Neuron Importance Meets Data-Aware Low Rank Approximation for Language Model Compressioncited by 0
- 2026Unifying Depth and Width Pruning for LLMs via Binary Knapsack Optimizationcited by 0
- 2026PruneShift: A Framework for Evaluating Decision Reliability in Structured Pruningcited by 0
- 2026XMerge: Cross-Axis Selection and Reconstructive Layer Merging for LLM Depth Compressioncited by 0
- 2026Output-aware Residual Stream Pruning for Large Language Modelscited by 0
- 2026Learning Functional Subspaces for Neural Network Compressioncited by 0
- 2026XOR-Trellis: Ultra-Low-Complexity Dequantization and Curvature-Aware Hadamard-Free LLM Quantizationcited by 0
- 2026ReCal: Calibrating Structured Pruning for On-Policy Distillation Recoverycited by 0
- 2026ReCal: Calibrating Structured Pruning for On-Policy Distillation Recovery8 Oct 2026
- 2026XOR-Trellis: Ultra-Low-Complexity Dequantization and Curvature-Aware Hadamard-Free LLM Quantization30 Sep 2026
- 2026Learning Functional Subspaces for Neural Network Compression30 Sep 2026
- 2026Output-aware Residual Stream Pruning for Large Language Models28 Sep 2026
- 2026XMerge: Cross-Axis Selection and Reconstructive Layer Merging for LLM Depth Compression2 Sep 2026
- 2026PruneShift: A Framework for Evaluating Decision Reliability in Structured Pruning30 Aug 2026
- 2026Unifying Depth and Width Pruning for LLMs via Binary Knapsack Optimization13 Aug 2026
- 2026Understanding Calibration and Truncation Error Propagation in Training-Free Low-Rank Compression for LLMs9 Aug 2026
- 2026Compressing What Matters: Neuron Importance Meets Data-Aware Low Rank Approximation for Language Model Compression30 Jun 2026
- 2026SVD-Surgeon: Optimal Singular-Value Surgery for Large Language Model Compression22 Jun 2026
- 2026DOT-MoE: Differentiable Optimal Transport for MoEfication1 Jun 2026
- 2026Reasoning-preserved Efficient Distillation of Large Language Models via Activation-aware Initialization28 May 2026
24 more
- 2026Optimal Brain Decomposition for Accurate LLM Low-Rank Approximation1 Apr 2026
- 2026FlattenGPT: Depth Compression for Transformer with Layer Flattening9 Feb 2026
- 2026Compressing LLMs with MoP: Mixture of Pruners5 Feb 2026
- 2025Elastic ViTs from Pretrained Models without Retraining20 Oct 2025
- 2025Accelerating Attention with Basis Decomposition2 Oct 2025
- 2025laplax -- Laplace Approximations with JAX22 Jul 2025
- 2025BLaST: High Performance Inference and Pretraining using BLock Sparse Transformers3 Jul 2025
- 2025SparseSSM: Efficient Selective Structured State Space Models Can Be Pruned in One-Shot11 Jun 2025
- 2025Olica: Efficient Structured Pruning of Large Language Models without Retraining10 Jun 2025
- 2025ReplaceMe: Network Simplification via Depth Pruning and Transformer Block Linearization5 May 2025
- 2025Efficient LLMs with AMP: Attention Heads and MLP Pruning29 Apr 2025
- 2025Skrr: Skip and Re-use Text Encoder Layers for Memory Efficient Text-to-Image Generation12 Feb 2025
- 2025EfficientLLM: Scalable Pruning-Aware Pretraining for Architecture-Agnostic Edge Language Models10 Feb 2025
- 2025Position: Curvature Matrices Should Be Democratized via Linear Operators31 Jan 2025
- 2025Pivoting Factorization: A Compact Meta Low-Rank Representation of Sparsity for Efficient Inference in Large Language Models31 Jan 2025
- 20252SSP: A Two-Stage Framework for Structured Pruning of LLMs29 Jan 2025
- 2025ToMoE: Converting Dense Large Language Models to Mixture-of-Experts through Dynamic Structural Pruning25 Jan 2025
- 2024DISP-LLM: Dimension-Independent Structural Pruning for Large Language Models15 Oct 2024
- 2024A deeper look at depth pruning of LLMs23 Jul 2024
- 2024Pruner-Zero: Evolving Symbolic Pruning Metric from scratch for Large Language Models5 Jun 2024
- 2024STAT: Shrinking Transformers After Training29 May 2024
- 2024HW-GPT-Bench: Hardware-Aware Architecture Benchmark for Language Models16 May 2024
- 2024OpenBA-V2: Reaching 77.3% High Compression Ratio with Fast Multi-Stage Pruning9 May 2024
- 2024Model Compression and Efficient Inference for Large Language Models: A Survey15 Feb 2024
- 2024Model Compression and Efficient Inference for Large Language Models: A Surveycited by 62
- 2024Pruner-Zero: Evolving Symbolic Pruning Metric from scratch for Large Language Modelscited by 40
- 2024DISP-LLM: Dimension-Independent Structural Pruning for Large Language Modelscited by 24
- 2024A deeper look at depth pruning of LLMscited by 23
- 20252SSP: A Two-Stage Framework for Structured Pruning of LLMscited by 12
- 2024HW-GPT-Bench: Hardware-Aware Architecture Benchmark for Language Modelscited by 10
- 2025EfficientLLM: Scalable Pruning-Aware Pretraining for Architecture-Agnostic Edge Language Modelscited by 10
- 2025ReplaceMe: Network Simplification via Depth Pruning and Transformer Block Linearizationcited by 10
- 2025Position: Curvature Matrices Should Be Democratized via Linear Operatorscited by 9
- 2025Olica: Efficient Structured Pruning of Large Language Models without Retrainingcited by 6
- 2025Efficient LLMs with AMP: Attention Heads and MLP Pruningcited by 5
- 2025laplax -- Laplace Approximations with JAXcited by 5
12 most cited of 31 All 31 (JSON)
- 2026ReCal: Calibrating Structured Pruning for On-Policy Distillation Recovery8 Oct 2026
- 2026XOR-Trellis: Ultra-Low-Complexity Dequantization and Curvature-Aware Hadamard-Free LLM Quantization30 Sep 2026
- 2026Learning Functional Subspaces for Neural Network Compression30 Sep 2026
- 2026Output-aware Residual Stream Pruning for Large Language Models28 Sep 2026
- 2026XMerge: Cross-Axis Selection and Reconstructive Layer Merging for LLM Depth Compression2 Sep 2026
- 2026PruneShift: A Framework for Evaluating Decision Reliability in Structured Pruning30 Aug 2026
- 2026Unifying Depth and Width Pruning for LLMs via Binary Knapsack Optimization13 Aug 2026
- 2026Understanding Calibration and Truncation Error Propagation in Training-Free Low-Rank Compression for LLMs9 Aug 2026
- 2026Compressing What Matters: Neuron Importance Meets Data-Aware Low Rank Approximation for Language Model Compression30 Jun 2026
- 2026SVD-Surgeon: Optimal Singular-Value Surgery for Large Language Model Compression22 Jun 2026
- 2026Optimal Brain Decomposition for Accurate LLM Low-Rank Approximation1 Apr 2026
- 2026FlattenGPT: Depth Compression for Transformer with Layer Flattening9 Feb 2026
12 newest of 31 All 31 (JSON)