解析视觉语言模型中注意力头如何传递图像到文本信息。
Interpreting Attention Heads for Image-to-Text Information Flow in Large Vision-Language Models
- 提出头归因方法,识别关键注意力头的作用模式。
- 发现注意力头选择由图像语义决定而非外观特征。
- 揭示文本与图像信息在令牌层面的分步传递路径。
大型视觉语言模型(LVLMs)通过一系列注意力头将图像信息传递至文本以回答视觉问题。由于大量注意力头同时运作,这种图像到文本的信息流机制难以解释。为此,我们提出头归因技术,借鉴组件归因方法,识别在信息传递中起关键作用的注意力头一致模式。通过该方法,我们研究了LVLM如何依赖特定注意力头识别并回答关于图像主体对象的问题。分析发现,一组特定注意力头促进图像到文本的信息流动;令人惊讶的是,这些头的选择由输入图像的语义内容决定,而非其视觉外观。我们在令牌层面进一步分析信息流动,发现:(1) 文本信息先传播至角色相关令牌和最后一个令牌,随后才接收图像信息;(2) 图像信息被嵌入于对象相关和背景相关令牌中。本工作表明,图像到文本的信息流具有结构性,且注意力头层级分析为理解LVLM机制提供了有前景的方向。
原文摘要 · Abstract (English)
Large Vision-Language Models (LVLMs) answer visual questions by transferring information from images to text through a series of attention heads. While this image-to-text information flow is central to visual question answering, its underlying mechanism remains difficult to interpret due to the simultaneous operation of numerous attention heads. To address this challenge, we propose head attribution, a technique inspired by component attribution methods, to identify consistent patterns among attention heads that play a key role in information transfer. Using head attribution, we investigate how LVLMs rely on specific attention heads to identify and answer questions about the main object in an image. Our analysis reveals that a distinct subset of attention heads facilitates the image-to-text information flow. Remarkably, we find that the selection of these heads is governed by the semantic content of the input image rather than its visual appearance. We further examine the flow of information at the token level and discover that (1) text information first propagates to role-related tokens and the final token before receiving image information, and (2) image information is embedded in both object-related and background tokens. Our work provides evidence that image-to-text information flow follows a structured process, and that analysis at the attention-head level offers a promising direction toward understanding the mechanisms of LVLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。