Constrained Edge AI Deployment: Fine-Tuning vs Distillation for LLM Compression
Not on the map. Why?
Abstract (arXiv)
Modern foundational models are often compressed via a combination of structured pruning and re-training to meet the strict compute, memory, and connectivity constraints of edge deployments. While state-of-the-art pruning schemes target the entire Transformer, we adopt a simple, layer-wise L2-norm pruning on only the MLP blocks as a fixed baseline. Our focus is not on achieving maximal compression, but on isolating the impact of the re-training loss function: (i) Fine-tuning with Cross- Entropy (L2PFT), which requires labeled data, versus (ii) Self-Distillation with KL-divergence, which leverages only teacher logits (no labels) (L2PSD). We evaluate both pipelines on the OLMo2- 7B-SFT model for CommonsenseQA suitable for intermittent or denied connectivity scenarios typical of edge networks. Under identical pruning schedules, KL-based distillation matches or exceeds CE fine-tuning in test accuracy, demonstrating that, even with a basic MLP-only pruning, the choice of loss function materially affects compressed model recovery in resource-constrained environments.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 17 of its 25 references that are in the map. First posted on arXiv 13 May 2025.
Builds on
17 of 25 references.
References · 17
- 2017Attention Is All You Needcited by 60,361
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2023GPT-4 Technical Reportcited by 18,396
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,822
- 2023SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shotcited by 1,121
- 2017Empirical Analysis of the Hessian of Over-Parametrized Neural Networkscited by 290
- 2020Self-Distillation Amplifies Regularization in Hilbert Spacecited by 173
- 2023Neural Architecture Search: Insights from 1000 Paperscited by 139
- 2021Comparing Kullback-Leibler Divergence and Mean Squared Error Loss in Knowledge Distillationcited by 134
- 2017Stochastic gradient descent performs variational inference, converges to limit cycles for deep networkscited by 109
5 more
- 2024LLM Pruning and Distillation in Practice: The Minitron Approachcited by 82
- 2023A Survey on Uncertainty Quantification Methods for Deep Learningcited by 81
- 2024Self-Data Distillation for Recovering Quality in Pruned Large Language Modelscited by 8
- 2024EPSD: Early Pruning with Self-Distillation for Efficient Model Compressioncited by 6
- 2025On Accelerating Edge AI: Optimizing Resource-Constrained Environmentscited by 5
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026A Comparative Study of CNN Optimization Methods for Edge AI: Exploring the Role of Early Exits
- 2021Deep Neural Compression Via Concurrent Pruning and Self-Distillation
- 2022A Novel Architecture Slimming Method for Network Pruning and Knowledge Distillation
- 2026Prune-Quantize-Distill: An Ordered Pipeline for Efficient Neural Network Compression
- 2021PQK: Model Compression via Pruning, Quantization, and Knowledge Distillation
- 2023PERP: Rethinking the Prune-Retrain Paradigm in the Era of LLMs
- 2023Hardware-Aware DNN Compression via Diverse Pruning and Mixed-Precision Quantization
- 2025MI-to-Mid Distilled Compression (M2M-DC): An Hybrid-Information-Guided-Block Pruning with Progressive Inner Slicing Approach to Model Compression