Search papers, labs, and topics across Lattice.
This paper introduces Direct On-Policy Distillation (Direct-OPD), a novel approach that enhances strong language models by leveraging the reinforcement learning (RL) outcomes of weaker models, thereby circumventing the costly process of generating extensive rollouts for each new model. By comparing the post-RL policy of a weak teacher with its pre-RL state, Direct-OPD effectively transfers RL-induced policy shifts to the stronger model, using the log-ratio as an implicit reward signal. Empirical results demonstrate that this method significantly improves the performance of the Qwen3-1.7B model on the AIME 2024 benchmark, achieving a notable increase from 48.3% to 62.4% in just four hours of training on eight A100 GPUs.
Weak models can supercharge strong models through efficient policy shift transfers, achieving significant performance gains without the usual rollout costs.
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 training an explicit reward model or 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 62.4% 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.
Tsinghua AI