arXiv:2503.06287cs.CVcs.AI2025-03CVPR被引 98

仅用少数注意力头即可实现高效视觉定位,无需微调模型。

Your Large Vision-Language Model Only Needs A Few Attention Heads For Visual Grounding

论文配图:Your Large Vision-Language Model Only Needs A Few Attention Heads For Visual Grounding
图 1 · 摘自论文原文
  • 从冻结的视觉语言模型中提取特定注意力头,直接生成文本-图像对齐图。
  • 仅需3个注意力头就达到主流方法水平,性能媲美需微调的方法。
  • 适合追求轻量化、零训练部署的视觉定位场景。

视觉定位旨在定位与自由文本描述对应的图像区域。近期大型视觉语言模型(LVLM)的强大多模态能力推动了视觉定位的显著进步,但通常需要微调及额外组件来显式生成边界框或分割掩码。我们发现,冻结的LVLM中少数注意力头已具备强大的视觉定位能力。这些头在文本语义相关物体位置上持续聚焦,称为定位头。基于定位头,我们提出一种简单高效的无训练视觉定位框架,利用定位头生成的文本-图像注意力图识别目标对象。令人惊讶的是,仅需数千个注意力头中的3个,即可达到与需微调的现有方法相当的定位性能。研究结果表明,LVLM能基于对图文关系的深层理解,隐式关注相关图像区域以生成信息丰富的文本输出。所有源代码将公开发布。

原文摘要 · Abstract (English)

Visual grounding seeks to localize the image region corresponding to a free-form text description. Recently, the strong multimodal capabilities of Large Vision-Language Models (LVLMs) have driven substantial improvements in visual grounding, though they inevitably require fine-tuning and additional model components to explicitly generate bounding boxes or segmentation masks. However, we discover that a few attention heads in frozen LVLMs demonstrate strong visual grounding capabilities. We refer to these heads, which consistently capture object locations related to text semantics, as localization heads. Using localization heads, we introduce a straightforward and effective training-free visual grounding framework that utilizes text-to-image attention maps from localization heads to identify the target objects. Surprisingly, only three out of thousands of attention heads are sufficient to achieve competitive localization performance compared to existing LVLM-based visual grounding methods that require fine-tuning. Our findings suggest that LVLMs can innately ground objects based on a deep comprehension of the text-image relationship, as they implicitly focus on relevant image regions to generate informative text outputs. All the source codes will be made available to the public.

视觉定位注意力头零训练

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