From Pruning to Grafting: Dynamic Knowledge Redistribution via Learnable Layer Fusion
Not on the map. Why?
Abstract (arXiv)
Structured pruning of Generative Pre-trained Transformers (GPTs) offers a promising path to efficiency but often suffers from irreversible performance degradation due to the discarding of transformer blocks. In this paper, we introduce FuseGPT, a compression paradigm that reframes structured pruning as iterative knowledge grafting rather than simple removal. Motivated by the observation that linear block merging fails to capture non-linear feature disparities and that block importance fluctuates dynamically during pruning, FuseGPT employs a dual-strategy pipeline. First, we propose Macro Influence (MI), a dynamic fusion-aware metric that continuously re-evaluates block redundancy as the network topology evolves. Second, instead of rigid parameter averaging, we introduce a learnable low-rank fusion mechanism that adaptively grafts the knowledge of pruned blocks onto surviving layers via lightweight local distillation. Extensive experiments on LLaMA, Mistral, Qwen, and Phi families demonstrate that FuseGPT establishes a new state-of-the-art on the compression-accuracy Pareto frontier: at 25% sparsity, FuseGPT achieves lower perplexity than prior methods at 20% sparsity, improves zero-shot reasoning by up to 4.5 points, and delivers 1.33× inference speedup with 25% memory reduction. Furthermore, FuseGPT is orthogonal to quantization, achieving 52.1% total compression with negligible quality loss when combined with 4-bit GPTQ. We make our code publicly available at https://github.com/JarvisPei/FuseGPT.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 39 of its 47 references that are in the map. First posted on arXiv 21 Nov 2024.
Builds on
39 of 47 references.
References · 39
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2023Visual Instruction Tuningcited by 7,869
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2023Improved Baselines with Visual Instruction Tuningcited by 4,418
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
27 more
- 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,611
- 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,512
- 2016Pointer Sentinel Mixture Modelscited by 3,096
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
- 2019HuggingFace's Transformers: State-of-the-art Natural Language Processingcited by 2,218
- 2015Learning both Weights and Connections for Efficient Neural Networkscited by 2,031
- 2022GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformerscited by 1,919
- 2023AWQ: Activation-aware Weight Quantization for LLM Compression and Accelerationcited by 1,298
- 2023SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shotcited by 1,121
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
- 2021A Survey of Quantization Methods for Efficient Neural Network Inferencecited by 815
- 2023LLM-Pruner: On the Structural Pruning of Large Language Modelscited by 650
- 2017Learning Sparse Neural Networks through L0 Regularizationcited by 583
- 2021Sparsity in Deep Learning: Pruning and growth for efficient inference and training in neural networkscited by 527
- 2022ZeroQuant: Efficient and Affordable Post-Training Quantization for Large-Scale Transformerscited by 526
- 2024SliceGPT: Compress Large Language Models by Deleting Rows and Columnscited by 385
- 2024ShortGPT: Layers in Large Language Models are More Redundant Than You Expectcited by 383
- 2021Post-Training Quantization for Vision Transformercited by 278
- 2019Structured Pruning of Large Language Modelscited by 203
- 2022Structured Pruning Learns Compact and Accurate Modelscited by 170
- 2024LaCo: Large Language Model Pruning via Layer Collapsecited by 126
- 2022The Optimal BERT Surgeon: Scalable and Accurate Second-Order Pruning for Large Language Modelscited by 115
- 2024SLEB: Streamlining LLMs through Redundancy Verification and Elimination of Transformer Blockscited by 97
- 2024Shortened LLaMA: Depth Pruning for Large Language Models with Comparison of Retraining Methodscited by 82
- 2024EvoPress: Accurate Dynamic Model Compression via Evolutionary Searchcited by 20
- 2024Pruning via Merging: Compressing LLMs via Manifold Alignment Based Layer Mergingcited by 19
- 2024FoldGPT: Simple and Effective Large Language Model Compression Schemecited by 9
Built on by
Citing papers per year: 2025: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2021Greedy-layer Pruning: Speeding up Transformer Models for Natural Language Processing
- 2022A Novel Architecture Slimming Method for Network Pruning and Knowledge Distillation
- 2026Post-Training Probability Manifold Correction via Structured SVD Pruning and Self-Referential Distillation
- 2024RL-Pruner: Structured Pruning Using Reinforcement Learning for CNN Compression and Acceleration
- 2025Teacher-Guided One-Shot Pruning via Context-Aware Knowledge Distillation
- 2025GoPrune: Accelerated Structured Pruning with ℓ2,p-Norm Optimization
- 2025Strategic Fusion Optimizes Transformer Compression
- 2026On the Limits of Layer Pruning for Generative Reasoning in Large Language Models