Are Full Rollouts Necessary for On-Policy Distillation?
Not on the map. Why?
Abstract (arXiv)
On-policy distillation (OPD) provides dense teacher feedback along student-generated rollouts rather than fixed teacher traces and has emerged as a promising post-training paradigm. However, standard OPD typically generates full rollouts during training, which is computationally expensive and may expose the student to unreliable teacher feedback at late rollout positions, especially during early training. We identify the rollout horizon as a key bottleneck in OPD that substantially impacts training efficiency. Unlike Reinforcement Learning with Verifiable Rewards (RLVR), OPD does not require a final answer reward to provide learning signals. Therefore, full rollouts may not always be necessary for OPD. Motivated by this insight, we propose two simple horizon-control strategies: Progressive OPD (POPD), which gradually expands the rollout horizon during training, and Truncated OPD (TOPD), which permanently performs distillation on reliable truncated rollouts. Experiments on mathematical reasoning show that POPD improves the training efficiency of OPD by up to 3×, while TOPD matches OPD performance using only 10% of the rollout horizon, leading to substantial wall-clock and memory reductions. These results demonstrate that controlling the rollout horizon offers a simple and practical path to more efficient OPD.
Key facts. Cited by 14 papers in the map (data through 7 Oct 2026). Builds on 16 of its 20 references that are in the map. First posted on arXiv 29 May 2026.
Builds on
16 of 20 references.
References · 16
- 2017Attention Is All You Needcited by 59,932
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,678
- 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 948
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
- 2016Sequence-Level Knowledge Distillationcited by 761
- 2026GLM-5: from Vibe Coding to Agentic Engineeringcited by 502
- 2025ReTool: Reinforcement Learning for Strategic Tool Use in LLMscited by 377
- 2026Reinforcement Learning via Self-Distillationcited by 285
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 262
- 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixescited by 122
- 2026OpenClaw-RL: Train Any Agent Simply by Talkingcited by 83
- 2026Privileged Information Distillation for Language Modelscited by 79
4 more
- 2025CriticSearch: Fine-Grained Credit Assignment for Search Agents via a Retrospective Criticcited by 29
- 2025Nemotron-Math: Efficient Long-Context Distillation of Mathematical Reasoning from Multi-Mode Supervisioncited by 22
- 2026Learning to Foresee: Unveiling the Unlocking Efficiency of On-Policy Distillationcited by 17
- 2026π-Play: Multi-Agent Self-Play via Privileged Self-Distillation without External Datacited by 7
Built on by
Citing papers per year: 2026: 14.
14 papers.
Sort:
By area:
- 2026TurnOPD: Making On-Policy Distillation Turn-Aware for Efficient Long-Horizon Agent Trainingcited by 14
- 2026UCOB: Learning to Utilize and Evolve Agentic Skills via Credit-Aware On-Policy Bidirectional Self-Distillationcited by 10
- 2026Prefix-Guided On-Policy Distillation: Mining Golden Trajectories from Rolloutscited by 7
- 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillationcited by 5
- 2026Contrastive On-Policy Distillationcited by 3
- 2026Beyond Token-Local Imitation: Reward-Compatible Temporal Credit Assignment for On-Policy Distillationcited by 2
- 2026DASH-OPD: Discrepancy-Aware Switching with Hysteresis for On-Policy Distillationcited by 1
- 2026From Parameters to Behaviors: A Survey of Model Fusion for Large Language Modelscited by 1
- 2026Teacher Should Think Ahead: Adaptive Continuations for Reliable On-Policy Distillationcited by 1
- 20261% of Tokens Can Be Enough: On Gradient Estimation in On-Policy Distillationcited by 1
- 2026Blockwise Policy-Drift Gating for On-Policy Distillationcited by 0
- 2026Rollout Efficiency in Reinforcement Learning for Reasoning Large Language Models: A Taxonomy and Future Directionscited by 0
2 more
- 2026On the Off-Policy Teacher in On-Policy Distillation29 Sep 2026
- 2026UOPD: Uncertainty-Aware Intervention for On-Policy Distillation of Multi-Turn Agents27 Sep 2026
- 2026Rollout Efficiency in Reinforcement Learning for Reasoning Large Language Models: A Taxonomy and Future Directions21 Sep 2026
- 20261% of Tokens Can Be Enough: On Gradient Estimation in On-Policy Distillation21 Sep 2026
- 2026From Parameters to Behaviors: A Survey of Model Fusion for Large Language Models17 Sep 2026
- 2026Beyond Token-Local Imitation: Reward-Compatible Temporal Credit Assignment for On-Policy Distillation15 Sep 2026
- 2026Teacher Should Think Ahead: Adaptive Continuations for Reliable On-Policy Distillation6 Sep 2026
- 2026DASH-OPD: Discrepancy-Aware Switching with Hysteresis for On-Policy Distillation31 Jul 2026
- 2026Contrastive On-Policy Distillation21 Jul 2026
- 2026TurnOPD: Making On-Policy Distillation Turn-Aware for Efficient Long-Horizon Agent Training7 Jul 2026
- 2026UCOB: Learning to Utilize and Evolve Agentic Skills via Credit-Aware On-Policy Bidirectional Self-Distillation28 Jun 2026
- 2026Blockwise Policy-Drift Gating for On-Policy Distillation23 Jun 2026
- 2026Prefix-Guided On-Policy Distillation: Mining Golden Trajectories from Rolloutscited by 7
- 2026Teacher Should Think Ahead: Adaptive Continuations for Reliable On-Policy Distillationcited by 1
- 2026Rollout Efficiency in Reinforcement Learning for Reasoning Large Language Models: A Taxonomy and Future Directionscited by 0
- 2026Rollout Efficiency in Reinforcement Learning for Reasoning Large Language Models: A Taxonomy and Future Directions21 Sep 2026
- 2026Teacher Should Think Ahead: Adaptive Continuations for Reliable On-Policy Distillation6 Sep 2026
- 2026Prefix-Guided On-Policy Distillation: Mining Golden Trajectories from Rollouts20 Jun 2026
- 2026TurnOPD: Making On-Policy Distillation Turn-Aware for Efficient Long-Horizon Agent Trainingcited by 14
- 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillationcited by 5
- 2026Contrastive On-Policy Distillationcited by 3
- 2026Beyond Token-Local Imitation: Reward-Compatible Temporal Credit Assignment for On-Policy Distillationcited by 2
- 2026From Parameters to Behaviors: A Survey of Model Fusion for Large Language Modelscited by 1
- 20261% of Tokens Can Be Enough: On Gradient Estimation in On-Policy Distillationcited by 1
- 2026Blockwise Policy-Drift Gating for On-Policy Distillationcited by 0
- 2026UOPD: Uncertainty-Aware Intervention for On-Policy Distillation of Multi-Turn Agentscited by 0
- 2026On the Off-Policy Teacher in On-Policy Distillationcited by 0
- 2026On the Off-Policy Teacher in On-Policy Distillation29 Sep 2026
- 2026UOPD: Uncertainty-Aware Intervention for On-Policy Distillation of Multi-Turn Agents27 Sep 2026
- 20261% of Tokens Can Be Enough: On Gradient Estimation in On-Policy Distillation21 Sep 2026
- 2026From Parameters to Behaviors: A Survey of Model Fusion for Large Language Models17 Sep 2026
- 2026Beyond Token-Local Imitation: Reward-Compatible Temporal Credit Assignment for On-Policy Distillation15 Sep 2026
- 2026Contrastive On-Policy Distillation21 Jul 2026
- 2026TurnOPD: Making On-Policy Distillation Turn-Aware for Efficient Long-Horizon Agent Training7 Jul 2026
- 2026Blockwise Policy-Drift Gating for On-Policy Distillation23 Jun 2026
- 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillation22 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2026On-Policy Distillation with Best-of-N Teacher Rollout Selection
- 2026ADWIN: Adaptive Windows for Horizon-Aware On-Policy Distillation
- 2026Spend Teacher Tokens Where They Matter: Success-Referenced On-Policy Distillation
- 2026On-Policy or Off-Policy Learning? A Systematic Study of Distillation Dynamics
- 2026Learning beyond Teacher: Generalized On-Policy Distillation with Reward Extrapolation
- 2026Gains and Collapse in On-Policy Distillation:A Reinforcement Learning Perspective
- 2026OPD+: Rethinking the Advantage Design for On-Policy Distillation
- 2026Outcome-Guided On-Policy Self-Distillation