Learn What's Left, Not What's Mastered: Saturation Aware Advantage Reweighting for Multi-Reward Policy Optimization
学习未掌握部分而非已掌握部分:多奖励策略优化的饱和感知优势重加权
🧭 LLM 后训练 · 多奖励 RL 优化
University of Florida · UC San Diego · Northeastern University · Northwestern University · Stanford University
提出 SA-MRPO,独立标准化各奖励并按批次饱和度动态折扣优势,将优化重心转向未饱和目标,提升困难指标性能
📌 工作参考价值(4 点)
- 多目标 RL 微调时,不要先固定加权求和再标准化优势;改为对每个奖励独立标准化,并在每个 batch 估计目标饱和度(如 batch 平均奖励接近上限)来动态折扣其梯度贡献,可让模型自动将优化重心转向未达标目标,避免简单指标吸干预算。
- 推荐系统多任务学习可借鉴“饱和度感知动态损失权重”:每隔若干 step 根据验证集或 batch 上各目标当前水平计算饱和度,自动降低已达标任务的损失权重,把容量留给困难目标(如转化率)。
- 工程实现成本低:只需在 batch 内统计每个奖励的均值或分位数,设计递减函数(如 1 - 均值/上限)生成权重,无需额外超参或元学习,易嵌入现有 GRPO/PPO 流程。
- 多目标场景下应监控指标是否已达标,适时冻结或降低其训练信号,将 RL 信号导向更难提升的目标,符合“保持已满意水平而提升困难目标”的实践原则。