Weak-to-Strong Generalization via Direct On-Policy Distillation
Summary
Direct-OPD distills the policy shift from a small model's pre- and post-RL checkpoints to improve a larger student model via on-policy distillation, achieving significant gains without expensive RL on the student.
View Cached Full Text
Cached at: 07/14/26, 04:12 AM
Paper page - Weak-to-Strong Generalization via Direct On-Policy Distillation
Source: https://huggingface.co/papers/2607.05394 RLVR is powerful, but repeating it for every larger target model is expensive: each target must generate its own rollouts and rediscover useful learning signals from sparse outcome rewards.
Can RL on a small, weaker model improve a stronger student—even when the student already outperforms the small model after RL?
We found that it can—but not by distilling the weak model itself.
Today, we’re excited to share Direct-OPD, developed at SIA-Lab, a joint lab of Tsinghua AIR and ByteDance Seed. https://bytedtsinghua-sia.github.io/Direct-OPD/
Our alternative is simple:
- Run RL on a small model, where exploration and rollouts are cheaper.
- Treat the model’s pre- and post-RL checkpoints as a teacher pair, whose difference captures the direction learned through RL.
- On-policy distill this policy shift—what RL changed—using the stronger student’s own rollouts.
In one setting on AIME24: • 1.5B teacher pair: pre-RL and post-RL checkpoints (Post-RL teacher score: 51.3) • 7B student before transfer: 56.7 • 7B student + vanilla on-policy distillation: ~50 • 7B student + Direct-OPD: 63.1 (+6.4)
The 7B student already starts stronger than the post-RL teacher. Distilling the teacher itself makes the student worse. But distilling what the teacher pair learned through RL improves it further.
In other words, the reusable outcome of an RL run is not only the final checkpoint—it can also be the policy shift encoded by the pre- and post-RL checkpoint pair.
Similar Articles
Eliciting Weak-to-Strong Generalization with On-Policy Reverse Distillation
The paper introduces On-Policy Reverse Distillation (OPRD), a method that enables stronger AI models to exceed weaker supervisors by amplifying verifier-supported policy gradients along the teacher's shift direction, achieving higher performance with fewer updates in distillation scenarios.
Gains and Collapse in On-Policy Distillation:A Reinforcement Learning Perspective
This paper analyzes on-policy distillation (OPD) for language model post-training through a reinforcement learning lens, explaining why OPD both improves performance and can collapse into repetitive generation. It finds that masking unhealthy responses during training and SFT initialization can mitigate collapse, shifting focus from teacher generation quality to its ability to evaluate student rollouts.
On-Policy Distillation Meets Off-Policy GRPO: Training Compact Instruction-Following Rerankers
This paper introduces a reinforcement learning-based distillation framework for training compact instruction-following rerankers, using off-policy GRPO for teacher enhancement and on-policy distillation for student learning, demonstrating superior performance under distribution shift.
The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes
This paper presents a comprehensive empirical study on on-policy distillation for large language models, identifying failure mechanisms like distribution mismatch and optimization instability, and proposing fixes such as stop-gradient objectives and RLVR-adapted teachers.
OPRD: On-Policy Representation Distillation
OPRD proposes a new knowledge distillation method that aligns student and teacher hidden states across layers during on-policy rollouts, eliminating sampling variance from token-space KL estimation. Empirically, OPRD outperforms output-space baselines on math reasoning benchmarks (AIME 2024/2025, AIMO) while being 1.44x faster and using 54% less memory.