破解异步强化学习策略偏差!Proximal KL 首次实现 Token 级离线重要性采样:Qwen3 实测超越 GRPO,训练提速防坍塌
在可验证奖励强化学习(RLVR)中,异步采样生成与响应复用导致严重离线策略偏差。传统 Token 级加权会导致策略梯度有偏,而序列级加权则面临方差爆炸。UIUC 与港科大团队提出 Proximal KL,数学证明在覆盖度与局部表达性条件下,完整未来 KL 积分可使近端优化器具备唯一稳态策略。实测在 Qwen3-1.7B 与 4B 上超越 GRPO 高达 2.58%,并彻底根除周期性刷新下的模型崩溃。
核心要点速览
- 数学破局策略梯度偏差:证明未来 KL 积分使近端优化器具备唯一稳态,解耦参考策略与采样策略,告别序列级方差爆炸
- 基准跑分全面超越 GRPO:在 Qwen3-1.7B 与 4B 的 8 项数学测试集微平均准确率提升最高达 2.58%,AIME24 pass@16 显著跃升
- 根除周期性刷新训练崩溃:在异步多步离线复用下杜绝标准 PG 与 CISPO 常见的灾难性坍塌,以更少更新步数达到更高收敛精度