‹ 返回 2026-06-21

基于视觉的思考

Thinking with Visual Grounding

▲ 5 💬 3 2026-06-21

Junkai Zhang, Yihe Deng, Kai-Wei Chang, Wei Wang

摘要

视觉性思维不仅应该听起来合理,还应该有相应的证据支持。虽然最近的视觉-语言模型能够产生自然语言推理过程,但这些过程往往无法明确显示出相关的图像区域,因此难以验证和监管。我们提出了“基于图像的思考”这种推理方式——在这种方式中,模型将自然语言表达与每一步所使用的视觉证据的明确点或框标记相结合。这样,模型就可以用语言来表达中间推理过程,同时让关键对象与它们所指的图像区域相关联。为了训练这种行为,我们构建了一个可扩展的合成流程:该流程能够提取正确的视觉推理过程,找出这些过程中所需的视觉对象,然后利用基于SAM3的代理来为这些对象提供明确的定位信息,最后从得到的掩码中获取相应的点、框标记信息。我们还提出了“关注定位的强化学习”方法,它将答案的正确性奖励与明确的定位奖励结合起来,以评估生成的物体引用是否与正确的图像证据相符。在两个计数基准测试和四个空间推理基准测试中,加入基于图像的思考机制后,Gemma3-4B-IT模型的性能明显优于原始模型和没有基于图像的思考机制的模型。在空间推理方面,基于图像的思考机制的4B模型与同一模型家族中的Gemma3-27B-IT模型相当,甚至在某些情况下更优。我们的分析表明,点定位非常适合用于计数任务,而框定位则更受益于明确的定位奖励。总体而言,我们的结果表明,当视觉-语言模型的中间推理与对应的图像区域相关联时,其性能会得到提升。

English Abstract

Visual thinking should not only sound right; it should show its evidence. While recent vision-language models (VLMs) can produce natural-language reasoning traces, these traces often leave the supporting image regions implicit, making them hard to verify and difficult to supervise. We introduce visually grounded thinking, a reasoning process in which models interleave natural-language thoughts with explicit point or box groundings of the visual evidence used at each step. This lets the model express intermediate reasoning in language while grounding key objects in the image regions they refer to. To train this behavior, we construct a scalable synthesis pipeline that distills correct visual reasoning traces, extracts the visual objects required by the traces, grounds them with a SAM3-based agent, and derives aligned point and box supervision from the resulting masks. We further propose grounding-aware reinforcement learning, which combines answer correctness rewards with dense grounding rewards that score whether generated object references match the correct image evidence. Across two counting benchmarks and four spatial reasoning benchmarks, adding visually grounded thinking to Gemma3-4B-IT consistently improves performance over the original model and the non-grounded thinking baseline. On spatial reasoning, the visually grounded thinking 4B models match, and in some cases surpass, Gemma3-27B-IT from the same model family. Our analysis shows that point grounding is well suited to counting, while box grounding benefits most from explicit grounding rewards on spatial tasks. Overall, our results show that VLMs think better when their intermediate thoughts are tied to the image regions that make them true.