arXiv:2505.02835cs.CVcs.CL2025-05被引 64

用稳定强化学习训练多模态奖励模型,性能显著提升。

R1-Reward: Training Multimodal Reward Model Through Stable Reinforcement Learning

  • 将奖励建模重构为基于规则的强化学习任务
  • 在VL Reward-Bench上提升8.4%,Multimodal Reward Bench上提升14.3%
  • 适合追求高精度多模态评估的模型开发者

多模态奖励模型(MRMs)对提升多模态大语言模型(MLLMs)性能至关重要。尽管近期研究主要聚焦于模型结构与训练数据的改进,但对长程推理能力在奖励建模中的有效性及其激活方式仍缺乏探索。本文探讨了强化学习(RL)在奖励建模中的应用,将奖励建模问题重新定义为基于规则的强化学习任务。然而,直接使用现有算法如Reinforce++常导致训练不稳定甚至崩溃。为此,我们提出StableReinforce算法,通过优化训练损失、优势估计策略和奖励设计,实现更稳定的训练动态与更优性能。为支持MRM训练,我们从多个数据集收集了20万条偏好数据。基于该数据集,使用StableReinforce训练的R1-Reward模型,在多模态奖励建模基准上表现优异:相比先前最先进模型,在VL Reward-Bench上提升8.4%,在Multimodal Reward Bench上提升14.3%。此外,随着推理计算资源增加,其性能进一步增强,凸显强化学习在优化MRMs方面的潜力。

原文摘要 · Abstract (English)

Multimodal Reward Models (MRMs) play a crucial role in enhancing the performance of Multimodal Large Language Models (MLLMs). While recent advancements have primarily focused on improving the model structure and training data of MRMs, there has been limited exploration into the effectiveness of long-term reasoning capabilities for reward modeling and how to activate these capabilities in MRMs. In this paper, we explore how Reinforcement Learning (RL) can be used to improve reward modeling. Specifically, we reformulate the reward modeling problem as a rule-based RL task. However, we observe that directly applying existing RL algorithms, such as Reinforce++, to reward modeling often leads to training instability or even collapse due to the inherent limitations of these algorithms. To address this issue, we propose the StableReinforce algorithm, which refines the training loss, advantage estimation strategy, and reward design of existing RL methods. These refinements result in more stable training dynamics and superior performance. To facilitate MRM training, we collect 200K preference data from diverse datasets. Our reward model, R1-Reward, trained using the StableReinforce algorithm on this dataset, significantly improves performance on multimodal reward modeling benchmarks. Compared to previous SOTA models, R1-Reward achieves a $8.4\%$ improvement on the VL Reward-Bench and a $14.3\%$ improvement on the Multimodal Reward Bench. Moreover, with more inference compute, R1-Reward's performance is further enhanced, highlighting the potential of RL algorithms in optimizing MRMs.

奖励模型强化学习多模态稳定训练

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