Weak-to-Strong Generalization via Direct On-Policy Distillation
arXiv:2607.05394v2 · This workspace stays on this version. Check arXiv for newer versions
Text from the PDF. The web (HTML) version of this paper couldn’t be retrieved, so this is the PDF’s text without its layout: formulas, tables and figures are not shown, and nothing was compared or corrected.
Source details
The text was extracted from the original PDF as is; it was not compared with another source or corrected.
- The HTML source is unavailable (SOURCE_NETWORK_OR_TIMEOUT).
Figures and tables are only in the original PDF. Open original PDF
Weak-to-Strong Generalization via Direct On-Policy Distillation
Shiyuan Feng*1,2,4, Huan-ang Gao*, ‡1,2,3, Haohan Chi*1,2,3, Hanlin Wu1,2, Zhilong Zhang1,2, Zheng Jiang3, Bingxiang He3, Wei-Ying Ma1,2, Ya-Qin Zhang1,2, Hao Zhou†1,2
1SIA-Lab of Tsinghua AIR and ByteDance Seed
2Institute for AI Industry Research (AIR), Tsinghua University
3Department of Computer Science and Technology, Tsinghua University
4Peking University
Abstract
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.
Date: July 9, 2026 Correspondence: zhouhao@air.tsinghua.edu.cn Project Page: https://bytedtsinghua-sia.github.io/Direct-OPD/
1 Introduction
Reinforcement learning with verifiable rewards (RLVR) has become a dominant post-training recipe for eliciting strong reasoning in large language models [ 1, 2]. Its cost, however, is tied to the model being trained: every update requires the current policy to generate rollouts, receive verifiable outcome scores, and update on those trajectories. Larger target models make each rollout slower and each RL iteration more expensive. Thus, as reasoning models continue to scale, re-running RLVR from scratch for every new strong model risks making post-training itself a bottleneck.
In this paper, we propose Direct On-Policy Distillation (Direct-OPD), a weak-to-strong post-training paradigm.
*Equal contribution. ‡Project Lead. †Corresponding author.
1
0 100 200 300 Training step
50
54
58
62
AIME 24 ave@32
Direct-OPD
OPD
Initial
Teacher
Teacher ref 28.5
(a) Direct-OPD vs. OPD
Qwen3 1.7B
Qwen3 4B
R1-Distill 7B
40
50
60
70
80
AIME 24 ave@32
+10.0
+5.1
+6.4
Teacher
Teacher ref 28.5
(b) Weak-to-Strong Generalization
Initial Direct-OPD
Figure 1 Direct-OPD transfers the effect of small-model RL rather than imitating the small model. (a) Starting from R1-Distill-7B, vanilla OPD toward the post-RL JustRL-1.5B teacher degrades performance, whereas Direct-OPD transfers the JustRL-1.5B − R1-Distill-1.5B policy shift and improves the student. (b) The same policy shift improves Qwen3-1.7B, Qwen3-4B, and R1-Distill-7B on AIME 2024, including students whose initial accuracy already exceeds the post-RL teacher.
The setting is simple: run RL where it is cheap, on a weak model, and use what that RL run learned to train a stronger target model. Crucially, the weak model is not assumed to be a better reasoner than the student; it is merely an inexpensive vehicle which ships the behavior the RL signal has already reshaped. Direct-OPD transfers the improvement induced by weak-model RL, not the weak model itself.
The central question is what exactly to transfer from the weak RL run. One candidate is the post-RL teacher’s final policy: on-policy distillation (OPD) lets the student sample its own states and trains it to match the teacher there. Yet this is the wrong object for weak-to-strong transfer, because the final policy entangles the useful change induced by RL with the weak model’s intrinsic capacity limits. When the student already surpasses the teacher, imitating that policy can overwrite stronger behavior. Figure 1(a) illustrates this failure: R1-Distill-7B [ 1] begins at 56.7, already above the post-RL JustRL-1.5B teacher [2] at 51.3, yet vanilla OPD drags it down to roughly 50. The weak RL run carries useful supervision, but its final policy is a poor carrier of that supervision.
Direct-OPD instead isolates the supervision by contrasting the weak model with itself, before and after RL. Let πTref denote the weak model before RL and πT its post-RL counterpart. We define the teacher policy shift
∆T (y | x) = log πT (y | x) − log πTref (y | x), (1)
which is positive on responses that RL made more likely and negative on those it suppressed. The subtraction discards what the weak model already preferred before RL and retains only what RL changed. More importantly, under the KL-regularized RL objective, this policy shift is mathematically equivalent to the reward that trained the weak model: optimizing against ∆T recovers the same objective form as optimizing against that reward, as proved in Section 2.2. A pair of weak-model checkpoints therefore stores the RL supervision signal directly in policy space. Direct-OPD applies this signal on the student’s own on-policy states, anchored to the student initialization by a KL term, without training an explicit reward model or running sparse-reward RL on the target.
This mechanism converts weak-model RL into a transferable post-training signal. Figure 1(b) shows that the policy shift from R1-Distill-1.5B to JustRL-1.5B improves Qwen3-1.7B, Qwen3-4B [3], and R1-Distill-7B alike, including students that already start above the post-RL teacher. The transfer is also inexpensive: with 8 A100 GPUs for about 4 hours, Direct-OPD raises Qwen3-1.7B from 48.3 to 58.3 on AIME 2024, comparable to what Polaris attains by running RL directly on Qwen3-1.7B with 32 A100 GPUs for at least a week [4].
Contributions.
2
• A weak-to-strong post-training paradigm. We recast small-model RL as a cheap generator of an implicit reward for stronger models: rather than copying a weak teacher’s final policy, we evaluate its RL-induced policy shift on the student’s own on-policy states.
• Consistent gains across teacher pairs and students. With two teacher pairs (R1-Distill-1.5B → JustRL-1.5B and Nemotron-1.5B → QuestA-Nemotron-1.5B) and three students from two families (R1- Distill-7B, Qwen3-1.7B, and Qwen3-4B), Direct-OPD improves every student—including R1-Distill-7B and Qwen3-4B, which already start above the post-RL teacher—and lifts Qwen3-1.7B from 48.3 to 58.3 on AIME 2024 in about 4 hours on 8 A100 GPUs.
• Cheaper than direct large-scale RL. At matched RL steps, running RL on a 1.5B model and transferring with Direct-OPD outperforms running RL directly on R1-Distill-7B in both accuracy and compute.
• Analysis of when the signal is reliable. Unlike imitating the teacher’s final policy, Direct-OPD does not require high teacher–student top-k overlap and transfers across thinking patterns; we identify the response-length and KL conditions under which the implicit reward stays aligned with validation accuracy.
2 Direct On-Policy Distillation
2.1 Preliminaries: On-Policy Distillation
We consider autoregressive language-model policies. Let x ∼ D be a prompt and y = (y1, . . . , yT ) a response, so that a policy π factorizes as log π(y | x) =
PT t=1 log π(yt | st) over prefixes st = (x, y<t). We distinguish four policies: the policy being trained πθ and its initial checkpoint πS ; and a teacher pair consisting of a pre-RL reference πTref and the corresponding post-RL teacher πT . Our weak-to-strong setting is the regime in which πT may be smaller or weaker than the target student, yet the transition from πTref to πT still encodes information worth transferring.
On-policy distillation (OPD) supervises the student on trajectories it samples itself. Given a prompt x, the student draws ˆy ∼ πθ (· | x), and at each visited prefix st = (x, ˆy<t) we read off the student and teacher next-token distributions pt(v) = πθ (v | st) and qt(v) = πT (v | st). Standard OPD minimizes the sequence-level KL to the teacher on these rollouts,
LOPD(θ) = Ex∼D
DKL