让多模态大模型用图像坐标思考,提升视觉推理准确性
GRIT: Teaching MLLMs to Think with Images
- 推理时交替输出语言和图像框坐标,实现视觉定位与思维结合
- 仅需20组图文问答数据即可训练,大幅降低标注成本
- 无需标注推理链或框位置,适合资源有限的视觉推理研究
近期研究证明强化学习可有效构建先生成思维链再输出答案的推理模型。然而,现有开源视觉语言推理模型多仅生成纯自然语言推理,缺乏对视觉信息的显式整合,限制了其推理过程的清晰性与视觉一致性。为此,我们提出一种新方法GRIT(Grounded Reasoning with Images and Texts),使多模态大模型在推理中结合图像。GRIT采用一种基于图像的推理范式,模型生成的语言推理链中穿插明确的边界框坐标,指向输入图像中所参考的区域。同时,GRIT引入基于GRPO算法的强化学习方法GRPO-GR,采用聚焦于最终答案准确率与推理输出格式的鲁棒奖励机制,无需依赖带推理链标注或显式框标签的数据。实验表明,仅需20个来自现有数据集的图像-问题-答案三元组,GRIT即能高效训练出连贯且视觉锚定的推理链,成功融合推理与视觉定位能力。
原文摘要 · Abstract (English)
Recent studies have demonstrated the efficacy of using Reinforcement Learning (RL) in building reasoning models that articulate chains of thoughts prior to producing final answers. However, despite ongoing advances that aim at enabling reasoning for vision-language tasks, existing open-source visual reasoning models typically generate reasoning content with pure natural language, lacking explicit integration of visual information. This limits their ability to produce clearly articulated and visually grounded reasoning chains. To this end, we propose Grounded Reasoning with Images and Texts (GRIT), a novel method for training MLLMs to think with images. GRIT introduces a grounded reasoning paradigm, in which models generate reasoning chains that interleave natural language and explicit bounding box coordinates. These coordinates point to regions of the input image that the model consults during its reasoning process. Additionally, GRIT is equipped with a reinforcement learning approach, GRPO-GR, built upon the GRPO algorithm. GRPO-GR employs robust rewards focused on the final answer accuracy and format of the grounded reasoning output, which eliminates the need for data with reasoning chain annotations or explicit bounding box labels. As a result, GRIT achieves exceptional data efficiency, requiring as few as 20 image-question-answer triplets from existing datasets. Comprehensive evaluations demonstrate that GRIT effectively trains MLLMs to produce coherent and visually grounded reasoning chains, showing a successful unification of reasoning and grounding abilities.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。