arXiv:2503.01773cs.CL2025-03ICML被引 108

通过注意力机制分析发现,视觉语言模型空间推理失败主因是注意力未对准物体位置。

Why Is Spatial Reasoning Hard for VLMs? An Attention Mechanism Perspective on Focus Areas

论文配图:Why Is Spatial Reasoning Hard for VLMs? An Attention Mechanism Perspective on Focus Areas
图 1 · 摘自论文原文
  • 从注意力分布角度解析模型内部状态,发现对准物体位置是关键
  • 提出无需训练的ADAPTVIS方法,在WhatsUp上提升50个百分点
  • 适合研究模型可解释性与提升视觉语言模型空间理解能力

大型视觉语言模型(VLMs)长期在空间推理任务上表现不佳。令人意外的是,即使仅涉及两个物体的简单空间关系(如'under'或'behind'),当前模型也难以准确识别。本文从机制可解释性视角出发,深入分析图像与文本标记间的内部交互,通过追踪中间层的注意力分布发现:成功完成空间推理与模型注意力对准真实物体位置密切相关,尤其在熟悉与不熟悉的空间关系间差异显著。基于此,我们提出ADAPTVIS方法,利用推理时置信度分数动态调整注意力:高置信时聚焦关键区域,低置信时扩大上下文视野。该无需训练的解码策略在WhatsUp和VSR等基准上实现显著提升(最高达50绝对百分点),且计算开销极低。代码与数据已公开于https://github.com/shiqichen17/AdaptVis。

原文摘要 · Abstract (English)

Large Vision Language Models (VLMs) have long struggled with spatial reasoning tasks. Surprisingly, even simple spatial reasoning tasks, such as recognizing "under" or "behind" relationships between only two objects, pose significant challenges for current VLMs. In this work, we study the spatial reasoning challenge from the lens of mechanistic interpretability, diving into the model's internal states to examine the interactions between image and text tokens. By tracing attention distribution over the image through out intermediate layers, we observe that successful spatial reasoning correlates strongly with the model's ability to align its attention distribution with actual object locations, particularly differing between familiar and unfamiliar spatial relationships. Motivated by these findings, we propose ADAPTVIS based on inference-time confidence scores to sharpen the attention on highly relevant regions when confident, while smoothing and broadening the attention window to consider a wider context when confidence is lower. This training-free decoding method shows significant improvement (e.g., up to a 50 absolute point improvement) on spatial reasoning benchmarks such as WhatsUp and VSR with negligible cost. We make code and data publicly available for research purposes at https://github.com/shiqichen17/AdaptVis.

空间推理注意力机制视觉语言模型可解释性

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