arXiv:2603.25108cs.CV2026-03中稿 · CVPR被引 5

用多阶段强化学习让少数据也能训练出强 multimodal 奖励模型

MSRL: Scaling Generative Multimodal Reward Modeling via Multi-Stage Reinforcement Learning

  • 先用文本偏好数据学通用推理能力,再逐步迁移到图文任务
  • 在两个评测集上性能提升超9个百分点,无需额外标注
  • 适合想低成本训练视觉理解与生成模型的研究者

近年来,多模态奖励建模的进展主要源于从判别式到生成式范式的转变。在此基础上,研究者采用可验证奖励的强化学习(RLVR)来增强多模态奖励模型(MRMs)。然而,基于RLVR的训练通常依赖于昂贵且耗时的多模态偏好数据,难以规模化。为此,我们提出多阶段强化学习(MSRL),可在有限多模态数据下实现可扩展的强化学习训练。MSRL通过三阶段策略:首先从大规模文本偏好数据中学习通用奖励推理能力,再通过基于字幕的阶段逐步迁移至多模态任务,最后完成全多模态强化学习。同时引入跨模态知识蒸馏,提升偏好泛化能力。大量实验表明,MSRL有效扩展了生成式MRM的RLVR训练,在视觉理解与生成任务上均显著提升性能(如在VL-RewardBench上从66.6%提升至75.9%,在GenAI-Bench上从70.2%提升至75.7%),且无需额外多模态偏好标注。代码已公开。

原文摘要 · Abstract (English)

Recent advances in multimodal reward modeling have been largely driven by a paradigm shift from discriminative to generative approaches. Building on this progress, recent studies have further employed reinforcement learning from verifiable rewards (RLVR) to enhance multimodal reward models (MRMs). Despite their success, RLVR-based training typically relies on labeled multimodal preference data, which are costly and labor-intensive to obtain, making it difficult to scale MRM training. To overcome this limitation, we propose a Multi-Stage Reinforcement Learning (MSRL) approach, which can achieve scalable RL for MRMs with limited multimodal data. MSRL replaces the conventional RLVR-based training paradigm by first learning a generalizable reward reasoning capability from large-scale textual preference data, and then progressively transferring this capability to multimodal tasks through caption-based and fully multimodal reinforcement-learning stages. Furthermore, we introduce a cross-modal knowledge distillation approach to improve preference generalization within MSRL. Extensive experiments demonstrate that MSRL effectively scales the RLVR-based training of generative MRMs and substantially improves their performance across both visual understanding and visual generation tasks (e.g., from 66.6% to 75.9% on VL-RewardBench and from 70.2% to 75.7% on GenAI-Bench), without requiring additional multimodal preference annotations. Our code is available at: https://github.com/wangclnlp/MSRL.

多模态强化学习奖励模型生成模型

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