arXiv:2505.17018cs.CV2025-05被引 52

给多模态大模型的推理过程打分,提升逻辑能力和泛化性能。

SophiaVL-R1: Reinforcing MLLMs Reasoning with Thinking Reward

  • 训练思维奖励模型,评估推理全过程质量。
  • 在数学和多模态任务上超越多个主流模型,7B版本胜过72B大模型。
  • 适合关注模型推理可信度与高效训练的研究者。

近期研究通过基于规则的强化学习(RL)与结果奖励,成功激发多模态大语言模型(MLLMs)的强推理能力。然而,该范式通常缺乏对最终结果背后推理过程的监督,导致模型可能习得次优推理策略,影响泛化能力。为此,我们提出SophiaVL-R1,旨在为推理过程引入奖励信号。首先训练一个思维奖励模型,评估完整推理过程的质量。针对部分样本的思维奖励可能因奖励劫持而不可靠的问题,我们提出信任-GRPO方法,在训练中根据正确与错误回答的思维奖励对比计算可信权重,以缓解不可靠奖励的影响。此外,设计了渐进式训练策略,逐步降低思维奖励权重,使模型后期更依赖精确的规则化结果奖励。实验表明,SophiaVL-R1在多个基准测试(如MathVisita、MMMU)上优于一系列推理型MLLMs,展现出强大的推理与泛化能力。值得注意的是,其7B版本在多数基准上甚至超过参数量达10倍的LLaVA-OneVision-72B。所有代码、模型与数据集均公开于https://github.com/kxfan2002/SophiaVL-R1。

原文摘要 · Abstract (English)

Recent advances have shown success in eliciting strong reasoning abilities in multimodal large language models (MLLMs) through rule-based reinforcement learning (RL) with outcome rewards. However, this paradigm typically lacks supervision over the thinking process leading to the final outcome. As a result, the model may learn sub-optimal reasoning strategies, which can hinder its generalization ability. In light of this, we propose SophiaVL-R1, as an attempt to add reward signals for the thinking process in this paradigm. To achieve this, we first train a thinking reward model that evaluates the quality of the entire thinking process. Given that the thinking reward may be unreliable for certain samples due to reward hacking, we propose the Trust-GRPO method, which assigns a trustworthiness weight to the thinking reward during training. This weight is computed based on the thinking reward comparison of responses leading to correct answers versus incorrect answers, helping to mitigate the impact of potentially unreliable thinking rewards. Moreover, we design an annealing training strategy that gradually reduces the thinking reward over time, allowing the model to rely more on the accurate rule-based outcome reward in later training stages. Experiments show that our SophiaVL-R1 surpasses a series of reasoning MLLMs on various benchmarks (e.g., MathVisita, MMMU), demonstrating strong reasoning and generalization capabilities. Notably, our SophiaVL-R1-7B even outperforms LLaVA-OneVision-72B on most benchmarks, despite the latter having 10 times more parameters. All code, models, and datasets are made publicly available at https://github.com/kxfan2002/SophiaVL-R1.

多模态推理增强强化学习模型效率

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