mlsanityData to 14 Oct 2026

UniSD: Towards a Unified Self-Distillation Framework for Large Language Models

arXiv 2605.06597 ↗First posted 7 May 2026cs.CL cs.AI cs.LGcited by 7

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

  1. 2026Embarrassingly Simple Self-Distillation Improves Code Generation1 Apr
  2. 2024A Survey on Knowledge Distillation of Large Language Models20 Feb
  3. 2024DistiLLM: Towards Streamlined Distillation for Large Language Models6 Feb
  4. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakes23 Jun

Other references · 37

  1. 2024The Llama 3 Herd of Modelscited by 14,715
  2. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  3. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  4. 2021Evaluating Large Language Models Trained on Codecited by 8,320
  5. 2023Visual Instruction Tuningcited by 7,869
  6. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
  7. 2021Program Synthesis with Large Language Modelscited by 3,637
  8. 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,455
  9. 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,879
  10. 2022Learn to Explain: Multimodal Reasoning via Thought Chains for Science Question Answeringcited by 1,919
  11. 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,822
  12. 2019Energy and Policy Considerations for Deep Learning in NLPcited by 1,254
25 more
  1. 2024SimPO: Simple Preference Optimization with a Reference-Free Rewardcited by 963
  2. 2023A Comprehensive Survey of Continual Learning: Theory, Method and Applicationcited by 872
  3. 2019Green AIcited by 627
  4. 2019Quantifying the Carbon Emissions of Machine Learningcited by 521
  5. 2026Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Modelscited by 455
  6. 2023ToolAlpaca: Generalized Tool Learning for Language Models with 3000 Simulated Casescited by 393
  7. 2019Explain Yourself! Leveraging Language Models for Commonsense Reasoningcited by 367
  8. 2026Reinforcement Learning via Self-Distillationcited by 332
  9. 2026Self-Distillation Enables Continual Learningcited by 247
  10. 2020MiniLMv2: Multi-Head Self-Attention Relation Distillation for Compressing Pretrained Transformerscited by 245
  11. 2020How Context Affects Language Models' Factual Predictionscited by 181
  12. 2026Entropy-Aware On-Policy Distillation of Language Modelscited by 175
  13. 2026A Survey of On-Policy Distillation for Large Language Modelscited by 163
  14. 2022Instruction Induction: From Few Examples to Natural Language Task Descriptionscited by 152
  15. 2022Less is More: Task-aware Layer-wise Distillation for Language Model Compressioncited by 93
  16. 2026Privileged Information Distillation for Language Modelscited by 90
  17. 2023Visual Program Distillation: Distilling Tools and Programmatic Reasoning into Vision-Language Modelscited by 66
  18. 2023Federated Continual Learning via Knowledge Fusion: A Surveycited by 53
  19. 2026SCOPE: Signal-Calibrated On-Policy Distillation Enhancement with Dual-Path Adaptive Weightingcited by 43
  20. 2026Demystifying OPD: Length Inflation and Stabilization Strategies for Large Language Modelscited by 30
  21. 2026AgentArk: Distilling Multi-Agent Intelligence into a Single LLM Agentcited by 13
  22. 2026VLA-OPD: Bridging Offline SFT and Online RL for Vision-Language-Action Models via On-Policy Distillationcited by 13
  23. 2025Beyond Magic Words: Sharpness-Aware Prompt Evolving for Robust Large Language Models with TAREcited by 9
  24. 2023GPT4All: An Ecosystem of Open Source Compressed Language Modelscited by 8
  25. 2026MASCOT: Multi-Agent Socio-Collaborative Companion Systemscited by 7

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 7.

7 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026OPSDL: On-Policy Self-Distillation for Long-Context Language Models
  2. 2024Self-Distillation Bridges Distribution Gap in Language Model Fine-Tuning
  3. 2026Learning with Rare Success but Rich Feedback via Reflection-Enhanced Self-Distillation
  4. 2026Rethinking Selective Knowledge Distillation
  5. 2024Pre-training Distillation for Large Language Models: A Design Space Exploration
  6. 2024Dynamic Self-Distillation via Previous Mini-batches for Fine-tuning Small Language Models
  7. 2026Disentangling Self-Distillation: Measuring and Modeling Acquisition and Retention
  8. 2026Self-Distillation as a Performance Recovery Mechanism for LLMs: Counteracting Compression and Catastrophic Forgetting