Dr. GRPO:去长度与组方差偏置的组策略优化¶
本页在公开候选策略或确定性 Agent mini-suite 上复现可隔离的 RL 更新机制;不把轻量实验写成原论文规模模型或 benchmark 结论。
论文信息¶
| 字段 | 内容 |
|---|---|
| 论文链接 | Dr. GRPO:去长度与组方差偏置的组策略优化(arXiv 2503.20783) |
| 公司 / 机构 | SAIL 研究团队 |
| 首次公开日期 | 2025-03-26 |
| 原作者代码 | 已开源 |
| 本地 adapter / 算法键 | dr-grpo |
| 本地复现代码 | src/auto_research/post_training/ |
原始论文总结¶
背景与主要改动¶
原始 GRPO 的 response 内长度平均和组内标准差会引入长度与题目难度偏置。Dr. GRPO 移除这两个归一化项,保留中心化的组相对奖励,让每条轨迹以原始尺度参与更新。
flowchart LR
G["response group"] --> C["mean-center rewards"]
C --> N["remove std / length normalization"]
N --> P["clipped policy update"]
原论文关键图¶
原论文 Figure 2(性能比较):不同 R1-Zero 训练配置在多个数学基准上的准确率对比。图片来自原论文,版权归原作者所有;点击图片可查看来源。
核心公式¶
\[
\hat A_i=r_i-\frac1G\sum_{j=1}^G r_j,\qquad \mathcal L_{\rm DrGRPO}=-\frac1G\sum_i\min(r_i(\theta)\hat A_i,\operatorname{clip}(r_i(\theta))\hat A_i).
\]
论文离线与线上效果¶
论文报告在保持推理效果的同时提升 token efficiency,并以极简 R1-Zero recipe 在 7B base 上得到 AIME24 43.3%;未报告线上 A/B。
本地复现¶
在相同 GSM8K candidate-policy、步数和 seed 下真实移除组 std 与伪长度归一化,并与 GRPO 的更新统计对比。
auto-research post-train --algorithm dr-grpo --dataset gsm8k-candidate --maximum-examples 256 --steps 120 --seed 42
固定 seed 汇总指标见 rl-papers-summary-seed42.json。
复现边界¶
候选策略没有自由生成 response 长度,伪长度只用于暴露归一化差异,不等同于论文的 token-level 长 CoT 训练。
