Abstract
Direct On-Policy Distillation transfers reinforcement learning improvements from smaller to larger models by using the policy shift induced by RL as an implicit reward signal, enabling efficient scaling of training without re-running expensive RL on the target model.
Reinforcement learning with verifiable rewards (RLVR) is a powerful recipe for improving language-model reasoning, but it is expensive to repeat on every new strong model because the target model must generate many rollouts during training. As models scale, post-training itself becomes a bottleneck. We study a weak-to-strong alternative: run RL on a smaller model where rollouts are cheaper, then reuse what that RL run learned to improve a stronger target model. Directly distilling the post-RL weak teacher is not enough, because the teacher's final policy mixes useful RL gains with the limitations of the smaller model. We propose Direct On-Policy Distillation (Direct-OPD), which transfers the teacher's RL-induced policy shift instead. Direct-OPD compares the post-RL teacher with its own pre-RL reference and treats their log-ratio as a dense implicit reward for the student. In plain terms, the checkpoint pair tells us which actions RL made the weak model more or less likely to take, and Direct-OPD applies that signal on the stronger student's own on-policy states. This directly reuses the weak model's RL supervision signal without running sparse-reward RL on the target model. Empirically, Direct-OPD consistently leverages weaker teachers to improve stronger target models; notably, it boosts Qwen3-1.7B from 48.3% to 58.3% on AIME 2024 in just 4 hours on 8 A100 GPUs. It outperforms step-matched direct RL and enables the sequential composition of multiple policy shifts. Our results show that RL outcomes can be reused across model scales as implicit reward signals, not merely as final models to imitate.
Community
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.
Get this paper in your agent:
hf papers read 2607.05394 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper
Collections including this paper 0
No Collection including this paper