When Do We Need On-Policy Distillation? Distilling on Offline Student Rollouts Is Often Better (1 minute read)

TLDR AI Papers

Summary

NVIDIA的研究提出Semi-OPD替代方案,发现在初始学生的离线rollout上进行蒸馏通常优于on-policy蒸馏,在17组教师-学生配对中14次胜出,最高提升13.6%准确率和11.4倍训练加速。

On-policy distillation has become increasingly popular for transferring teacher capabilities to student models. Semi-OPD, which distills from offline rollouts generated by the initial student, can often outperform OPD in both accuracy and training efficiency. The choice between OPD and Semi-OPD depends on the alignment between the initial teacher and student. OPD is beneficial only when the two are highly aligned with high overlap ratios.
Original Article
View Cached Full Text

Cached at: 10/13/26, 06:29 PM

# When Do We Need On-Policy Distillation? Distilling on Offline Student Rollouts Is Often Better | Research Source: [https://research.nvidia.com/publication/2026-12_when-do-we-need-policy-distillation-distilling-offline-student-rollouts-often](https://research.nvidia.com/publication/2026-12_when-do-we-need-policy-distillation-distilling-offline-student-rollouts-often) 1. [Publications](https://research.nvidia.com/publications) 2. When Do We Need On\-Policy Distillation? Distilling on Offline Student Rollouts Is Often Better ![Publication image](https://research.nvidia.com/sites/default/files/styles/wide/public/default_images/default.jpeg?itok=TfIobf92) 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\. ## Authors Siyan Zhao \(NVIDIA\) Shih\-Yang Liu \(NVIDIA\) Sharath Turuvekere Sreenivas \(NVIDIA\) Wenliang Dai \(NVIDIA\) Aditya Grover \(NVIDIA\) ## Publication Date Sunday, December 6, 2026 ## Published in ## Research Area

Similar Articles

Demystifying On-Policy Distillation: Where Does the Teacher's Advantage Live?

arXiv cs.LG

这篇论文从知识蒸馏视角重新审视on-policy distillation(OPD),通过受控token翻转实验测量教师在学生生成轨迹上的优势来源,发现基于mass和KL的选择性干预比随机翻转高出15-18个百分点,主张在保持分布稳定性的同时学习更有价值的教师纠正。

OPMD: On-Policy Mechanism Distillation Beyond Output Supervision

Hugging Face Daily Papers

The paper introduces On-Policy Mechanism Distillation (OPMD), which transfers teacher intermediate computations compatible with the student's architecture during on-policy distillation, outperforming output-only distillation baselines across multiple teacher-student settings.