English

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

Computation and Language 2026-05-22 v2 Artificial Intelligence Machine Learning

Abstract

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.

Keywords

Cite

@article{arxiv.2605.06597,
  title  = {UniSD: Towards a Unified Self-Distillation Framework for Large Language Models},
  author = {Yiqiao Jin and Yiyang Wang and Lucheng Fu and Yijia Xiao and Yinyi Luo and Haoxin Liu and B. Aditya Prakash and Josiah Hester and Jindong Wang and Srijan Kumar},
  journal= {arXiv preprint arXiv:2605.06597},
  year   = {2026}
}

Comments

Website: https://unifiedsd.github.io/ Code: https://github.com/Ahren09/UniSD

R2 v1 2026-07-01T12:55:39.297Z