arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2512.03438cs.AI
收藏

具有自适应验证器的多模态强化学习

Multimodal Reinforcement Learning with Adaptive Verifier for AI Agents

Microsoft Research(微软研究院);UMass Amherst(马萨诸塞大学阿默斯特分校);ETH Zurich(苏黎世联邦理工学院);UW–Madison(威斯康星大学麦迪逊分校)
Reuben Tan, Baolin Peng, Zhengyuan Yang, Hao Cheng, Oier Mees, Theodore Zhao, Andrea Tupini, Isar Meijer, Qianhui Wu, Yuncong Yang, Lars Liden, Yu Gu, Sheng Zha… 展开作者

Reuben Tan, Baolin Peng, Zhengyuan Yang, Hao Cheng, Oier Mees, Theodore Zhao, Andrea Tupini, Isar Meijer, Qianhui Wu, Yuncong Yang, Lars Liden, Yu Gu, Sheng Zhang, Xiaodong Liu, Lijuan Wang, Marc Pollefeys, Yong Jae Lee, Jianfeng Gao

AI总结:

本文提出Argos验证器,通过结合SFT数据筛选与RL训练,提升多模态推理模型在空间推理、视觉幻觉及机器人任务中的性能,同时减少奖励黑客问题。

摘要:

训练多模态强化学习(MMRL)的代理推理模型已变得越来越有能力,但几乎普遍使用基于最终答案的稀疏、结果导向的奖励进行优化。通过计算推理标记的更丰富的奖励可以显著提高学习效果,因为它提供了更细粒度的指导。然而,在MMRL中计算更多信息的奖励具有挑战性,因为不同样本可能需要不同的评分函数,且教师模型可能提供嘈杂的奖励信号。在本文中,我们引入了Argos(用于基于地面和客观评分的代理奖励),一种原理化的奖励代理,用于训练多模态推理模型以完成代理任务。对于每个样本,Argos从一组教师模型衍生和基于规则的评分函数中选择,同时评估:(i)最终响应的准确性,(ii)所指实体和动作的空间时间定位,以及(iii)推理过程的质量。我们发现,通过利用我们的代理验证器在SFT数据筛选和RL训练中,我们的模型在多个代理任务如空间推理、视觉幻觉以及机器人和具身AI基准上均取得最佳结果。关键的是,我们证明仅依赖于在高度筛选的推理数据上进行SFT后训练是不够的,因为代理在RL过程中会不可避免地崩溃到无根的解决方案,而没有我们的在线验证。我们还展示了我们的代理验证器可以帮助减少MMRL中的奖励黑客问题。最后,我们还通过帕累托最优的概念提供了Argos有效性的理论依据。

英文摘要:

Agentic reasoning models trained with multimodal reinforcement learning (MMRL) have become increasingly capable, yet they are almost universally optimized using sparse, outcome-based rewards computed based on the final answers. Richer rewards computed from the reasoning tokens can improve learning significantly by providing more fine-grained guidance. However, it is challenging to compute more informative rewards in MMRL beyond those based on outcomes since different samples may require different scoring functions and teacher models may provide noisy reward signals too. In this paper, we introduce the Argos (Agentic Reward for Grounded & Objective Scoring), a principled reward agent to train multimodal reasoning models for agentic tasks. For each sample, Argos selects from a pool of teacher-model derived and rule-based scoring functions to simultaneously evaluate: (i) final response accuracy, (ii) spatiotemporal localization of referred entities and actions, and (iii) the quality of the reasoning process. We find that by leveraging our agentic verifier across both SFT data curation and RL training, our model achieves state-of-the-art results across multiple agentic tasks such as spatial reasoning, visual hallucination as well as robotics and embodied AI benchmarks. Critically, we demonstrate that just relying on SFT post-training on highly curated reasoning data is insufficient, as agents invariably collapse to ungrounded solutions during RL without our online verification. We also show that our agentic verifier can help to reduce reward-hacking in MMRL. Finally, we also provide a theoretical justification for the effectiveness of Argos through the concept of pareto-optimality.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