Make Sparse Rewards Count: Density-Aware Reward Aggregation for Multi-Reward RL
让稀疏奖励起作用:多奖励 RL 的密度感知奖励聚合
🧭 多奖励 RL 的密度感知聚合
University of Chinese Academy of Sciences · University of Minnesota Twin Cities · University of Wisconsin–Madison · Stony Brook University · Kuaishou Technology
提出 DARA,通过逆平方根密度校正放大稀疏奖励的信号,加速多奖励 RL 训练且不改变策略优化目标
📌 工作参考价值(4 点)
- 多目标 LLM 后训练中,若直接对多个 reward 求和或平均,稀疏目标(如格式合规、工具调用、长度合规)容易被稠密目标淹没。可仿照 DARA 在 GRPO/GDPO 实现中按 batch 内每个 reward 非零优势的比例动态计算权重,对稀疏目标施加 inverse-square-root 加权,无需人工反复调 reward coefficient。
- 监控每个 reward 的 advantage energy 或 active-group density,可以量化判断哪些目标学习停滞;当某个目标 density 骤降时自动提高其聚合权重,比训练后看指标再手动调参更早发现信号失衡。
- DARA 只改 reward aggregation,不改变底层 policy optimization objective 或梯度计算,可作为即插即用模块嵌入现有 RLHF/GRPO 流程,工程风险低;只需在 batch 内按 reward 保存 group-level advantages 即可计算 density。
- 对电商 Agent、对话式推荐、商品文案生成等需要同时满足「格式/调用/质量」等多约束的 LLM 应用,稀疏约束信号密度很低,DARA 的密度感知聚合能减少达到高合规率所需训练步数,降低迭代成本。