Mismatch Matters: On-Policy Distillation Beyond Token Agreement
On the map: Knowledge distillation → On-policy and self-distillation for LLMs
Abstract (arXiv)
On-policy distillation (OPD) has emerged as a core component of modern LLM post-training pipelines, yet we reveal a failure mode: degenerate agreement, where students exploit repetitive loops to achieve near-perfect token agreement with the teacher despite globally flawed responses. We therefore shift our focus from agreement to teacher-student mismatch, and find that mismatch tokens can be mainly categorized into two types: student-excess tokens and student-deficit tokens. Student-excess tokens are generated by the student but assigned near-zero probability by the teacher; their log-ratio corrections grow unbounded and destabilize the update. Student-deficit tokens, in contrast, are preferred by the teacher but rarely sampled by the student; their absence blocks the transfer of the teacher's reasoning patterns. To tackle these mismatch directions, we propose TIDE (Token-level Independent Deficit-Excess correction), which applies bounded Hellinger shaping to suppress the most severe sampled excesses and an analytic teacher top-K injection to restore deficient probability mass without requiring deficit tokens to be sampled. Across mathematical reasoning benchmarks with multiple Qwen3 teacher-student pairs, TIDE consistently outperforms standard OPD and recent token-selection and reward-shaping baselines. Moreover, the gains of TIDE are more pronounced under strong teacher-student mismatch, where it improves Avg@8 from 6.9% to 20.3%, reduces average response length by a factor of 3.6, and substantially reduces formatting failures. Code is available at https://github.com/yzc-666/TIDE
Key facts. Cited by 10 papers in the map (data through 14 Oct 2026). Placed in On-policy and self-distillation for LLMs, one of 12 directions in Knowledge distillation. Builds on 35 of its 41 references that are in the map. First posted on arXiv 10 Aug 2026.
Builds on
35 of 41 references.
Closest ancestors · 23
Same direction, newest first
- 2026Rethinking On-Policy Self-Distillation for Thinking Models6 Jul
- 2026Multi-Turn On-Policy Distillation with Prefix Replay6 Jul
- 2026Prefix-Guided On-Policy Distillation: Mining Golden Trajectories from Rollouts20 Jun
- 2026PowerOPD: Stabilizing On-Policy Distillation with Bounded Power Transformation15 Jun
- 2026Escaping the KL Agreement Trap in On-Policy Distillation8 Jun
- 2026Trajectory-Refined Distillation7 Jun
- 2026OPRD: On-Policy Representation Distillation4 Jun
- 2026Filter, Then Reweight: Rethinking Optimization Granularity in On-Policy Distillation1 Jun
- 2026Trust Region On-Policy Distillation31 May
- 2026Your Teacher Can't Help You Here: Combating Supervision Fidelity Decay in On-Policy Distillation29 May
- 2026Not All Disagreement Is Learnable: Token Teachability in On-Policy Distillation26 May
- 2026Respecting Self-Uncertainty in On-Policy Self-Distillation for Efficient LLM Reasoning13 May
11 more
- 2026Teacher-Guided Policy Optimization for On-Policy Reasoning Distillation under Large Policy Divergence13 May
- 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes11 May
- 2026KL for a KL: On-Policy Distillation with Control Variate Baseline8 May
- 2026Prune-OPD: Efficient and Reliable On-Policy Distillation for Long-Horizon Reasoning8 May
- 2026Asymmetric On-Policy Distillation: Bridging Exploitation and Imitation at the Token Level7 May
- 2026TIP: Token Importance in On-Policy Distillation15 Apr
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe14 Apr
- 2026Demystifying OPD: Length Inflation and Stabilization Strategies for Large Language Models9 Apr
- 2026Scaling Reasoning Efficiently via Relaxed On-Policy Distillation11 Mar
- 2026Entropy-Aware On-Policy Distillation of Language Models7 Mar
- 2026Fast and Effective On-policy Distillation from Reasoning Prefixes16 Feb
Other references · 12
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2025Qwen3 Technical Reportcited by 8,820
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
- 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 803
- 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 660
- 2026GLM-5: from Vibe Coding to Agentic Engineeringcited by 587
- 2026MiMo-V2-Flash Technical Reportcited by 191
- 2025JustRL: Scaling a 1.5B LLM with a Simple RL Recipecited by 64
Built on by
Citing papers per year: 2026: 10.
10 papers.
- 2026Understanding On-Policy Distillation: A Mechanistic Interpretability Perspective via Sparse Crosscoderscited by 1
- 2026TV-Regulated OPD: Direction Matters in On-Policy Distillationcited by 0
- 2026Expected Reasoning-Step Return Unifies On-Policy Learning from Rewards and Teacherscited by 0
- 2026From Static to Dynamic: On-Policy Distillation from Image to Video Diffusion Modelscited by 0
- 2026The Missing Primitive: Diagnosing and Repairing Mathematical Reasoning in Large Language Modelscited by 0
- 2026When Do We Need On-Policy Distillation? Distilling on Offline Student Rollouts Is Often Bettercited by 0
- 2026DIAL-OPD: Learning More from Fewer Tokens in On-Policy Distillationcited by 0
- 2026MetaOPD: Meta-Learned Token Weighting for On-Policy Distillationcited by 0
- 2026Branching On-Policy Distillationcited by 0
- 2026Token Utility for On-Policy Distillation via Local Trust-Region Optimizationcited by 0
- 2026Branching On-Policy Distillation9 Oct 2026
- 2026MetaOPD: Meta-Learned Token Weighting for On-Policy Distillation8 Oct 2026
- 2026DIAL-OPD: Learning More from Fewer Tokens in On-Policy Distillation8 Oct 2026
- 2026When Do We Need On-Policy Distillation? Distilling on Offline Student Rollouts Is Often Better8 Oct 2026
- 2026Token Utility for On-Policy Distillation via Local Trust-Region Optimization1 Oct 2026
- 2026The Missing Primitive: Diagnosing and Repairing Mathematical Reasoning in Large Language Models1 Oct 2026
- 2026Understanding On-Policy Distillation: A Mechanistic Interpretability Perspective via Sparse Crosscoders28 Sep 2026
- 2026From Static to Dynamic: On-Policy Distillation from Image to Video Diffusion Models28 Sep 2026
- 2026Expected Reasoning-Step Return Unifies On-Policy Learning from Rewards and Teachers26 Sep 2026
- 2026TV-Regulated OPD: Direction Matters in On-Policy Distillation8 Sep 2026
Similar, not cited
Close in wording, no citation either way
- 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixes
- 2026SimpleOPD: Simple Tokenizer-Agnostic On-Policy Distillation for Long-Context Reasoning
- 2026Outcome-Confounded Local Supervision in On-Policy Distillation
- 2026Rethinking Cross-Tokenizer On-Policy Distillation: From Alignment Coverage to Supervision Reliability
- 2026SimCT: Recovering Lost Supervision for Cross-Tokenizer On-Policy Distillation
- 2026Overcoming Scaling Limits in On-Policy Self-Distillation for LLM Reasoning
- 2026Calibrating Teacher--Student Discrepancy for On-Policy Distillation
- 2026A Token-Level Analysis of Sampled-Token Reverse-KL On-Policy Distillation