Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Your Teacher Can't Help You Here: Combating Supervision Fidelity Decay in On-Policy Distillation

About

On-policy distillation transfers reasoning capabilities by training a student model on its own generated trajectories using token-level feedback from a teacher. However, we identify a critical bottleneck, \textbf{Supervision Fidelity Decay (SFD)}: as student-generated prefixes lengthen, the teacher's next-token distribution becomes less confident and less discriminative. Consequently, the teacher-dependent corrective signal in reverse-KL distillation weakens, causing student drift to compound across long reasoning chains. To mitigate SFD, we introduce \textbf{Lookahead Group Reward (\ours{})}. Building on the insight that next-step teacher confidence reflects the discriminative strength of future reverse-KL supervision, \ours{} evaluates the student's top-K candidate tokens by the teacher confidence they induce at the subsequent step and assigns a group-normalized reward. To maintain computational efficiency, we further design an entropy-triggered tree-attention mechanism. Across six math and code benchmarks, \ours{} improves mean@8 by \textbf{2.57} points over OPD for a 7B student, with gains increasing in longer-generation and reaching +\textbf{4.92} points on AIME-26 at 39k tokens.

Yanjiang Liu, Jie Lou, Xinyan Guan, Yuqiu Ji, Hongyu Lin, Ben He, Xianpei Han, Le Sun, Xing Yu, Yaojie Lu• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningAIME 25--
112
Mathematical ReasoningAIME 24
Pass Rate@883.33
55
Mathematical ReasoningAIME 26
Mean@853.75
28
Mathematical ReasoningHMMT 25
Mean@831.67
18
Mathematical ReasoningHMMT 26
Mean@834.85
18
Showing 5 of 5 rows

Other info

Follow for update