Know When to Stop, Where to Restart: Accelerating Multi-Turn Agentic On-Policy Distillation
On the map: LLM agents and tool use → Reinforcement learning for LLM agents
Abstract (arXiv)
On-policy distillation (OPD) has become a standard approach for transferring capabilities from large teachers to compact students. Its cost, however, is dominated by autoregressive student rollouts and scales poorly in multi-turn agentic settings. Existing acceleration methods truncate or relocate the supervision signal according to fixed, offline budgets, despite substantial variation in teacher-signal reliability both within and across trajectories. Our empirical analysis on τ2-bench reveals a clear structure in this variation: informative supervision is concentrated in the prefix of each turn, and, most importantly for multi-turn agentic training, the cross-turn loss of teacher endorsement is temporally locked to the student's first erroneous action rather than accumulating gradually over turns. Building on these findings, we propose STRIDE (Stop-and-Restart on-policy Distillation acceleration), which combines two complementary techniques: adaptive early stopping, which terminates a rollout once the cumulative teacher log-probability falls below an out-of-distribution threshold, and a prefix buffer, which caches high-quality prefixes and restarts generation at the weakest correct turn. Together, these mechanisms induce a data-driven curriculum that progressively extends coverage to later turns. On τ2-bench retail, our method matches full-trajectory OPD and exceeds the 30B teacher at a 3.73× speedup, surpasses the baseline itself at 2.34×, and retains a 4.51× speedup under cross-domain multi-teacher training. As a supplementary generalization test beyond the agentic setting, STRIDE outperforms full OPD on AIME 2025 at a 5.10× speedup and on AIME 2024 at a 3.08× speedup; averaged across the two evaluations, both fixed-budget truncation baselines remain below full OPD.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Reinforcement learning for LLM agents, one of 16 directions in LLM agents and tool use. Builds on 22 of its 33 references that are in the map. First posted on arXiv 13 Sep 2026.
Builds on
22 of 33 references.
References · 22
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2025Qwen3 Technical Reportcited by 8,195
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
- 2026DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligencecited by 877
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
- 2025τ2-Bench: Evaluating Conversational Agents in a Dual-Control Environmentcited by 558
- 2015Policy Distillationcited by 414
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 262
- 2026Learning beyond Teacher: Generalized On-Policy Distillation with Reward Extrapolationcited by 157
- 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixescited by 122
- 2026TIP: Token Importance in On-Policy Distillationcited by 90
- 2026Fast and Effective On-policy Distillation from Reasoning Prefixescited by 43
10 more
- 2026TCOD: Exploring Temporal Curriculum in On-Policy Distillation for Multi-turn Autonomous Agentscited by 38
- 2025The First Few Tokens Are All You Need: An Efficient and Effective Unsupervised Prefix Fine-Tuning Method for Reasoning Modelscited by 31
- 2026Lightning OPD: Efficient Post-Training for Large Reasoning Models with Offline On-Policy Distillationcited by 28
- 2026Prune-OPD: Efficient and Reliable On-Policy Distillation for Long-Horizon Reasoningcited by 23
- 2026From Self-Evolving Synthetic Data to Verifiable-Reward RL: Post-Training Multi-turn Interactive Tool-Using Agentscited by 17
- 2026TurnOPD: Making On-Policy Distillation Turn-Aware for Efficient Long-Horizon Agent Trainingcited by 14
- 2026Prefix Teach, Suffix Fade: Local Teachability Collapse in Strong-to-Weak On-Policy Distillationcited by 13
- 2026Your Teacher Can't Help You Here: Combating Supervision Fidelity Decay in On-Policy Distillationcited by 11
- 2026EasyOPD: An Easy-to-use On-Policy Distillation Framework for Large Language Modelscited by 3
- 2026Post-Training Shifts Confidence: A Three-Stage Analysis of How SFT, RL, and OPD Shape CoT Calibrationcited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Multi-Turn On-Policy Distillation with Prefix Replay
- 2026Look Ahead Before You Distill: Future Trajectory Validation of Teacher Guidance for Agentic On-Policy Distillation
- 2026Stochastic Teacher Intervention for Agentic On-Policy Distillation
- 2026DASH-OPD: Discrepancy-Aware Switching with Hysteresis for On-Policy Distillation
- 2026Graph-Conditioned On-Policy Agent Distillation from Off-the-Shelf Teachers
- 2026SAGE-OPD: Selective Agent-Guided Intervention for Multi-Turn On-Policy Distillation
- 2026ATOD: Annealed Turn-Aware On-Policy Distillation for Multi-Turn Agentic Tasks
- 2026On-Policy Distillation with Best-of-N Teacher Rollout Selection