arXiv:2511.12738cs.CV2025-11被引 7

让视觉模型更准确地关注图像相关区域,提升问答与指代理解能力。

Direct Visual Grounding by Directing Attention of Visual Tokens

  • 设计新损失函数,直接引导语言令牌关注对应视觉区域。
  • 在合成与真实数据上,几何任务与指代理解显著提升。
  • 适用于需要精准视觉定位的视觉语言模型优化场景。

视觉语言模型(VLM)将视觉标记与文本标记混合处理,但实验发现:在语言模型的最终层中,与查询最相关的视觉标记却极少获得回答标记的关注,这可能导致错误答案。标准的下一个词预测(NTP)损失对引导视觉标记注意力作用不足。为此,我们提出一种新损失函数——KL注意力损失(KLAL),通过最小化视觉标记注意力分布与真实注意力图之间的KL散度,直接监督视觉标记对语言标记的注意力。真实注意力图来自标注(如边界框或点标注)或合成任务中的几何关系,无需额外标签即可用于语言模型内部的注意力指导。该方法结合NTP损失,在几何任务、指代理解和指向任务中均实现显著性能提升。我们还构建了一个新数据集用于评估视觉语言模型的线段追踪能力,结果表明连商业级模型也表现不佳。

原文摘要 · Abstract (English)

Vision Language Models (VLMs) mix visual tokens and text tokens. A puzzling issue is the fact that visual tokens most related to the query receive little to no attention in the final layers of the LLM module of VLMs from the answer tokens, where all tokens are treated equally, in particular, visual and language tokens in the LLM attention layers. This fact may result in wrong answers to visual questions, as our experimental results confirm. It appears that the standard next-token prediction (NTP) loss provides an insufficient signal for directing attention to visual tokens. We hypothesize that a more direct supervision of the attention of visual tokens to corresponding language tokens in the LLM module of VLMs will lead to improved performance on visual tasks. To demonstrate that this is indeed the case, we propose a novel loss function that directly supervises the attention of visual tokens. It directly grounds the answer language tokens in images by directing their attention to the relevant visual tokens. This is achieved by aligning the attention distribution of visual tokens to ground truth attention maps with KL divergence. The ground truth attention maps are obtained from task geometry in synthetic cases or from standard grounding annotations (e.g., bounding boxes or point annotations) in real images, and are used inside the LLM for attention supervision without requiring new labels. The obtained KL attention loss (KLAL) when combined with NTP encourages VLMs to attend to relevant visual tokens while generating answer tokens. This results in notable improvements across geometric tasks, pointing, and referring expression comprehension on both synthetic and real-world data, as demonstrated by our experiments. We also introduce a new dataset to evaluate the line tracing abilities of VLMs. Surprisingly, even commercial VLMs do not perform well on this task.

视觉定位注意力机制多模态模型优化

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