← 返回 2026-10-08

基于负策略采样序列的在线策略蒸馏 On-Policy Distillation with Negative-Policy Rollouts

Jaehui Hwang, Dongyoon Han, Sangdoo Yun, Byeongho Heo 📅 2026-10-06 👍 19 2026-11-11 18:30
LLM后训练 在线策略蒸馏 推理增强 知识蒸馏 负样本学习

在rollout阶段混入更弱负策略的序列,不改蒸馏奖励即为OPD补上负向信号并全面提升

前置知识

在线策略蒸馏(OPD)

OPD是一种LLM后训练方法:让学生模型 $\pi_\theta$ 对每个问题 $x$ 自己生成序列(rollout),教师在相同前缀 $(x, y_{<t})$ 下逐token给出监督,通过RL式更新把学生拉向教师分布。与在教师生成序列上做SFT的SeqKD不同,OPD在学生自己的分布上训练,减少训练-推理失配、保留学生既有能力,近年被证明可以替代RLVR做推理后训练。

本文的全部改动都建立在OPD框架之上——只有先理解rollout与token级监督这两个要素,才能看懂负策略是从哪个环节被注入的。

token级蒸馏奖励

OPD把蒸馏写成RL形式的逐token奖励 $R^{OPD}_t=\log\pi^*(y_t|x,y_{<t})-\log\pi_\theta(y_t|x,y_{<t})$,即教师与学生对该token对数概率之差:学生采到教师也喜欢的token得正奖励,反之得负奖励,整体梯度以该奖励加权策略梯度。ExOPD、OPD2等近期变体都是在改造这个 $R_t$。

NP-OPD的核心证明是:负策略token的期望 $R^{OPD}_t<0$,因此不动奖励、只换采样来源就能产生'推离'效果;不看懂这个量就无法理解3.2节的推导。

Rollout与rollout策略

Rollout指模型对一个问题自回归采样出的完整回答序列;rollout策略即产生这些序列的模型。OPD中rollout策略就是学生自己(on-policy);本文把它推广为混合分布:以概率 $1-\alpha$ 用学生、概率 $\alpha$ 用负策略 $\pi_n$ 采样。rollout来源决定了训练序列的分布,进而决定哪些token会被教师打分。

这正是本文唯一改动的地方——'负信号从rollout阶段进入'是全文区别于以往奖励层面改进的根本点。

DPO与负样本

直接偏好优化(DPO)利用成对回答 $(y^+, y^-)$,通过 $-\mathbb{E}\log\sigma\big(\beta[\log\tfrac{\pi_\theta(y^+)}{\pi_{ref}(y^+)}-\log\tfrac{\pi_\theta(y^-)}{\pi_{ref}(y^-)}]\big)$ 同时学习'靠近preferred'与'远离rejected'两个方向,其中被拒绝回答常由更弱模型采样得到。负样本提供了显式的'该避免什么'信号。

NP-OPD的负策略概念直接类比DPO中的rejected回答,理解这一对应关系才能明白'负'的确切含义与设计动机。

KL散度

KL散度 $D_{KL}(p\|q)=\mathbb{E}_{y\sim p}[\log p(y)-\log q(y)]$ 衡量两个分布的差异且方向不对称。本文用它刻画三个模型的两两距离:负策略离教师更远、离学生更近,即 $D_{KL}(\pi_n\|\pi^*)>D_{KL}(\pi_n\|\pi_\theta)$,这一假设保证了负策略token的期望蒸馏奖励为负。

3.2节的核心论证(式12)本质上就是一次KL散度的代数变形,是'为什么只换采样就能产生负信号'的数学根据。

GRPO组相对优势

GRPO对每个问题采样一组回答,用组内奖励归一化得到优势 $A_i$,正优势回答被强化、负优势被抑制,天然同时包含正负两种信号,是基于可验证奖励的推理RL的主流算法。

论文用GRPO做结构对比:RL类方法天然拥有负信号,OPD没有——这正是NP-OPD要填补的空缺,也是理解论文动机的关键参照。

研究动机

