Retrieval Heads are Dynamic
Not on the map. Why?
Abstract (arXiv)
Recent studies have identified "retrieval heads" in Large Language Models (LLMs) responsible for extracting information from input contexts. However, prior works largely rely on static statistics aggregated across datasets, identifying heads that perform retrieval on average. This perspective overlooks the fine-grained temporal dynamics of autoregressive generation. In this paper, we investigate retrieval heads from a dynamic perspective. Through extensive analysis, we establish three core claims: (1) Dynamism: Retrieval heads vary dynamically across timesteps; (2) Irreplaceability: Dynamic retrieval heads are specific at each timestep and cannot be effectively replaced by static retrieval heads; and (3) Correlation: The model's hidden state encodes a predictive signal for future retrieval head patterns, indicating an internal planning mechanism. We validate these findings on the Needle-in-a-Haystack task and a multi-hop QA task, and quantify the differences on the utility of dynamic and static retrieval heads in a Dynamic Retrieval-Augmented Generation framework. Our study provides new insights into the internal mechanisms of LLMs.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 21 of its 44 references that are in the map. First posted on arXiv 7 Jan 2026.
Builds on
21 of 44 references.
References · 21
- 2017Attention Is All You Needcited by 59,932
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,686
- 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,383
- 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
- 2021Improving language models by retrieving from trillions of tokenscited by 1,278
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,130
- 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,085
- 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,054
- 2022In-context Learning and Induction Headscited by 930
9 more
- 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 915
- 2024SnapKV: LLM Knows What You are Looking for Before Generationcited by 848
- 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 844
- 2019Are Sixteen Heads Really Better than One?cited by 803
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 700
- 2020Asymmetric Loss For Multi-Label Classificationcited by 379
- 2024Retrieval Head Mechanistically Explains Long-Context Factualitycited by 219
- 2024Not All Heads Matter: A Head-Level KV Cache Compression Method with Integrated Retrieval and Reasoningcited by 96
- 2025Query-Focused Retrieval Heads Improve Long-Context Reasoning and Re-rankingcited by 30
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2025QuCo-RAG: Quantifying Uncertainty from the Pre-training Corpus for Dynamic Retrieval-Augmented Generation
- 2024DRAGIN: Dynamic Retrieval Augmented Generation based on the Information Needs of Large Language Models
- 2026To Memorize or to Retrieve: Scaling the Interaction Between Pretraining and Retrieval
- 2025To Retrieve or Not to Retrieve? Uncertainty Detection for Dynamic Retrieval Augmented Generation
- 2025Modeling Uncertainty Trends for Timely Retrieval in Dynamic RAG
- 2026Predictive Prefetching for Retrieval-Augmented Generation
- 2024Quantifying reliance on external information over parametric knowledge during Retrieval Augmented Generation (RAG) using mechanistic analysis
- 2024Think-then-Act: A Dual-Angle Evaluated Retrieval-Augmented Generation