UniSD: Towards a Unified Self-Distillation Framework for Large Language Models
On the map: Knowledge distillation → Knowledge distillation for LLMs
Abstract (arXiv)
Self-distillation (SD) offers a promising path for adapting large language models (LLMs) without relying on stronger external teachers. However, SD in autoregressive LLMs remains challenging because self-generated trajectories are free-form, correctness is task-dependent, and plausible rationales can still provide unstable or unreliable supervision. Existing methods mainly examine isolated design choices, leaving their effectiveness, roles, and interactions unclear. In this paper, we propose UniSD, a unified framework to systematically study self-distillation. UniSD integrates complementary mechanisms that address supervision reliability, representation alignment, and training stability, including multi-teacher agreement, EMA teacher stabilization, token-level contrastive learning, feature matching, and divergence clipping. Across six benchmarks and six models from three model families, UniSD reveals when self-distillation improves over static imitation, which components drive the gains, and how these components interact across tasks. Guided by these insights, we construct UniSDfull, an integrated pipeline that combines complementary components and achieves the strongest overall performance, improving over the base model by +5.4 points and the strongest baseline by +2.8 points. Extensive evaluation highlights self-distillation as a practical and steerable approach for efficient LLM adaptation without stronger external teachers.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Placed in Knowledge distillation for LLMs, one of 12 directions in Knowledge distillation. Builds on 41 of its 53 references that are in the map. First posted on arXiv 7 May 2026.
Builds on
41 of 53 references.
Closest ancestors · 4
Same direction, newest first
- 2026Embarrassingly Simple Self-Distillation Improves Code Generation1 Apr
- 2024A Survey on Knowledge Distillation of Large Language Models20 Feb
- 2024DistiLLM: Towards Streamlined Distillation for Large Language Models6 Feb
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakes23 Jun
Other references · 37
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2021Evaluating Large Language Models Trained on Codecited by 8,320
- 2023Visual Instruction Tuningcited by 7,869
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2021Program Synthesis with Large Language Modelscited by 3,637
- 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,455
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,879
- 2022Learn to Explain: Multimodal Reasoning via Thought Chains for Science Question Answeringcited by 1,919
- 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,822
- 2019Energy and Policy Considerations for Deep Learning in NLPcited by 1,254
25 more
- 2024SimPO: Simple Preference Optimization with a Reference-Free Rewardcited by 963
- 2023A Comprehensive Survey of Continual Learning: Theory, Method and Applicationcited by 872
- 2019Green AIcited by 627
- 2019Quantifying the Carbon Emissions of Machine Learningcited by 521
- 2026Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Modelscited by 455
- 2023ToolAlpaca: Generalized Tool Learning for Language Models with 3000 Simulated Casescited by 393
- 2019Explain Yourself! Leveraging Language Models for Commonsense Reasoningcited by 367
- 2026Reinforcement Learning via Self-Distillationcited by 332
- 2026Self-Distillation Enables Continual Learningcited by 247
- 2020MiniLMv2: Multi-Head Self-Attention Relation Distillation for Compressing Pretrained Transformerscited by 245
- 2020How Context Affects Language Models' Factual Predictionscited by 181
- 2026Entropy-Aware On-Policy Distillation of Language Modelscited by 175
- 2026A Survey of On-Policy Distillation for Large Language Modelscited by 163
- 2022Instruction Induction: From Few Examples to Natural Language Task Descriptionscited by 152
- 2022Less is More: Task-aware Layer-wise Distillation for Language Model Compressioncited by 93
- 2026Privileged Information Distillation for Language Modelscited by 90
- 2023Visual Program Distillation: Distilling Tools and Programmatic Reasoning into Vision-Language Modelscited by 66
- 2023Federated Continual Learning via Knowledge Fusion: A Surveycited by 53
- 2026SCOPE: Signal-Calibrated On-Policy Distillation Enhancement with Dual-Path Adaptive Weightingcited by 43
- 2026Demystifying OPD: Length Inflation and Stabilization Strategies for Large Language Modelscited by 30
- 2026AgentArk: Distilling Multi-Agent Intelligence into a Single LLM Agentcited by 13
- 2026VLA-OPD: Bridging Offline SFT and Online RL for Vision-Language-Action Models via On-Policy Distillationcited by 13
- 2025Beyond Magic Words: Sharpness-Aware Prompt Evolving for Robust Large Language Models with TAREcited by 9
- 2023GPT4All: An Ecosystem of Open Source Compressed Language Modelscited by 8
- 2026MASCOT: Multi-Agent Socio-Collaborative Companion Systemscited by 7
Built on by
Citing papers per year: 2026: 7.
7 papers.
- 2026RLCSD: Reinforcement Learning with Contrastive On-Policy Self-Distillationcited by 45
- 2026On-Policy Self-Distillation without Any Supervisioncited by 17
- 2026DemoPSD: Disagreement-Modulated Policy Self-Distillationcited by 6
- 2026SMOPD: Selective Token-Entropy Masking for Dirty-History Multi-Turn On-Policy Self-Distillationcited by 0
- 2026Train Ahead, Distill Back: Bootstrapping On-Policy Self-Distillation for Large Language Modelscited by 0
- 2026Science Utopia? Closed-Loop LLM Simulation of Academic Research Ecosystemscited by 0
- 2026Privileged Context as Drift in On-Policy Self-Distillationcited by 0
- 2026Privileged Context as Drift in On-Policy Self-Distillation6 Oct 2026
- 2026Science Utopia? Closed-Loop LLM Simulation of Academic Research Ecosystems1 Oct 2026
- 2026Train Ahead, Distill Back: Bootstrapping On-Policy Self-Distillation for Large Language Models29 Sep 2026
- 2026On-Policy Self-Distillation without Any Supervision6 Aug 2026
- 2026SMOPD: Selective Token-Entropy Masking for Dirty-History Multi-Turn On-Policy Self-Distillation30 Jul 2026
- 2026DemoPSD: Disagreement-Modulated Policy Self-Distillation2 Jul 2026
- 2026RLCSD: Reinforcement Learning with Contrastive On-Policy Self-Distillation10 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2026OPSDL: On-Policy Self-Distillation for Long-Context Language Models
- 2024Self-Distillation Bridges Distribution Gap in Language Model Fine-Tuning
- 2026Learning with Rare Success but Rich Feedback via Reflection-Enhanced Self-Distillation
- 2026Rethinking Selective Knowledge Distillation
- 2024Pre-training Distillation for Large Language Models: A Design Space Exploration
- 2024Dynamic Self-Distillation via Previous Mini-batches for Fine-tuning Small Language Models
- 2026Disentangling Self-Distillation: Measuring and Modeling Acquisition and Retention
- 2026Self-Distillation as a Performance Recovery Mechanism for LLMs: Counteracting Compression and Catastrophic Forgetting