arXiv:2608.03092cs.LGcs.AI2026-08被引 1

解决多奖励学习中稀疏奖励信号弱的问题,提升模型综合能力。

SMOPD: Multi-Reward Reinforcement Learning via Specialize-and-Merge Online Policy Distillation

论文配图:SMOPD: Multi-Reward Reinforcement Learning via Specialize-and-Merge Online Policy Distillation
图 1 · 摘自论文原文
  • 分阶段训练:先针对不同奖励专精教学,再融合成统一策略。
  • 在1.5B至7B模型上均优于现有方法,尤其改善稀疏奖励表现。
  • 适合需要平衡多种奖励目标的强化学习任务,如工具调用与安全生成。

本文旨在提升多奖励强化学习中的模型性能。现有方法GDPO虽通过分维归一化缓解了奖励信号相互遮蔽的问题,但实验表明其仍难以平衡粒度不同的奖励。例如,在某些任务中,模型同时接收细粒度奖励(0.1~1.0)和二值稀疏奖励(0或1),后者常因信号过弱导致能力无法有效强化。为此,本文提出SMOPD——一种两阶段多奖励优化方法。第一阶段‘专精’:基于奖励优先级配置,训练多个专注于特定奖励的教师模型,确保各奖励信号能有效驱动优化;第二阶段‘融合’:采用在线策略蒸馏,将各教师的专长能力整合进一个学生策略,实现任务层面的均衡优化。在互补奖励(工具调用准确率与格式)和冲突奖励(有用性与安全性)设置下,SMOPD在1.5B、3B和7B模型上均显著超越GDPO。

原文摘要 · Abstract (English)

We aim to improve model performance in multi-reward reinforcement learning training process. Existing Group reward-Decoupled Normalization Policy Optimization (GDPO) has mitigated the issue of reward signals masking one another during direct scalarization by normalizing each reward dimension separately before aggregation. However, our experiments show that GDPO still struggles to balance reward signals with different granularities. Specifically, in some particular training tasks, the model may receive a dense reward that assigns fine-grained scores ranging from 0.1 to 1.0, together with a sparse reward that provides only binary feedback of either 0 or 1. In such cases, we find that the sparse reward may provide an insufficient optimization signal, preventing its corresponding capability from being effectively reinforced. Therefore, how can we strengthen the optimization signal from the sparse reward without sacrificing the capability already learned from the fine-grained reward? To overcome this limitation, we propose Specialize-and-Merge Online Policy Distillation (SMOPD), a two-stage training method for multi-reward optimization. Stage1-Specialize: SMOPD first employs reward-priority configurations to train multiple reward-specialized teachers, allowing each reward to be learned under conditions where its signal can effectively drive optimization. Stage2-Merge: SMOPD then utilizes online policy distillation to combine the reward-specialized capabilities of these teachers into a single student policy, while maintaining balanced task-level optimization. To validate our method, we conduct experiments on two multi-reward settings: complementary rewards(tool-calling accuracy and format) and conflicting rewards (helpful and harmless rewards). Based on above settings, SMOPD outperforms GDPO across 1.5B, 3B and 7B backbones.

强化学习多奖励策略蒸馏模型优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。