arXiv:2608.27417cs.CV2026-08

发现视觉语言模型中定位图文关联的微型关键注意力头。

Retrieval Heads Meet Vision: Uncovering How VLMs Locate and Extract Visual Information

论文配图:Retrieval Heads Meet Vision: Uncovering How VLMs Locate and Extract Visual Information
图 1 · 摘自论文原文
  • 提出视觉检索头(VRH),仅占总注意力头1.7%-2.6%,负责图文对齐。
  • 屏蔽前20个VRH导致定位准确率最高下降80个百分点,远超随机头影响。
  • VRH通用性强,跨任务、跨模型结构仍保持因果性,适合模型可解释性研究者。

视觉语言模型(VLMs)能根据文本提示定位图像区域并将对应视觉证据传递至输出,但其内部机制尚不明确。受大语言模型中检索头的启发,本文提出视觉检索头(VRHs),即一组小型注意力头(约占总头数的1.7%-2.6%),对文本描述与图像区域的对齐具有因果作用。通过统一设计空间重构现有头评分方法,并采用基于真实参照区域的输出预测词求和策略,可最可靠地识别出因果头。在十一款VLM和五个指代表达基准上验证:仅屏蔽前20个VRH,定位准确率最高下降80个百分点;而屏蔽同等数量的随机头则影响微弱。除复制文本检索头的因果-稀疏-通用三特性外,VRH还表现出新属性:跨视觉参照任务泛化能力,即使在属性、空间、计数和视觉数学任务上依然保持因果性;功能特异性,破坏定位但保留输出格式;架构共享性,可在共享大语言模型主干但视觉编码器、投影器和指令微调不同的模型间转移因果作用。

原文摘要 · Abstract (English)

Vision-language models (VLMs) can locate an image region referred to by a text prompt and route the corresponding visual evidence to the output, yet the internal mechanism behind this behavior is not understood. Inspired by retrieval heads in large language models, we ask whether VLMs contain an analogous mechanism for visual retrieval. We answer affirmatively by introducing Visual Retrieval Heads (VRHs), a small subset of attention heads (about 1.7-2.6%) that are causally responsible for grounding text descriptions to image regions. To find them, we recast existing head-scoring methods under a unified design space over query tokens, key aggregation, and cross-sample aggregation. We then show that scoring attention from output prediction tokens with a sum over the ground-truth referent region most reliably identifies causal heads. Across eleven VLMs and five referring-expression benchmarks, masking only the top 20 VRHs reduces grounding accuracy by up to 80 percentage points, while masking the same number of random heads has little effect. Beyond replicating the causal-sparse-universal triad established for text retrieval heads, VRHs exhibit several properties not previously reported: they generalize across visual reference tasks, remaining causal on attribute, spatial, counting, and visual-math benchmarks despite being discovered through bounding-box prediction; they are functionally specific, preserving output format while corrupting localization; and they are architecturally shared, transferring causally across VLMs that share an LLM backbone but differ in vision encoder, projector, and instruction tuning.

视觉语言模型注意力机制可解释性因果分析

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