mlsanityData to 14 Oct 2026

Mismatch Matters: On-Policy Distillation Beyond Token Agreement

arXiv 2608.09836 ↗First posted 10 Aug 2026cs.AI cs.CLcited by 10

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

  1. 2026Rethinking On-Policy Self-Distillation for Thinking Models6 Jul
  2. 2026Multi-Turn On-Policy Distillation with Prefix Replay6 Jul
  3. 2026Prefix-Guided On-Policy Distillation: Mining Golden Trajectories from Rollouts20 Jun
  4. 2026PowerOPD: Stabilizing On-Policy Distillation with Bounded Power Transformation15 Jun
  5. 2026Escaping the KL Agreement Trap in On-Policy Distillation8 Jun
  6. 2026Trajectory-Refined Distillation7 Jun
  7. 2026OPRD: On-Policy Representation Distillation4 Jun
  8. 2026Filter, Then Reweight: Rethinking Optimization Granularity in On-Policy Distillation1 Jun
  9. 2026Trust Region On-Policy Distillation31 May
  10. 2026Your Teacher Can't Help You Here: Combating Supervision Fidelity Decay in On-Policy Distillation29 May
  11. 2026Not All Disagreement Is Learnable: Token Teachability in On-Policy Distillation26 May
  12. 2026Respecting Self-Uncertainty in On-Policy Self-Distillation for Efficient LLM Reasoning13 May
11 more
  1. 2026Teacher-Guided Policy Optimization for On-Policy Reasoning Distillation under Large Policy Divergence13 May
  2. 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes11 May
  3. 2026KL for a KL: On-Policy Distillation with Control Variate Baseline8 May
  4. 2026Prune-OPD: Efficient and Reliable On-Policy Distillation for Long-Horizon Reasoning8 May
  5. 2026Asymmetric On-Policy Distillation: Bridging Exploitation and Imitation at the Token Level7 May
  6. 2026TIP: Token Importance in On-Policy Distillation15 Apr
  7. 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe14 Apr
  8. 2026Demystifying OPD: Length Inflation and Stabilization Strategies for Large Language Models9 Apr
  9. 2026Scaling Reasoning Efficiently via Relaxed On-Policy Distillation11 Mar
  10. 2026Entropy-Aware On-Policy Distillation of Language Models7 Mar
  11. 2026Fast and Effective On-policy Distillation from Reasoning Prefixes16 Feb

Other references · 12

  1. 2017Proximal Policy Optimization Algorithmscited by 15,215
  2. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  3. 2025Qwen3 Technical Reportcited by 8,820
  4. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
  5. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
  6. 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
  7. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
  8. 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 803
  9. 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 660
  10. 2026GLM-5: from Vibe Coding to Agentic Engineeringcited by 587
  11. 2026MiMo-V2-Flash Technical Reportcited by 191
  12. 2025JustRL: Scaling a 1.5B LLM with a Simple RL Recipecited by 64

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 10.

10 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixes
  2. 2026SimpleOPD: Simple Tokenizer-Agnostic On-Policy Distillation for Long-Context Reasoning
  3. 2026Outcome-Confounded Local Supervision in On-Policy Distillation
  4. 2026Rethinking Cross-Tokenizer On-Policy Distillation: From Alignment Coverage to Supervision Reliability
  5. 2026SimCT: Recovering Lost Supervision for Cross-Tokenizer On-Policy Distillation
  6. 2026Overcoming Scaling Limits in On-Policy Self-Distillation for LLM Reasoning
  7. 2026Calibrating Teacher--Student Discrepancy for On-Policy Distillation
  8. 2026A Token-Level Analysis of Sampled-Token Reverse-KL On-Policy Distillation