arXiv Machine Learning

CoRD: Coupled Rollouts Fill the Blind Spot of On-Policy Distillation

CoRD (Coupled Rollout Distillation) enhances on‑policy distillation by aligning the student’s training state distribution with the teacher’s through a geometric interpolation of their rollouts, using maximal coupling to keep tokens where the distributions agree and substitute tokens otherwise. This approach keeps the on‑policy loss intact while reducing supervision gaps, leading to higher pass@1 and pass@8 scores across multiple Qwen3 and Qwen3.5 student models on public math benchmarks. Notably, a 4B student trained with CoRD closes performance gaps in just 16 steps, far faster than on‑policy distillation.

arXiv Computation and Language
Oct 9

When Do We Need On-Policy Distillation? Distilling on Offline Student Rollouts Is Often Better

The paper investigates whether on‑policy distillation (OPD) is always advantageous for transferring knowledge from a teacher to a student model. It introduces Semi‑OPD, which uses offline rollouts from the initial student, and demonstrates that across 17 teacher‑student pairs (1.5B–235B parameters) Semi‑OPD outperforms OPD in 14 cases, achieving up to +13.6% accuracy and an 11.4× speedup. The authors show that the benefit of OPD depends on the alignment between teacher and student, measured by an output‑token overlap ratio, and argue that effective distillation requires on‑policy behavior relative to both models.

By Siyan Zhao, Yonggan Fu, Jindong Jiang, Shih-Yang Liu, Song Bian, Byung-Kwan Lee, Sharath Turuvekere Sreenivas, Wenliang Dai, Hanrong Ye, Aditya Grover, Pavlo Molchanov
arXiv Machine Learning
1d ago

Learning Dynamics of On-Policy Distillation

The paper investigates the learning dynamics of on‑policy distillation (OPD), a method that combines teacher supervision with training on student‑generated trajectories. It identifies two stages of the feedback loop: converting the distillation signal into parameter updates, and broadcasting those updates to alter predictions and reshape the rollout distribution. The study explains why students may be unresponsive to teacher disagreement and how updates that improve local agreement can lead to degeneration by increasing the likelihood of poor continuations. It also shows that offline distillation can prepare a student for OPD without boosting immediate accuracy, yet OPD can still yield significant accuracy gains after sufficient offline preparation.

By Guan Zhe Hong, Sumeet Ramesh Motwani, Liang Chen, Yangchen Pan, Junchi Yu, Jialin Yu, Philip Torr, Yi Ren
arXiv Machine Learning
Oct 23

When to Follow the Teacher: Advantage-Based On-Policy Distillation

The paper introduces Advantage-Based On-Policy Distillation (AOPD), a method that selectively weights teacher actions during online policy distillation by their counterfactual advantage to the student. Unlike standard OPD, which follows the teacher at every state, AOPD reinforces only those teacher actions that improve the student's expected outcome, using an estimator that relies solely on standard student rollouts and terminal rewards. Experiments on math, coding, and Medqa tasks demonstrate that AOPD outperforms both distillation and reinforcement learning baselines in external-teacher and self-distillation settings.

By Mahdi Sabbaghi, George Pappas, Hamed Hassani, Adel Javanmard
arXiv AI
Jun 16

On-Policy Distillation with Curriculum Turn-level Guidance for Multi-turn Agents

arXiv:2606. 15912v1 Announce Type: cross Abstract: Multi-turn agents that plan, invoke tools, and interact with environments offer a promising paradigm for solving complex tasks, yet their capabilities typically rely on very large models whose inference cost is prohibitive in practice.

By Gengsheng Li, Mao Zheng, Mingyang Song, Ruiqi Liu, Tianyu Yang, Jie Sun, Qiyong Zhong, Haiyun Guo, Junfeng Fang, Dan Zhang, Jinqiao Wang
arXiv AI
Oct 15

Correcting or Continuing? What On-Policy Distillation Learns from Student Errors

The paper investigates on‑policy distillation (OPD), a method that trains a student model using its own rollouts while a teacher remains frozen, to see if it can help a student improve on problems it currently fails. Experiments with Qwen‑3 and Phi‑4 models on a competition‑mathematics benchmark show that OPD actually degrades performance when training focuses on failure cases, even though training loss (KL divergence) continues to drop. However, when OPD is applied to a broader set of problems, it improves accuracy by 4.6 percentage points, indicating that OPD works well when most training data contains correct student prefixes but can reinforce errors if the training regime concentrates on failures.

By Shuo Yang, Zheyu Zhang, Mahito Sugiyama, Gjergji Kasneci
arXiv AI
Jul 29

Pass the Baton: Trajectory-Relayed On-Policy Distillation

arXiv:2607. 26057v1 Announce Type: cross Abstract: On-policy distillation (OPD) grounds token-level supervision in the student's own trajectory, yet suffers from prefix failure: once the student commits to a wrong reasoning direction, all subsequent generation builds on this deviation, producing misdirected continuations that elicit unreliable supervision and waste compute.

By Haolei Xu, Xiaowen Xu, Haiwen Hong, Zixuan Ni, Hongxing Li, Yiwen Qiu, Weiming Lu, Yongliang Shen