mlsanityData to 7 Oct 2026

Can Retrieval Heads See Images? Multimodal Retrieval Heads in Long-Context Vision-Language Models

arXiv 2605.27243 ↗First posted 26 May 2026cs.CVcited by 1

On the map: Multimodal LLMs / vision-language models → Document understanding and multimodal RAG

Abstract (arXiv)

Large vision-language models increasingly rely on long-context modeling to reason over documents, hour-level videos, and long-horizon agent trajectories, requiring them to locate relevant evidence across interleaved text and images. Prior work has studied this behavior using retrieval heads in large language models, but its copy-based criterion does not directly apply when evidence appears in images. We introduce a multimodal retrieval head detection method that scores attention from question tokens to textual or visual evidence. With this method, we show that multimodal retrieval heads are sparse, intrinsic, and causally important: only 4.4-10.2% of attention heads account for 50% of the positive retrieval-score mass, and masking the top-5% selected heads reduces performance on two tasks from MMLongBench by 42.5 and 62.3 percentage points, while random-head masking is far less damaging. Further analysis shows that these heads are partly shared across modalities yet remain dynamic within each modality, with image retrieval heads changing more than text retrieval heads as context length and haystack modality change. Without further training, we find that these heads can also be used directly to rank visually rich documents: on MMDocIR, Qwen3-VL-8B selected-head scoring improves Recall@1 by 7.7/7.4 macro/micro points for page retrieval and 6.3/6.8 points for layout retrieval over the strongest reported baseline.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Document understanding and multimodal RAG, one of 16 directions in Multimodal LLMs / vision-language models. Builds on 22 of its 36 references that are in the map. First posted on arXiv 26 May 2026.

Builds on

22 of 36 references.

References · 22

  1. 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,779
  2. 2023MMMU: A Massive Multi-discipline Multimodal Understanding and Reasoning Benchmark for Expert AGIcited by 2,044
  3. 2023MathVista: Evaluating Mathematical Reasoning of Foundation Models in Visual Contextscited by 1,582
  4. 2024Video-MME: The First-Ever Comprehensive Evaluation Benchmark of Multi-modal LLMs in Video Analysiscited by 1,375
  5. 2021Unsupervised Dense Information Retrieval with Contrastive Learningcited by 1,203
  6. 2020ColBERT: Efficient and Effective Passage Search via Contextualized Late Interaction over BERTcited by 1,160
  7. 2019Attention is not Explanationcited by 916
  8. 2024Measuring Multimodal Mathematical Reasoning with MATH-Vision Datasetcited by 516
  9. 2024MMMU-Pro: A More Robust Multi-discipline Multimodal Understanding Benchmarkcited by 494
  10. 2023OPERA: Alleviating Hallucination in Multi-Modal Large Language Models via Over-Trust Penalty and Retrospection-Allocationcited by 468
  11. 2019Is Attention Interpretable?cited by 387
  12. 2023Label Words are Anchors: An Information Flow Perspective for Understanding In-Context Learningcited by 235
10 more
  1. 2025NoLiMa: Long-Context Evaluation Beyond Literal Matchingcited by 120
  2. 2025WebWatcher: Breaking New Frontier of Vision-Language Deep Research Agentcited by 114
  3. 2024Unifying Multimodal Retrieval via Document Screenshot Embeddingcited by 107
  4. 2025Your Large Vision-Language Model Only Needs A Few Attention Heads For Visual Groundingcited by 78
  5. 2022VL-InterpreT: An Interactive Visualization Tool for Interpreting Vision-Language Transformerscited by 53
  6. 2024Attention in Large Language Models Yields Efficient Zero-Shot Re-Rankerscited by 42
  7. 2024Unveiling Visual Perception in Language Models: An Attention Head Analysis Approachcited by 34
  8. 2025Query-Focused Retrieval Heads Improve Long-Context Reasoning and Re-rankingcited by 30
  9. 2025MMLongBench: Benchmarking Long-Context Vision-Language Models Effectively and Thoroughlycited by 29
  10. 2026Training Long-Context Vision-Language Models Effectively with Generalization Beyond 128K Contextcited by 1

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026Retrieval Heads Meet Vision: Uncovering How VLMs Locate and Extract Visual Information
  2. 2026Mechanistic Insights into Functional Sparsity in Multimodal LLMs via CoRe Heads
  3. 2025mRAG: Elucidating the Design Space of Multi-modal Retrieval-Augmented Generation
  4. 2026Benchmarking Deflection and Hallucination in Large Vision-Language Models
  5. 2024MRAG-Bench: Vision-Centric Evaluation for Retrieval-Augmented Multimodal Models
  6. 2026VERA: Identifying and Leveraging Visual Evidence Retrieval Heads in Long-Context Understanding
  7. 2026MultiHaystack: Benchmarking Multimodal Retrieval and Reasoning over 40K Images, Videos, and Documents
  8. 2026Indexing Multimodal Language Models for Large-scale Image Retrieval