Why On-Policy Distillation Sometimes Fails: Vanishing Learning Signals
On the map: Knowledge distillation → On-policy and self-distillation for LLMs
Abstract (arXiv)
On-policy distillation (OPD) enables effective capability transfer between language models, yet the mechanisms underlying its failures are not fully understood. Across code generation and mathematical reasoning, OPD with larger-scale teachers exhibits early loss plateaus, with an average final loss reduction of 25.1% after 200 updates, compared with 96.2% for self-RL teachers, obtained by further reinforcement learning (RL) training of the initial student. To understand this difference, we analyze OPD as an idealized continuous-time dynamical system in the small-learning-rate limit. Our training-log diagnostics associate these plateaus with an early decline in a gradient-based learning-signal proxy while substantial loss remains; these measurements do not establish why the underlying gradient weakens. We further prove a local recovery guarantee for teachers sufficiently close to the initial student in a shared parameterization under regularity conditions, offering a conditional explanation for the success of self-RL teachers in our experiments. Across runs with and without loss plateaus, we observe small relative parameter changes (0.025-0.098%) and high similarity between the student's representations before and after OPD (linear CKA >0.98 across layers). These observations suggest that limited representation adaptation may contribute to learning-signal collapse, a hypothesis that remains to be tested. Code is available at https://github.com/leizhao7/opd-learning-signals.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in On-policy and self-distillation for LLMs, one of 12 directions in Knowledge distillation. Builds on 23 of its 52 references that are in the map. First posted on arXiv 8 Oct 2026.
Builds on
23 of 52 references.
Closest ancestors · 13
Same direction, newest first
- 2026Rethinking On-Policy Distillation of Large Language Models II: One Training Example3 Sep
- 2026Demystifying On-Policy Distillation: Roles, Pathologies, and Regulations15 Jul
- 2026When Does Online Imitation Learning Help in LLM Post-Training? The Role of (Non-)Realizability Beyond Horizon29 Jun
- 2026On the Geometry of On-Policy Distillation5 Jun
- 2026Not All Disagreement Is Learnable: Token Teachability in On-Policy Distillation26 May
- 2026Learning to Foresee: Unveiling the Unlocking Efficiency of On-Policy Distillation12 May
- 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes11 May
- 2026Unmasking On-Policy Distillation: Where It Helps, Where It Hurts, and Why11 May
- 2026Cornerstones or Stumbling Blocks? Deciphering the Rock Tokens in On-Policy Distillation10 May
- 2026Asymmetric On-Policy Distillation: Bridging Exploitation and Imitation at the Token Level7 May
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe14 Apr
- 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixes26 Mar
1 more
Other references · 10
- 2025Qwen3 Technical Reportcited by 8,820
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
- 2025Process Reinforcement through Implicit Rewardscited by 504
- 2026MiMo-V2-Flash Technical Reportcited by 191
- 2025Skywork Open Reasoner 1 Technical Reportcited by 165
- 2024Multi-IF: Benchmarking LLMs on Multi-Turn and Multilingual Instructions Followingcited by 102
- 2025JustRL: Scaling a 1.5B LLM with a Simple RL Recipecited by 64
- 2025The Path Not Taken: RLVR Provably Learns Off the Principalscited by 56
- 2025Reinforcement Learning Finetunes Small Subnetworks in Large Language Modelscited by 49
- 2026PRISM: Demystifying Retention and Interaction in Mid-Trainingcited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Scaling Properties of Same-Family On-Policy Distillation
- 2026Gains and Collapse in On-Policy Distillation:A Reinforcement Learning Perspective
- 2026Understanding Off- vs On-Policy Distillation: A Tale of Distinct Training Objectives
- 2026Correcting or Continuing? What On-Policy Distillation Learns from Student Errors
- 2026Does On-Policy Distillation Really Distill? From Noisy Teacher to Self-Improvement
- 2026AsyncOPD: How Stale Can On-Policy Distillation Be?
- 2026One Symptom, Three Levers: A Critical Review of On-Policy Self-Distillation
- 2026A Predictive Law for On-Policy Self-Distillation From World Feedback