SA-MRPO:面向多奖励策略优化的饱和感知优势重加权方法

HuggingFace Daily Papers(社区热门论文)·2026-08-17 08:00·9天前
AI 导读

针对多奖励强化学习中固定加权求和导致已饱和目标持续占用梯度预算的问题,研究者提出SA-MRPO,对每个奖励目标独立标准化,并按批次级饱和估计自适应折扣其贡献。在数学推理的15项基准对比中,SA-MRPO在12项上优于GDPO,AIME24最高提升5%;自适应推理五项基准平均提升3.8%,AMC23最高提升9.2%;代码基准通过率最高提升2.3%,同时保持已满足目标性能。

HuggingFace Daily Papers(社区热门论文)
50AI 编辑部评分,满分 100

SA-MRPO:面向多奖励策略优化的饱和感知优势重加权方法

2026-08-17 08:00· 9天前
AI 导读

针对多奖励强化学习中固定加权求和导致已饱和目标持续占用梯度预算的问题,研究者提出SA-MRPO,对每个奖励目标独立标准化,并按批次级饱和估计自适应折扣其贡献。在数学推理的15项基准对比中,SA-MRPO在12项上优于GDPO,AIME24最高提升5%;自适应推理五项基准平均提升3.8%,AMC23最高提升9.2%;代码基准通过率最高提升2.3%,同时保持已满足目标性能。

Reinforcement learning (RL) with group-relative advantages has become the de facto standard for post-training language model reasoners. However, when optimizing multiple reward objectives, existing methods typically scalarize the reward vector with a fixed weighted sum before group-wise standardization. We show that this design leads to two fundamental problems: rollouts with distinct reward profiles can receive identical advantages, and all objectives are optimized with fixed relative weights regardless of their current level of saturation. As a result, training continues to allocate gradient budget to already-solved objectives instead of focusing on those with greater remaining headroom. We introduce Saturation Aware Advantage Reweighting for Multi-Reward Policy Optimization (SA-MRPO), which standardizes each reward objective independently and adaptively discounts its contribution according to a batch-level estimate of objective saturation. This dynamically reallocates optimization effort toward under-optimized objectives while empirically maintaining performance on those that are already well satisfied. We further show that saturation-aware reweighting can reverse the sign of an update, rather than merely rescale its magnitude. Across mathematical reasoning with two- and three-objective reward combinations, SA-MRPO improves the harder correctness objective over GDPO in 12 of 15 benchmark comparisons, with gains of up to 5% on AIME24. On adaptive reasoning it improves accuracy on all five benchmarks, by 3.8% on average and up to 9.2 % on AMC23, and on coding benchmarks it improves pass rate by up to 2.3%, while in all settings maintaining the easier objectives near their already satisfied levels.

来源:HuggingFace Daily Papers(社区热门论文)· arxiv.org