arXiv AI By Sirui Chen, Yunzhe Qi, Haobo Xu, Jiaru Zou, Mengting Ai, Zihao Li, Jingrui He

Proposing Better Rollouts for On-Policy Distillation

Read the original on arXiv AI →

The paper proposes Mixture Proposal On-Policy Distillation (MPOPD) to address a mismatch issue in on‑policy distillation (OPD) where the rollout policy is fixed to the current student. By treating the rollout policy as an optimization target, MPOPD derives a geometric mixture of student and teacher policies and introduces a proposal policy for efficient rollout generation. Experiments on 16 reasoning benchmarks show consistent gains over OPD, notably a 6.91% accuracy improvement on math reasoning with Qwen3-4B-Base and a 41% reduction in generation length.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

arXiv AI
Sep 17

Trajectory Learnability for Offline On-Policy Distillation with Imperfect Teachers

The paper introduces a method for offline on‑policy distillation that addresses the problem of imperfect teacher supervision. By training on teacher‑successful problems and measuring changes in token likelihoods on teacher‑failed trajectories, the authors derive a learnability signal that weights the distillation loss. This approach improves performance on mathematical reasoning and code generation tasks while reducing computational cost compared to online distillation.

By Yihao Ai, Weilong Yan
arXiv AI
Oct 5

Spend Teacher Tokens Where They Matter: Success-Referenced On-Policy Distillation

The paper introduces Success-Referenced On-Policy Distillation (SR-OPD), a method that reduces teacher computation in on-policy distillation by selectively providing teacher supervision only for prompts where the student has both successful and failed rollouts. SR-OPD uses successful rollouts as references to prioritize failed rollouts that diverge significantly from the successful ones, while also considering estimated teacher-input cost. Experiments on three teacher-student pairs across six mathematical reasoning benchmarks show that SR-OPD requires only 3.46‑5.02% of the teacher-input tokens of Vanilla OPD in a one-pass setting, yet maintains comparable reasoning performance, and further validates its design choices under a 5% teacher-input budget.

By Xiang Chen, Futao Su, Kong Wang, Jiayi Chen, TanLin Li