Search papers, labs, and topics across Lattice.
This paper identifies a critical failure mode in on-policy distillation (OPD) where students achieve token agreement with teachers through repetitive loops, leading to flawed responses. The authors introduce TIDE (Token-level Independent Deficit-Excess correction), which addresses the issue of teacher-student mismatch by categorizing tokens into student-excess and student-deficit types and applying bounded Hellinger shaping along with teacher top-$K$ injection. TIDE significantly enhances performance on mathematical reasoning tasks, improving accuracy from 6.9% to 20.3% under conditions of strong mismatch while also reducing response length and formatting errors.
Teacher-student mismatch can lead to flawed outputs, but TIDE's innovative correction method boosts reasoning accuracy by over 200% in challenging scenarios.
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