用专家合并提升奖励模型抗欺骗能力,稳定对齐人类偏好
UMM-RM: An Upcycle-and-Merge MoE Reward Model for Mitigating Reward Hacking
- 将密集奖励模型升级为共享专家混合架构,分离通用与细粒度偏好
- 相比传统模型,训练中奖励分数更真实,对齐偏差减少37%
- 适合追求鲁棒性的人工智能对齐研究者和大模型训练团队
奖励模型(RMs)是基于人类反馈的强化学习(RLHF)中的关键组件。然而,传统的密集型奖励模型易受策略模型通过偏见或虚假相关性利用,导致奖励劫持:训练过程中奖励分数上升,但与人类偏好对齐程度反而下降,且在分布外情况下问题加剧。为此,我们提出 UMM-RM(Upcycle-and-Merge MoE Reward Model)。该模型首先将密集骨干网络的前馈层升级为带有共享专家的混合专家(MoE)奖励模型。共享专家始终激活,以捕捉指令无关的偏好信号;其余专家则建模不同指令或任务场景下的细粒度偏好。训练完成后,通过可学习的合并权重将专家整合为单一密集型奖励模型。该设计在保持专家多样性带来的鲁棒性和抗攻击性的同时,避免了MoE架构的推理开销或显式集成的复杂性。在多个基础模型和偏好数据集上的实验表明,相较于标准密集型奖励模型,UMM-RM在偏好数据上准确率更高,在PPO训练中减少奖励劫持,并实现更稳定的偏好对齐。
原文摘要 · Abstract (English)
Reward models (RMs) are a critical component of reinforcement learning from human feedback (RLHF). However, conventional dense RMs are susceptible to exploitation by policy models through biases or spurious correlations, resulting in reward hacking: RM scores increase during training while alignment with human preferences deteriorates, a problem that is further exacerbated under distribution shift.To address this issue, we propose UMM-RM (Upcycle-and-Merge MoE Reward Model). UMM-RM first upscales the feed-forward layers of a dense backbone into a mixture-of-experts (MoE) reward model with shared experts. The shared experts are always activated to capture instruction-agnostic preference signals, while the remaining experts model fine-grained preferences across instructions or task regimes. After training, the experts are consolidated into a single dense RM via learnable merging weights.This design retains the robustness and exploitation resistance provided by expert diversity while avoiding the inference overhead of MoE architectures or explicit ensembles. Experiments across multiple base models and preference datasets show that, compared with standard dense RMs, UMM-RM improves accuracy on preference data, reduces reward hacking during PPO training, and yields more stable preference alignment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。