When Do We Need On-Policy Distillation? Distilling on Offline Student Rollouts Is Often Better
cs.CL
Submitted: 2026-10-08
Updated: 2026-10-08
License: http://creativecommons.org/licenses/by/4.0/
The gist: On-policy distillation (OPD) has become increasingly popular for transferring teacher capabilities to student models.
Terminology
Abstract
On-policy distillation (OPD) has become increasingly popular for transferring teacher capabilities to student models. In this work, we ask a critical research question: Is on-policy sampling always beneficial for distilling arbitrary teacher-student pairs? We show that a simple alternative, Semi-OPD, which distills from offline rollouts generated by the initial student, can often outperform OPD in both accuracy and training efficiency. Across 17 teacher-student pairs ranging from 1.5B to 235B parameters, Semi-OPD outperforms OPD in 14 cases, with up to +13.6% accuracy and 11.4x training speedup. We further find that the choice between OPD and Semi-OPD depends on the alignment between the initial teacher and student, quantified by an output-token overlap ratio: OPD is beneficial only when the two are highly aligned with high overlap ratios. Our deeper investigation suggests that effective distillation requires on-policyness w.r.t. both the student and the teacher. For misaligned pairs, student rollouts can become increasingly off-policy w.r.t. the teacher as context length grows, weakening the distillation signal. In contrast, Semi-OPD is often more stable, as it distills on shorter contexts while covering full trajectories and exposing the student to more teacher-preferred tokens. Beyond proposing Semi-OPD as an efficient alternative, our work motivates the community to rethink when to use OPD and to study stronger OPD variants with meaningful teacher-student pairs.
Sources
- DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligence
- Demystifying OPD: Length Inflation and Stabilization Strategies for Large Language Models
- When EOS Tokens Disagree: Understanding Length Inflation in On-Policy Distillation
- Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixes
- Mismatch Matters: On-Policy Distillation Beyond Token Agreement
- Less is More: Early Stopping Rollout for On-Policy Distillation
- Prefix Teach, Suffix Fade: Local Teachability Collapse in Strong-to-Weak On-Policy Distillation
- Pass the Baton: Trajectory-Relayed On-Policy Distillation
- Trajectory-Refined Distillation
- Kimi K3: Open Frontier Intelligence
- Lightning OPD: Efficient Post-Training for Large Reasoning Models with Offline On-Policy Distillation
- Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe
- Qwen3 Technical Report
- Nemotron-Cascade 2: Post-Training LLMs with Cascade RL and Multi-Domain On-Policy Distillation
- JustRL: Scaling a 1.5B LLM with a Simple RL Recipe
- A Survey of On-Policy Distillation for Large Language Models
- Entropy-Aware On-Policy Distillation of Language Models
- Unmasking On-Policy Distillation: Where It Helps, Where It Hurts, and Why
- Rethinking On-Policy Distillation of Large Language Models II: One Training Example
- 1% of Tokens Can Be Enough: On Gradient Estimation in On-Policy Distillation
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering