叛逆学生:在自蒸馏 RLVR 中反转教师信号以探索推理 Rebellious Student: Reversing Teacher Signals for Reasoning Exploration with Self-Distilled RLVR
反转自蒸馏信号,放大正确 rollout 中模型自驱推理的 token
前置知识
RLVR (Reinforcement Learning with Verifiable Rewards)
可验证奖励强化学习,指用规则可以判定的稀疏标量奖励(如数学题最终答案对错)做 LLM 后训练范式。相比 RLHF,它奖励信号客观、自动可验证,典型代表是 GRPO:每个 prompt 采样 K 条 rollout,用组内相对优势替代 critic 网络。
RLVR 是本文所有实验的基线训练范式,论文在 RLVR 框架内提出 RLRT,理解 GRPO 的 group-relative advantage 和 token-level credit assignment 是读懂 RLRT 改动的必要前提。
Self-Distillation (自蒸馏)
单个模型同时扮演教师和学生:教师视图 $P^T_t(\cdot) = \pi_\theta(\cdot \mid h_t, c)$ 额外条件于特权上下文 $c$(如真实解题过程、同伴的成功 rollout),学生视图 $P^S_t(\cdot) = \pi_\theta(\cdot \mid h_t)$ 没有该上下文;两者之比 $\Delta_t = \log P^T_t - \log P^S_t$ 提供 token 级密集监督。
自蒸馏是 RLRT 的直接前置方法(OPSD、SDPO、SRPO、RLSD 都属此类),RLRT 的核心创新正是反转自蒸馏的对齐方向——把'让学生模仿教师'改成'强化学生和教师不同但仍答对的 token'。
Information Asymmetry (信息不对称)
教师和学生看到的条件信息不同所产生的分布差异,论文用 $\hat{D}_t(v) = \log \frac{P^S_t(v)}{P^T_t(v)}$ 度量 token 级不对称,用 $\bar{D}_t = \mathrm{KL}(P^S_t \| P^T_t)$ 度量位置级不对称;后者通过 Pinsker 型不等式控制位置的影响 $\mathrm{Inf}^S(t)$。
信息不对称是本文的'新设计轴',论文证明它能定位关键分支位置($\bar{D}_t$ 大)并区分探索/利用方向($\hat{D}_t$ 的符号),是理解 RLRT 为什么能成为有价值探索信号的关键数学工具。
GRPO (Group Relative Policy Optimization)
DeepSeek 提出的 RL 算法,采样 K 条 rollout 共享同一 prompt,用组内均值 $\mu$ 和标准差 $\sigma$ 把 reward 标准化为 advantage $A = (R - \mu)/\sigma$,无需 value network;目标函数类似 PPO 的 clipped surrogate $\min(\rho A, \mathrm{clip}(\rho) A)$,其中 $\rho = \pi_\theta/\pi_{\mathrm{old}}$。
RLRT 是 GRPO 的扩展('augments GRPO'),只改动 token 级 advantage 的加权方式(增加 reverse weight $w^{RLRT}_t$),不改变 rollout、reward 计算和 trust-region 机制;不熟悉 GRPO 就无法理解 RLRT 的'轻量改造'性质。
Bayesian Teacher 与位置影响 (Pinsker 界)
论文将教师建模为 $\pi_\theta(\cdot \mid h_t, R=1)$,由此推出引理 1:$P^T_t(v) = P^S_t(v) f(v)/\bar{f}^S_t$($f(v)$ 是 token 正确率),再由 Pinsker 不等式证 $\mathrm{Inf}^S(t) \le \sqrt{2\bar{D}_t}$,说明 $\bar{D}_t \approx 0$ 时该位置对结果无影响。
这是 RLRT 理论正确性的核心——证明 $\bar{D}_t$ 能定位'关键位置',没有这套理论,论文就只是一个经验性算法。
研究动机
在 RLVR 后训练中,GRPO 只提供稀疏的标量奖励,导致 credit-assignment 瓶颈;自蒸馏(SDPO、SRPO、OPSD、RLSD)被提出来缓解该问题,通过让教师(条件于 $c$,如成功同伴 rollout 或真实解题过程)把密集的 token 级监督'灌'给学生。然而,作者观察到一个关键缺陷:在已经正确的 rollout 上,自蒸馏的逻辑反转了。失败时教师指向修正方向是合理的,但成功时把学生拉向教师反而把学生'已经走通的自驱路径'覆盖成教师的路径,造成被近期工作[12]称为'优化歧义'的问题——学生本来独立推理对了,却被强迫模仿教师的解题过程,削弱了它自身的推理能力。此外,RLVR 普遍存在'推理边界坍缩',policy 概率质量集中在少数已有高 reward 策略上,现有解决方案(GRPO+EB 的 token 级熵调控、DIVER 的语义级多样性)都把多样性当作均匀目标,用 n-gram overlap、embedding similarity 等代理指标,只能鼓励表面变化,无法保证探索的 token 真正与正确解相关。
本文的目标是本文提出 RLRT (RLVR with Reversed Teacher),目标是在正确 rollout 上反转自蒸馏的方向——不再把学生拉向教师,而是放大那些学生与教师预测不同但仍答对的 token(称为'自驱推理' token),从而把'信息不对称'从'对齐信号'转化为'有价值的探索信号',在不引入额外多样性启发式的前提下,既保持 GRPO 的稀疏奖励框架,又针对性地强化学生真正'自己做对'的关键决策。
与已有工作不同的是,现有自蒸馏(RLSD)用 $w^{RLSD}_t = (P^T_t/P^S_t)^{\mathrm{sign}(A)}$ 作为 magnitude-only credit,在正确时放大教师偏好的 token;多样性方法把多样性当作均匀目标,不分'有用/无用'。本文切入角度独特之处在于三件事:(1)重新解读 teacher-student gap——在正确 rollout 上,它不是 alignment target 而是'self-driven reasoning locator';(2)严格只在 $r=1$ rollout 上用 $w^{RLRT}_t = (P^S_t/P^T_t)^{\mathrm{sign}(A)}$ 翻转 teacher/student 比例;(3)通过贝叶斯教师框架 + Pinsker 界证明 $\bar{D}_t$ 标记的位置就是真正影响结果的关键分支点,从而把经验算法升级到有理论保证的设计原则。
核心方法
RLRT 的整体思路是'反转自蒸馏方向 + 奖励门控'。直觉上,标准自蒸馏把教师(知道正确答案)当'权威',学生无条件地模仿;但当学生已经答对时,学生偏离教师预测的那些 token 恰恰是它独立推理的证据,这些偏离值得被强化而非被压制。技术上,RLRT 保持 GRPO 的一切基础设施(rollout、reward、trust-region)不变,只增加一个 token 级的 reverse weight $w^{RLRT}_t = \exp(\mathrm{sign}(A)\hat{D}_t) = (P^S_t/P^T_t)^{\mathrm{sign}(A)}$,并仅在 $r=1$ 的 rollout 上用 clip 后的 $(1-\lambda)A + \lambda w^{RLRT}_t$ 替代原始 advantage。训练时同一个模型在两组 prompt 上各采样一次:一组作为学生(无 $c$),另一组附加成功同伴 rollout 作为教师(条件 $c$),两者共享参数 $\pi_\theta$。
RLRT 与 RLSD 的本质区别只有一处:权重比值从 $P^T_t/P^S_t$ 翻转为 $P^S_t/P^T_t$。RLSD 在 $r=1$ 上放大 $P^T_t > P^S_t$ 的 token(教师偏好的、利用方向),RLRT 反过来放大 $P^S_t > P^S_t$ 的 token(学生自主偏离教师的、探索方向)。两个正则化项共同点:都靠 $\mathrm{sign}(A)$ 决定方向,$\Delta_t/\hat{D}_t$ 只调幅度。但 RLRT 严格把该权重门控到 $r=1$ rollout,杜绝在失败轨迹上把'无意义偏离'也当作'自驱探索'来强化——论文消融证明,没有 reward gate 的 RLRT-all 在 step 40 就因 response length 和 entropy 爆炸而训练坍缩。理论配套创新:用贝叶斯教师 $\pi_\theta(\cdot \mid h_t, R=1)$ 把 $\bar{D}_t = \mathrm{KL}(P^S_t \| P^T_t)$ 和位置影响力 $\mathrm{Inf}^S(t)$ 通过 Pinsker 界联系起来,证明 $\bar{D}_t$ 标记的位置就是影响结果的关键分支点。
方法步骤详情
RLRT 的训练步骤为:(1)Student rollouts:对 prompt $x$ 用 $\pi_\theta$ 采样 K 条 $y^{(k)}$,得到 $P^S_t(\cdot) = \pi_\theta(\cdot \mid h_t)$;(2)Teacher rollouts:取正确 $y^{(K)}$ 作为特权上下文 $c$,对同一组 prompt 在条件 $c$ 下再前向一次,得到 $P^T_t(\cdot) = \pi_\theta(\cdot \mid h_t, c)$;(3)计算 token 级不对称 $\hat{D}_t = \log P^S_t(y_t) - \log P^T_t(y_t)$,并据此得到 reverse weight $w^{RLRT}_t = \exp(\mathrm{sign}(A^{(k)})\hat{D}_t) = (P^S_t/P^T_t)^{\mathrm{sign}(A^{(k)})}$;(4)计算组内 advantage $A^{(k)} = (R^{(k)} - \mu)/\sigma$,然后对正确 rollout($r(y^{(k)})=1$)做 reward-gated 替换 $\mathcal{A}^{RLRT,(k)}_t = A^{(k)}(1-\lambda) + \lambda\,\mathrm{clip}(w^{RLRT}_t, 1-\varepsilon_w, 1+\varepsilon_w)$,错误 rollout 保持 $A^{(k)}$ 不变;$\lambda \in [0,1]$ 控制 reverse 信号强度($\lambda=0$ 退化为 GRPO),$\varepsilon_w$ 是 clip 范围防过激权重;(5)用替换后的 $\mathcal{A}^{RLRT,(k)}_t$ 替代原 advantage 喂给 PPO 风格的 clipped surrogate loss,反向更新 $\pi_\theta$。训练超参采用 DAPO 风格:batch size 256、PPO mini-batch 128、max response length 20480、非对称 clip $\varepsilon_{\mathrm{high}}=0.28, \varepsilon_{\mathrm{low}}=0.2$。
技术新颖性
技术新颖性体现在三个层次。第一,概念层:首次把'教师-学生差距'从 alignment target 翻转为 self-driven reasoning locator,提出了'valuable exploration'概念(在已正确 rollout 中既自我偏离又经验正确的 token)代替传统的'均匀多样性'。第二,算法层:RLRT 是首个把 $P^S_t/P^T_t$(而非 $P^T_t/P^S_t$)用作 token 级权重的 RLVR 算法,并配套 reward gate 机制,论文消融表明 reward gate 是不可或缺的——RLRT-all(无 gate)在 step 40 训练坍缩。第三,理论层:贝叶斯教师框架 + Pinsker 不等式给出 $\mathrm{Inf}^S(t)^2 \le 2\bar{D}_t$,严格证明 $\bar{D}_t \approx 0$ 的位置是'惰性'位置,反向用反射注入的因果干预(Fig. 6)在经验上验证了该理论——在 max-$\bar{D}_t$ 位置注入'Wait, let me reconsider.'比 random/min-$\bar{D}_t$ 位置的 wrong→right 翻转率高出 2 倍。
实验结果
在 4 个 Qwen3 backbone(4B-Base、8B-Base、4B-Instruct、8B-Thinking)上训练 100 step,eval 在 AIME24/25/26、HMMT26、AMC23、MATH500 六个数学 benchmark 上 avg@16/pass@16。表 1 显示 RLRT 在所有 4 个 backbone 上稳定超过 GRPO/SDPO/SRPO/RLSD,平均增益为:Qwen3-4B-Base +18.0%、Qwen3-8B-Base +12.0%、Qwen3-4B-Instruct +3.4%、Qwen3-8B(thinking off)+2.2%。具体数字:Qwen3-4B-Base 上 AIME24 从 GRPO 15.0 提升到 22.5(+7.5)、AIME26 从 12.3 提升到 19.8(+7.5)、HMMT26 从 10.0 提升到 15.9(+5.9);Qwen3-8B-Base 上 AIME24 从 19.8 提升到 27.9(+8.1)、AIME25 从 17.5 提升到 18.8、AMC23 从 62.5 提升到 67.3。Q1 答案(对比自蒸馏):RLRT 在 Qwen3-8B(thinking off)上比最佳 baseline(GRPO 70.2)在 AIME24 上还多 0.4,Q2 因果验证(Fig. 6):反射注入在 max-$\bar{D}_t$ 位置的 flip→R 率在 base checkpoint 是 random/min-$\bar{D}_t$ 的 2 倍,RLRT 训练 100 步后把该比值放大到超过 40%,而 GRPO 让 max-$\bar{D}_t$ 的优势坍缩到 random 水平。Q3 分布偏移(Fig. 7):在 hard prompt(ncorrect∈{0,1,2}/8)上,RLRT 让 JS>0.1 的 high-divergence token 比例显著超过 GRPO/RLSD,top-20 overlap 从 ~80% 掉到 ~50%,base prob<10^{-3} 的 token 被提升为 top-1 的频率是 GRPO/RLSD 的 10 倍以上。Q4 探索对比(Fig. 8):在 AIME24/AIME26 的 pass@k 曲线上,RLRT 从 pass@1 到 pass@256 全程超过 GRPO+EB 和 DIVER,DIVER 在大 k 才有微弱领先。消融(Fig. 9a)RLRT-all 训练 score 初始跟 RLRT 一致但在 step 40 因 length/entropy 爆炸坍缩,确认 reward gate 不可或缺;Fig. 9b 显示 ε_w=1.0 训练 score 最高、ε_w=0.2 退化为 GRPO,说明增益来自 reweighting 本身而非 r=1 过滤。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| AIME24 (Qwen3-4B-Base) | avg@16 | 22.5 | GRPO 15.0 / RLSD 13.3 | +7.5 (vs best baseline 15.0) |
| AIME24 (Qwen3-8B-Base) | avg@16 | 27.9 | GRPO 19.8 / RLSD 17.3 | +8.1 |
| AIME25 (Qwen3-4B-Base) | avg@16 | 18.5 | GRPO 14.4 / RLSD 11.2 | +4.1 |
| AIME26 (Qwen3-4B-Base) | avg@16 | 19.8 | GRPO 12.3 / RLSD 9.0 | +7.5 |
| HMMT26 (Qwen3-4B-Base) | avg@16 | 15.9 | GRPO 10.0 / RLSD 6.2 | +5.9 |
| AMC23 (Qwen3-4B-Base) | avg@16 | 63.9 | GRPO 58.3 / RLSD 55.2 | +5.6 |
| MATH500 (Qwen3-4B-Base) | avg@16 | 83.8 | GRPO 80.2 / RLSD 77.9 | +3.6 |
| AIME24 (Qwen3-8B thinking off) | avg@16 | 70.6 | GRPO 70.2 / RLSD 65.4 / SDPO 26.9 / SRPO 15.4 | +0.4 (vs GRPO) |
局限与改进
作者承认的局限:第一,只实验了数学 reasoning 任务,且只用了 DAPO-Math-17k 一个训练集,未验证 code、tool-use、开放式问答等领域;第二,教师上下文 $c$ 限制为'已知成功 rollout',而现实 RLVR 中可能没有这样的 ground-truth privileged context,论文未来工作明确提出要探索'noisier rewards'和'更广的 on-policy distillation(教师分布可来自多源)';第三,paper 指出 RLRT 增益在 Qwen3-8B(thinking-tuned)上只有 2.2%,远小于 4B-Base 的 18.0%,说明'对已被指令调教集中化的 policy 效果有限'。我自己观察的额外限制:(1)SDPO/SRPO 在 base 模型上 early collapse(Appendix F.2),所以 base 模型上的对比实际上只有 GRPO/RLSD,缺少与原始 OPSD 的对位(作者说 OPSD 需要外部 ground-truth solution,故排除);(2)教师 rollout 需要额外一次前向,算力开销 ~2x,论文未明确给出 FLOPs 或 GPU hours;(3)$\lambda$ 和 $\varepsilon_w$ 的最优值依赖 backbone(图 9b 显示 ε_w=1.0 在 4B/8B-Base 都最好,但作者没在 instruct/thinking 模型上 sweep);(4)论文在 Section 2.1 列了一长串特权上下文形式(ground-truth traces、运行时错误、judge 评估、critique 修订、专家 demo、in-context 知识),但只实验了'成功同伴 rollout'一种,其他 $c$ 的形式是否同样有效仍未知。
独立分析的弱点
独立分析的具体弱点与改进方向:(1)Teacher 形式单一——RLRT 只用'successful rollout'做 $c$,但 RLVR 训练初期很多 prompt 没有 successful rollout,论文没讨论冷启动问题,改进方向是用 ground-truth solution、self-consistency 多票等更鲁棒的 $c$;(2)Reward gate 阈值过严——硬截断 $r=1$ 意味着所有 $r=0$ rollout 退化为普通 GRPO,失去了对'差一点就答对'轨迹中的自驱 token 的利用,可以尝试 soft gate(如按 reward confidence 加权)或仅 gate 到 high-$\bar{D}_t$ 的位置;(3)算力翻倍——每个 prompt 要做 student rollout + teacher forward,改进方向是共享 KV cache 或在 teacher rollout 阶段关闭梯度、把 teacher 视作 $\pi_{\mathrm{ref}}$ 冻结;(4)Token-level 权重方差大——$w^{RLRT}_t$ 在不同 token 间可能跨几个数量级,虽然有 clip 仍可能放大局部噪声,可考虑加入 $\bar{D}_t$ 阈值过滤掉低不对称位置;(5)Qwen 单一家族——只在 Qwen3 4B/8B 上验证,改进方向是跨家族(Llama、DeepSeek、GPT-OSS)复现以排除架构特异性;(6)评测 benchmark 偏窄——AIME/HMMT/AMC/MATH500 全是同分布的高中竞赛数学,改进方向是加 BBH、MMLU-Pro、ARC-AGI 评估泛化。
未来方向
作者明确提出的方向:(1)把 RLRT 扩展到 noisier rewards(无 ground-truth 的开放式任务);(2)探索更广的 on-policy distillation——教师分布可来自不同源(更大的模型、不同 checkpoint、外部 verifier);(3)跨领域验证(code、tool-use、long-form QA)。基于成果可延伸的方向:(a)把 information asymmetry 从'self-distillation'推广到'cross-model distillation'——用不同模型做教师学生,看 $\hat{D}_t$ 是否仍能定位有价值的探索;(b)动态调整 $\lambda$——训练初期用小 $\lambda$(以稳定为先),后期逐步增大(以探索为先),类似于 entropy coefficient 的退火;(c)与 process reward model(PRM)结合——$\bar{D}_t$ 标记的关键位置天然适合插 PRM step-level reward;(d)在线选择 $c$——不固定用'successful rollout',而是每 step 在多种 $c$(GT solution、self-revised rollout、judge critique)之间按 $w^{RLRT}_t$ 的方差选择,自动发现最优特权上下文形式;(e)理论扩展——贝叶斯教师框架目前假设 $R \in \{0,1\}$ 的二值 reward,可推广到 dense reward。
复现评估
复现难度评估:数据公开——DAPO-Math-17k 训练集和 AIME24/25/26、HMMT26、AMC23、MATH500 评估集均为公开 benchmark,无外部私有数据;模型公开——所有 4 个 backbone(Qwen3-4B-Base、Qwen3-8B-Base、Qwen3-4B-Instruct、Qwen3-8B)均来自 HuggingFace 开源;算法透明——RLRT 只在 advantage 替换处修改,核心公式 Eq.(6)(7) 给出,无隐藏 trick;超参清楚——附录 G.2 列出 batch size 256、PPO mini-batch 128、max response 20480、非对称 clip 0.28/0.2、$\lambda$、$\varepsilon_w$ 等;算力要求较高——4 个 backbone × 100 step × 256 prompt/step × 20K token max response,且每个 prompt 要做 student + teacher 两次前向,粗估需要 8-16 张 H100/A100 训数天,中小实验室可能难以全部复现;代码情况——论文未明确标注 GitHub 链接,但作者在 Microsoft Research,通常后续会开源;基线可复现性——GRPO/SDPO/SRPO/RLSD 均为公开方法,但 SDPO/SRPO 在 base 模型上 early collapse,意味着完整复现需要做多组训练直到全部 baseline 都跑通,成本进一步上升。整体判断:中高复现难度,核心算法可实现,但完整 4 backbone × 多 baseline × 多 benchmark 的复现需要相当算力。
论文图表
左图(a)展示 Qwen3-4B-Base 上 GRPO、teacher-token upweight、RLRT 三个策略的训练 reward 曲线,RLRT 上升最快且最稳;右图(b)展示 4 个 backbone × 5 个方法(GRPO/SDPO/RLSD/SRPO/RLRT)在 6 个数学 benchmark 上的平均 avg@16 柱状对比,RLRT 在每个 backbone 上都是最高,+18.0%/+12.0%/+3.4%/+2.3% 增益标注清晰。
这张图是论文最核心的'一句话总结'——一张图同时展示训练动力学(RLRT 更快)和最终 benchmark 结果(RLRT 全面领先),读者扫一眼就能抓住论文主张。