arXiv:2511.13415cs.IRcs.CL2025-11被引 3

用多模态大模型注意力图指导文档检索,让模型知道哪部分内容相关。

Attention Grounded Enhancement for Visual Document Retrieval

  • 利用MLLM注意力图生成局部相关性信号,补充全局标签
  • 在ViDoRe V2上提升nDCG@1达12.82%,nDCG@5提升5.03%
  • 适合需要精准定位内容的视觉文档检索场景

视觉文档检索需理解异构多模态内容以满足隐式信息需求。现有方法采用截图编码与细粒度后期交互,显著提升性能,但训练仍依赖粗粒度全局相关性标签,无法揭示支持匹配的具体区域。这导致模型依赖表面线索,难以捕捉隐含语义关联,尤其在非抽取式查询下表现受限。为此,本文提出注意力引导增强框架AGREE,利用多模态大语言模型(MLLM)的跨模态注意力作为代理监督,引导检索器识别相关文档区域。具体地,AGREE从MLLM中提取注意力图,显示基于查询所关注的文档区域,这些注意力得分作为局部、区域级相关性信号。训练时,将局部信号与全局文档级相关标签结合,联合优化检索器。双重监督使模型不仅学习文档是否匹配,还学习驱动相关性的具体内容。在挑战性视觉文档检索基准ViDoRe V2上的实验表明,相比仅使用全局监督的基线,AGREE在平均nDCG@1和nDCG@5上分别提升12.82%和5.03%。定量与定性分析进一步证明,AGREE促进查询词与文档区域间的深层对齐,超越表面匹配,实现更准确且可解释的检索。代码已开源:https://github.com/VickiCui/AGREE。

原文摘要 · Abstract (English)

Visual document retrieval requires understanding heterogeneous and multi-modal content to satisfy implicit information needs. Recent advances use screenshot-based document encoding with fine-grained late interaction to encode holistic information and capture nuanced alignments, significantly improving retrieval performance. However, retrievers are still trained with coarse global relevance labels, without revealing which regions support the match. As a result, retrievers tend to rely on surface-level cues and struggle to capture implicit semantic connections, hindering their ability to handle non-extractive queries.To improve fine-grained relevance modeling, we propose a Attention-Grounded REtriever Enhancement (AGREE) framework. AGREE leverages cross-modal attention from multimodal large language models (MLLMs) as proxy supervision to guide the retriever in identifying relevant document regions. Specifically, AGREE extracts attention maps from the MLLM that highlight which document regions are attended to based on the query. These attention scores serve as local, region-level relevance signals. During training, AGREE combines local signals with the global document-level relevance label to jointly optimize the retriever. This dual-level supervision enables the model to learn not only whether documents match, but also which content drives relevance. Experiments on the challenging visual document retrieval benchmark, ViDoRe V2, show that AGREE significantly outperforms the global-supervision-only baseline by 12.82\% and 5.03\% in terms of average nDCG@1 and nDCG@5. Quantitative and qualitative analyses further demonstrate that AGREE promotes deeper alignment between query terms and document regions, moving beyond surface-level matching toward more accurate and interpretable retrieval. Our code is available at: https://github.com/VickiCui/AGREE.

视觉检索多模态注意力机制文档理解

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