Read this paper write-up in English
RSHD:只蒸馏经新重放验证有效的失败复盘¶
L1 算法组件/token 梯度诊断,不是自然语言反思或推理能力复现。 完整实验成本过高,不复现(审查 DONE);不进入 evolve 晋级。
论文信息¶
| 字段 | 内容 |
|---|---|
| 论文链接 | arXiv 2610.21018;固定 v1 全文;PDF |
| 公司 / 机构 | 一作 Sheldon Yu 属 University of California, San Diego,已核对 PDF 首页;合作机构含 Adobe Research、Virginia、Toronto、UNSW |
| 首次公开日期 | 2026-10-14T23:33:22Z(UTC,arXiv v1) |
| 原作者代码 | 未在论文或作者检索结果中找到 RSHD 官方方法仓库 |
| 代码许可 | 未找到方法代码,许可未确认;论文 CC BY-NC-SA 4.0 |
| 本地 adapter / 方法键 | rshd;adapter: null,独立组件 |
| 本地代码位置 | src/auto_research/post_training/reviewed_oct18.py:note_has_answer、admit_note、rshd_aux_loss |
| 运行入口 | scripts/run_oct18_post_components.py |
原始论文总结¶
背景与主要改动¶
一组 GRPO rollout 全部失败时,组内 advantage 全为0。RSHD 不把参考答案告诉模型,而让同一策略根据失败响应写纠错 note,再用 note 生成新的 replay。只有 replay 通过率严格超过原组,note 才被接纳;之后蒸馏的也不是旧失败响应,而是 note 实际诱导的新续写。推理时只需无 note 学生。原文算法1–2
flowchart LR
A[原始 GRPO 组] --> B[失败响应]
B --> C[不读答案写 note]
C --> D[训练专用泄漏与格式筛查]
D --> E[带 note 新采样 replay]
E --> F{新通过率严格高于原组}
F -->|是| G[均匀选一条 replay]
G --> H[同前缀带 note 与无 note 评分]
H --> I[正 lift 蒸馏]
A --> J[保持原 GRPO]
I --> J
F -->|否| K[修订或拒绝]
原文 Figure 2 展示上述完整数据流;Figure 1 说明难题组出现全失败的比例,不能拿它作为本地数据分布。
原论文关键图¶
原论文 Figure 2(关键图):展示原论文方法的总体设计和关键组成。图片来自原论文,版权归原作者所有;点击图片可查看来源。
核心公式¶
原始组通过率 \(\hat p=n^{-1}\sum_iR(y_i)\);note 诱导的 \(k\) 条新 replay 通过率为 \(\hat r(s)\)。接纳要求 \(\hat r(s)>\hat p\),不能把相等或仅“能运行”当成改进,也不能直接从 replay 中挑成功的一条冒充均匀抽样。
对接纳的 replay \(\tilde y\),相同响应前缀在带 note 冻结策略和无 note 学生下的 log-prob 分别是 \(\ell^T_t\)、\(\ell^S_t\):
分母包含每条接纳 replay 的所有有效 token,零 lift 也计数。没有接纳 note 时附加项精确为0,不退化成旧失败响应的自模仿;所有 teacher/lift 均 stop-gradient。
论文离线与线上效果¶
原文4B模型的 MATH-500准确率从 GRPO 的76.9到84.5;平均生成 token 从2545降至1979。GPQA 同时报告 exact 和模型匹配结果,不能混成一个口径。完整训练约增加一倍生成量,而非免费改进。未报告线上 A/B。原文 §6
本地复现¶
数据协议与运行¶
随机32词表因果模型按每题8条响应真实采样;末 token 余数条件只作人工二值 verifier。note writer 的输入只有原提示、失败响应和“失败”标记,不包含 reference token。诊断用精确 token 子序列筛查答案泄漏,重放每 note 两条、严格比较原组通过率,再均匀选取一条 replay 做真实同前缀双评分与梯度更新。
PYTHONPATH=src python scripts/run_oct18_post_components.py \
--methods rshd --device cpu --seeds 42,43,44 \
--output runs/rshd-component.json
CUDA 改为 --device cuda;默认每 seed 3次、最多8次更新,输出拒绝覆盖并保留完整 commit、源码 SHA256 和逐 seed 指标。失败、拒绝与泄漏 note 计数保留,不通过筛选删除负结果。
结果、代码映射与边界¶
2026-10-18 已在真实 NVIDIA A100 上完成 seeds 42/43/44,各3次更新,三种子组件验收通过:回执、逐 seed 指标。学生参数变化 \(L_2\) 均值0.017943,各 seed 均有非零梯度及参数更新;截断正 lift 和全接纳 token 分母的独立参考最大误差约 \(1.11\times10^{-16}\)。空接纳的附加 loss 精确为0,相等通过率和泄漏 note 均被拒绝,冻结策略参数变化和教师梯度数均为0。
保留全部接纳与拒绝:三个 seed 分别接纳37/44/32个 note,拒绝59/48/63个;其中6/6/7个含 reference token 的 note 被筛掉。正 lift token 分别为55/63/41个。它们都来自人工 token 条件,不能解释成自然语言反思通过率或真实推理收益;writer 未接收参考答案。
关键限制: 随机 token 不是有语义的纠错 note。只实现精确子序列泄漏筛查,未实现自然语言语义泄漏/畸形 note 审核;有界 runner 不运行论文的第二轮 note 修订。这些缺口不能被真实梯度和多种子“升级”为完整反思能力复现,也不接 evolve。
完整实验不复现: 原文1.7B、4B、8B模型各两个领域,150步、每步14题×8 rollout、3072-token生成、额外 note/replay及多组外域评测;完整多臂训练对单卡预算成本过高。未启动完整复现,审查 DONE 不代表已验证论文效果。