在线策略蒸馏(OPD)让学生在自己生成的序列上接受更强教师的逐token监督,已成为替代RL后训练的热门方案,但现有OPD的蒸馏目标始终只有一个方向——'模仿教师':教师是唯一的正向参照,训练信号从不显式告诉学生'应该远离什么'。当教师与学生能力差距过大、分布重叠有限时,这种纯正向引导的监督会变得稀薄——在学生与教师分布低重叠的区域,教师信号难以提供足够的学习梯度,先前研究已多次观察到这一现象。例如用Qwen3-30B级教师蒸馏1.7B学生时,大量token位置上教师与学生的偏好几乎不相交,蒸馏信号利用不足。相比之下,DPO通过成对的preferred/rejected回答同时提供'靠近'与'远离'两个方向,GRPO也通过正负优势同时强化正确回答、抑制错误回答,而OPD在结构上缺失了整个负半边。

本文的目标是本文的目标是在完全不修改蒸馏奖励公式的前提下,为OPD补上缺失的负向学习信号。具体包括四点:其一,引入一个能力低于学生的'负策略'模型,把它的生成序列混入训练批次并同样施加教师监督,使其扮演DPO中rejected回答的角色,为学生提供明确的'远离方向';其二,保留OPD原有的教师正向监督与token级奖励,保证方法与现有OPD变体(ExOPD、OPD2等)即插即用地兼容;其三,通过NSP、NOR等可测量的诊断指标验证负信号确实按预期作用——即抑制'负策略比教师更偏好'的token、拉大学生与负策略的分布距离;其四,利用负rollout可离线预生成的特性,顺带降低训练的rollout生成成本。

与已有工作不同的是,现有OPD改进几乎全部集中在修改token级奖励:ExOPD引入参考模型(教师的基座模型)做奖励外推,OPD2利用教师与其预训练基座之间的概率差作为delta信号,另有不确定性加权等方案;另一条线虽修改训练轨迹但目的仍是提升监督质量或效率,如Lightning OPD离线预计算教师信号、Relay-OPD让教师接管失败的学生前缀。这些工作都没有把'采样来源'当作传递负信号的通道。本文的独特切入在于:负信号不必通过奖励设计注入,而可以通过更换rollout策略注入——把一部分序列换成弱模型的生成,让OPD奖励在统计上自然对这些序列给出负期望回报,从而把学生推离负策略。这重新定义了rollout策略在OPD中的角色,且与所有奖励层面的改进正交、可自由叠加。

核心方法

直觉上,OPD像只告诉学生'向老师看齐'却从不说'别学坏榜样';NP-OPD给训练加了一个'反面教材':从同家族容量更小的弱模型(如Qwen3-0.6B之于1.7B学生)采样一部分序列,让教师在同一条序列上逐token打分。由于弱模型的token在教师看来普遍较差,OPD奖励对这些序列的期望值为负,学生便在统计上被推离负策略的生成模式,而教师偏好的token仍会被增强,正向监督不丢失。技术路线上方法极简:保持token级奖励 $R^{OPD}_t=\log\pi^*(y_t|x,y_{<t})-\log\pi_\theta(y_t|x,y_{<t})$ 完全不变,仅把式(1)中学生自己采样改为混合rollout分布 $\pi_{z_x}$:以概率 $1-\alpha$ 从学生采样、以 $z_x\sim\mathrm{Bernoulli}(\alpha)$ 决定的概率从负策略 $\pi_n$ 采样,再沿用RL式策略梯度更新 $\nabla_\theta J^{NP}(\theta)$。$\alpha=0$ 退化为标准OPD,$\alpha=1$ 完全使用负策略序列。全部改动只发生在'序列从哪来'这一处。

