mlsanityData to 7 Oct 2026

The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes

arXiv 2605.11182 ↗First posted 11 May 2026cs.AIcited by 37

Not on the map. Why?

Abstract (arXiv)

On-policy distillation (OPD) and on-policy self-distillation (OPSD) have emerged as promising post-training methods for large language models, offering dense token-level supervision on trajectories sampled from the model's own policy. However, existing results on their effectiveness remain mixed: while OP(S)D has shown promise in system prompt and knowledge internalization, recent studies also report instability and degradation. In this work, we present a comprehensive empirical study of when OPD and OPSD work, when they fail, and why. We find that OPD on mathematical reasoning is highly sensitive to teacher choice and loss formulation, whereas OPSD fails in our tested settings due to test-time absence of instance-specific privileged information (PI). In contrast, OPSD is effective when PI represents a shared latent rule, such as a system prompt or alignment preference. We identify three failure mechanisms: (1) distribution mismatch between teacher and student caused by conditioning on student-generated prefixes, (2) optimization instability from biased TopK reverse-KL gradients, and (3) an OPSD-specific limitation where the student learns a PI-free policy that aggregates PI-conditioned teachers, which is insufficient when PI is instance-specific. We further show that stop-gradient TopK objectives, RLVR-adapted teachers, and SFT-stabilized students mitigate these failures.

Key facts. Cited by 37 papers in the map (data through 7 Oct 2026). Builds on 9 of its 13 references that are in the map. First posted on arXiv 11 May 2026.

Builds on

9 of 13 references.

References · 9

  1. 2017Proximal Policy Optimization Algorithmscited by 14,441
  2. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
  3. 2019Persuasion for Good: Towards a Personalized Persuasive Dialogue System for Social Goodcited by 207
  4. 2026Entropy-Aware On-Policy Distillation of Language Modelscited by 151
  5. 2026Why Does Self-Distillation (Sometimes) Degrade the Reasoning Capability of LLMs?cited by 87
  6. 2023Emotionally Numb or Empathetic? Evaluating How LLMs Feel Using EmotionBenchcited by 54
  7. 2026CRISP: Compressed Reasoning via Iterative Self-Policy Distillationcited by 49
  8. 2021Greedification Operators for Policy Optimization: Investigating Forward and Reverse KL Divergencescited by 40
  9. 2026Demystifying OPD: Length Inflation and Stabilization Strategies for Large Language Modelscited by 24

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 37.

37 papers.

Sort:

By area:

  1. 2026RLCSD: Reinforcement Learning with Contrastive On-Policy Self-Distillationcited by 33
  2. 2026Rethinking On-Policy Distillation of Large Language Models II: One Training Examplecited by 15
  3. 2026Mismatch Matters: On-Policy Distillation Beyond Token Agreementcited by 7
  4. 2026Keep Policy Gradient in Charge: Sibling-Guided Credit Distillation for Long-Horizon Tool-Use Agentscited by 6
  5. 2026Rethinking Privileged Information in On-Policy Self-Distillationcited by 6
  6. 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillationcited by 5
  7. 2026Diagnosing and Mitigating Thinking Collapse in On-Policy Self-Distillationcited by 5
  8. 2026Towards Understanding On-Policy Distillation through the Lens of Test-Time Scalingcited by 5
  9. 2026CriPO: Enhancing Rubric-based RL via Self-Distillationcited by 4
  10. 2026MAIGO: Mitigating Lost-in-Conversation with History-Cleaned On-Policy Self-Distillationcited by 3
  11. 2026The Physics of Multi-Turn Long-Horizon Planning: From Pre-training to Post-training via Single- and Multi-Teacher On-Policy Agentic Distillationcited by 3
  12. 2026Privileged Likelihood Is Not Automatically Value: Three Checks for Token Credit in On-Policy Self-Distillationcited by 3
25 more
  1. 2026Tail-Aware Top-k On-Policy Distillationcited by 3
  2. 2026RISE: Recursive Improvement via Self-Extrapolating Policy Distillationcited by 3
  3. 2026Z-Reward: Beyond Scalar Rewards by Internalizing Reasoning into Score Distributionscited by 2
  4. 2026Trust the Right Teacher: Quality-Aware Self-Distillation for GUI Groundingcited by 2
  5. 2026DuDi: Dual-Signal Distillation with Cross-Lingual Verbalizercited by 1
  6. 2026DASH-OPD: Discrepancy-Aware Switching with Hysteresis for On-Policy Distillationcited by 1
  7. 2026Le Critique: Privileged Value Functions for LLM Reinforcement Learningcited by 1
  8. 2026When Teacher Guidance Misleads: Reward-Aligned On-Policy Distillationcited by 1
  9. 2026SR-OPSD: Self-Referenced On-Policy Self-Distillationcited by 0
  10. 2026SMOPD: Selective Token-Entropy Masking for Dirty-History Multi-Turn On-Policy Self-Distillationcited by 0
  11. 2026Distillation as Probability Transport: Routed On-Policy Distillationcited by 0
  12. 2026Understanding the Synergy between SFT, RLVR, and OPD in LLM Post-Trainingcited by 0
  13. 2026Teach Yourself Where to Look: On-Policy Attention Self-Distillation for Reasoningcited by 0
  14. 2026DuoOPD: Learning from Joint Teacher-Student Outcomes for Multi-Task On-Policy Distillationcited by 0
  15. 2026UOPD: Uncertainty-Aware Intervention for On-Policy Distillation of Multi-Turn Agentscited by 0
  16. 2026Unbiased Top-k Estimation for On-Policy Distillationcited by 0
  17. 2026No Pain, More Gain: Iterative Merging for Effective Multi-Teacher On-Policy Distillationcited by 0
  18. 2026Understanding On-Policy Distillation: A Mechanistic Interpretability Perspective via Sparse Crosscoderscited by 0
  19. 2026On-Policy or Off-Policy Learning? A Systematic Study of Distillation Dynamicscited by 0
  20. 2026Learning from Teacher Continuations at Student Statescited by 0
  21. 2026Diagnosing On-Policy Self-Distillation for Reasoning Language Modelscited by 0
  22. 2026From Given to Gathered Evidence: Agentic Learning for Longitudinal Medical Reasoningcited by 0
  23. 2026UBTree: Parallel Tree Drafting via Unigram and Bigram Models for Speculative Decodingcited by 0
  24. 2026From Gradients to Capabilities: Understanding Multi-Teacher On-Policy Distillationcited by 0
  25. 2026Why On-Policy Distillation Sometimes Fails: Vanishing Learning Signalscited by 0

Similar, not cited

Close in wording, no citation either way

  1. 2026A Brief Overview: On-Policy Self-Distillation In Large Language Models
  2. 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe
  3. 2026Uni-OPD: Unifying On-Policy Distillation with a Dual-Perspective Recipe
  4. 2026Unmasking On-Policy Distillation: Where It Helps, Where It Hurts, and Why
  5. 2026On-Policy Self-Distillation without Any Supervision
  6. 2026Demystifying On-Policy Distillation: Roles, Pathologies, and Regulations
  7. 2026Overcoming Scaling Limits in On-Policy Self-Distillation for LLM Reasoning
  8. 2026Train Ahead, Distill Back: Bootstrapping On-Policy Self-Distillation for Large Language Models