核心创新是把负信号的注入点从'奖励'挪到'rollout分布'。作者通过期望梯度分解证明这一改动的效果:$g^{NP}-g^{OPD}=\alpha\big(\prod_{i\le t}\pi_n(y_i|x,y_{D_{KL}(\pi_n\|\pi_\theta)$(负策略离教师比离学生更远,选更弱模型即自然满足),则有 $\mathbb{E}_{y_t\sim\pi_n}R^{OPD}_t=-D_{KL}(\pi_n\|\pi^*)+D_{KL}(\pi_n\|\pi_\theta)<0$,负策略token平均收到负奖励、被抑制。关键在于这是'选择性抑制':token级奖励仍是OPD奖励,凡教师偏好的token即使负策略也偏好照样被增强,因此不破坏OPD原有训练动力学;且负token期望奖励为负的论证对ExOPD、OPD2等新奖励同样近似成立,使方法无需改动任何奖励函数即可叠加。

方法步骤详情

第一步,选定负策略:取与同学族、容量更小、推理更弱的模型——Qwen3-1.7B/4B/8B学生分别配Qwen3-0.6B/1.7B/4B作为负策略,Gemma-4-E4B学生配Gemma-4-E2B;教师按规模与模式选择:思考模式下1.7B学生用Qwen3-8B教师、4B/8B用Qwen3-30B-A3B-Thinking,非思考模式用对应Instruct版。第二步,构造混合批次:每条问题按 $z_x\sim\mathrm{Bernoulli}(\alpha)$ 决定序列由学生(比例 $1-\alpha$)或负策略(比例 $\alpha$)以温度1.0生成,rollout上限16k token,训练集为Math/Science/Code三来源各三分之一共3万题。第三步,对两种来源的序列统一计算蒸馏奖励(OPD、ExOPD或OPD2公式)并施加RL式更新;优化用AdamW、学习率 $5\times10^{-6}$、余弦衰减、warmup 0.1、梯度裁剪1.0、bf16+ZeRO-3,全局batch 256,训练100步。第四步,机制验证:用NSP(式13)统计被抑制token中'负策略比教师更偏好'的比例,用NOR(式14)度量学生在教师与负策略首token分歧点上向教师侧偏移的程度。第五步,效率优化:负策略rollout可离线预生成并全程复用,$\alpha=1$ 时学生侧无需rollout生成。

技术新颖性

技术新颖性体现在三点。其一,注入位置新:首次系统地把rollout来源当作负信号的传递通道,与所有奖励层面的OPD改进(ExOPD、OPD2、不确定性加权等)正交,实验证明叠加后仍有一致增益(如Qwen3-4B思考模式OPD2上再+2.10点),而既有轨迹修改类工作(Lightning OPD、Relay-OPD)从未把rollout用作负参照。其二,机制刻画新:给出式(10)-(12)的梯度分解与KL论证,说明'只换采样、不改奖励'即可产生推离负策略的期望更新,并定义NSP、NOR两个可复用的诊断指标,把'负信号是否生效'从玄学变成可测量量。其三,视角新:作者发现ExOPD与OPD2在on-policy下本身就有更高的NSP,提示近期OPD变体的成功可能部分源于经由教师基座模型引入的隐式负信号,为理解OPD文献提供了统一解释;同时负rollout可预生成复用带来最高2.61×(思考)/4.14×(非思考)的训练加速,兼具新颖性与工程实用性。

实验结果

主结果覆盖3个Qwen3学生、思考/非思考两模式、13个基准(数学7、代码3、科学3)。非思考模式(Table 1,$\alpha=1$):Qwen3-1.7B数学均分48.3→56.3、代码27.2→31.1、科学37.2→41.5;Qwen3-4B数学62.2→70.5(AIME24单点53.8→67.9,HMMT25 26.2→43.8)、代码40.2→51.1、科学47.8→51.2;Qwen3-8B数学65.1→69.4。思考模式(Table 2):NP-OPD对三种奖励公式全部有效,1.7B总体均分+1.98(OPD)/+0.30(ExOPD)/+1.76(OPD2),4B分别+1.11/+1.86/+2.10;8B上OPD2均分75.0→77.1(AIME24 73.8→81.7)。跨家族验证(Table 3):Gemma-4-E4B数学63.0→66.4、科学49.1→49.8,但代码54.9→55.2虽略升仍低于未训练基座的56.5。效率上,负rollout复用使1.7B思考模式训练时间9h19m→3h34m(2.61×)、非思考5h15m→1h16m(4.14×)。机制分析(Figure 2-4):NP-OPD的NSP显著高于OPD且与平均准确率强相关;换成1.7B/4B/8B(含教师)做rollout时准确率反而下降,仅0.6B负策略有效;NOR均值NP-OPD 0.019对OPD 0.005,教师rollout为−0.030,held-out提示上0.016对−0.000,方向一致。消融(Table 4):高温1.5、小学生persona、去3层等'自劣化'rollout仅带来+0.9/+1.0/−0.8(均分42.9→43.8/43.9/42.1),远小于独立负策略的+3.4(46.3)。模型合并 $\alpha=0\oplus1$ 在8B ExOPD上数学74.2→76.3、OPD2 75.0→77.4;生成长度方面NP-OPD使4B/8B变短、1.7B略增。

Results of Qwen3 models on math, code, and science benchmarks (non-thinking mode).
Table 1: Results of Qwen3 models on math, code, and science benchmarks (non-thinking mode).
Effect of NP-OPD across OPD variants (thinking mode).
Table 2: Effect of NP-OPD across OPD variants (thinking mode).
Results of Gemma-4-E4B models on math, code, and science benchmarks (thinking mode).
Table 3: Results of Gemma-4-E4B models on math, code, and science benchmarks (thinking mode).
Negative-policy suppression precision (NSP) for OPD and different rollout policies.
Figure 2: Negative-policy suppression precision (NSP) for OPD and different rollout policies.
Role of lower-capability rollouts.
Figure 3: Role of lower-capability rollouts.
查看结构化数据
任务指标本文基线提升
数学推理(Qwen3-1.7B,非思考,α=1) 7个数学基准平均分 56.3 48.3(OPD) +8.0
数学推理(Qwen3-4B,非思考,α=1) 7个数学基准平均分 70.5 62.2(OPD) +8.3
AIME24(Qwen3-4B,非思考,α=1) avg@16 准确率 67.9 53.8(OPD) +14.1
代码推理(Qwen3-4B,非思考,α=1) 3个代码基准平均分 51.1 40.2(OPD) +10.9
13基准总体(Qwen3-1.7B,思考,OPD奖励) All平均分 51.6 49.7(OPD) +1.98
13基准总体(Qwen3-4B,思考,OPD2奖励) All平均分 67.5 65.4(OPD2) +2.10
数学推理(Gemma-4-E4B,思考,α=1) 7个数学基准平均分 66.4 63.0(OPD) +3.4
训练耗时(Qwen3-1.7B,思考模式,α=1) 单节点8×H100墙钟时间 3h34m 9h19m(OPD) 2.61×加速

局限与改进

作者承认的局限:一是并非所有设置都提升——例如Qwen3-8B非思考α=1时代码均分50.1→48.9、GPQA 60.4→56.8出现下降;Gemma代码两种方法均低于未训练基座(56.5),说明负信号在部分域会挤出正向监督。二是α需要按设置挑选,论文报告的是'在三个域均不差于基线的候选中选总体最高'的代表α,实际使用存在调参成本,且Math/Science偏好α=1而Code偏好中间值。三是思考模式的增益(约+1~2点)明显小于非思考模式(+4~8点)。我自己的观察:其一,负rollout预生成后全程静态复用,随学生进步,负策略的'负性'与学生-负策略重叠度会漂移,α=1时训练本质上是离线的,可能限制更长日程的训练;其二,负策略选择停留在'同学族小一号'的启发式,未探索跨家族弱模型、学生历史checkpoint或EMA学生等更灵活的负参考;其三,核心假设 $D_{KL}(\pi_n\|\pi^*)>D_{KL}(\pi_n\|\pi_\theta)$ 在学生快速变强后可能反转,届时负token期望奖励转正、机制失效,但论文未监测训练中假设的成立情况;其四,主实验未与RLVR(如GRPO)在相同数据与算力下对比,NP-OPD相对强化学习的性价比位置仍不清楚;其五,NSP/NOR机制分析集中在1.7B学生上,更大模型的机制行为未验证,且各设置仅单次训练,方差信息有限。

独立分析的弱点

独立分析的弱点与改进方向:(1) 静态负rollout的陈旧化——预生成序列反映训练初期负策略与学生状态,训练后期分布失配,α=1时学生完全不在自己的分布上训练;改进方向是周期性重采样负rollout,或用学生早期checkpoint/EMA权重在线充当'会成长的负策略'。(2) 域间增益不均——代码域多处下降(8B非思考α=1代码50.1→48.9,Gemma代码低于基座),可能与负策略代码能力过弱、序列噪声过大有关;改进方向是按域设置不同α,或以在线NSP为门控信号动态调节负样本比例。(3) 假设不可见则不可保障——建议训练中持续估计 $D_{KL}(\pi_n\|\pi_\theta)$,当其接近 $D_{KL}(\pi_n\|\pi^*)$ 时自动下调α或更换更弱负策略,形成负参考难度的课程化调度。(4) 缺RL对照——应在同数据同算力下与GRPO等RLVR比较,并测试NP-OPD与RLVR混合的互补性。(5) 实验稳健性——每个配置仅报告单次训练结果,α敏感性结论(大α常更好)在不同学生/模式间并不完全一致,需多seed与更多α网格确认。(6) 负策略仅有'更弱'这一个定义维度,'负'与'错'(如仅在某能力维度弱)之间的关系值得更细致解耦。

未来方向

作者提出的方向:开源代码与预生成负rollout;以'隐式负信号'视角统一解释ExOPD、OPD2等近期OPD变体并指导新奖励设计。基于本文成果可延伸的方向:(1) 动态/可学习负策略——用EMA学生、早期checkpoint或多负策略池(不同弱度)按课程难度调度,把负参考强度变成训练进程的函数;(2) 与RLVR混合——在可验证奖励的RL中混入负策略rollout,检验奖励稀疏场景下负信号的互补价值;(3) 把NSP/NOR从诊断指标升级为训练监控、早停信号乃至直接优化的正则项;(4) 扩展到推理之外的后训练目标,如安全对齐(负策略=越狱倾向模型)与指令跟随;(5) 与Relay-OPD等轨迹修复方法结合,研究'教师修复后的负样本'是否仍保留负信号;(6) 理论上刻画rollout分布形状与蒸馏梯度的关系,例如推导最优混合比例 $\alpha$ 的闭式解或随训练动态调整的调度律;(7) 探索负策略与学生能力差距的最优区间——0.6B对1.7B有效,但差距过大(如0.5B对70B)是否仍有效尚属空白。

复现评估

复现条件较好。代码承诺开源于 github.com/naver-ai/np-opd,并计划一并发布预生成的负策略rollout;所有模型均为公开权重(学生Qwen3-1.7B/4B/8B、教师Qwen3-8B与Qwen3-30B-A3B-Thinking/Instruct、负策略Qwen3-0.6B/1.7B/4B,以及Gemma-4-E4B/E2B/12B);训练数据为3万个公开题目(Math/Science/Code三来源1:1:1混合),无私有数据。算力门槛中等:Qwen实验为单节点8×H100(vLLM并发采样,8B用张量并行TP=4),Gemma用两节点;训练仅100步、全局batch 256、学习率 $5\times10^{-6}$、bf16+ZeRO-3。附录C.1完整给出优化器、warmup、梯度裁剪、KL正则系数(Qwen为0、Gemma为0.04)与评测协议(Evalchemy与lm-evaluation-harness,AIME24/25取avg@16、AMC23/HMMT25/MATH500/GPQA取avg@8、其余avg@3)。主要难点是多奖励变体×多α的完整消融矩阵算力开销大(附录跑了30余组训练);若只复现主结论(单一学生、标准OPD奖励、α=1),一到两个8×H100节点日内可完成;NSP/NOR机制分析只需推理即可复现,额外成本很低。

相似论文推荐