Understanding the Skill Gap in Recurrent Language Models: The Role of the Gather-and-Aggregate Mechanism
Not on the map. Why?
Abstract (arXiv)
State-space models (SSMs) offer efficient alternatives to Transformers for long sequences, but their fixed-size recurrent state limits capability on algorithmic tasks, such as retrieving past context. In this work, we examine how in-context retrieval operates in Transformer- and SSM-based language models and find that both rely on a similar Gather-and-Aggregate (G&A) mechanism: a Gather Head extracts relevant information pieces from context, which an Aggregate Head integrates into a single representation. In both architectures, G&A concentrates in a few heads, forming critical bottlenecks even for simple retrieval. For example, we show that disabling a single Gather or Aggregate Head in a pruned Llama-3.1-8B impairs retrieving the correct answer letter in MMLU, reducing its accuracy from 66% to 25% (random guessing). Moreover, this retrieval bottleneck can obscure limited knowledge demands of tasks as the pruned model succeeds on MMLU with functioning G&A heads yet fails on other knowledge benchmarks. The bottleneck similarly extends to tasks where SSMs typically underperform, such as GSM8K, BBH, and dialogue comprehension. We show that SSMs' retrieval challenges manifest in these heads, creating smoother attention patterns instead of the sharp token transitions effective G&A requires. Thus, the Transformer-SSM retrieval gap exists in just a few heads, rather than the entire language model. This suggests a unified explanation for Transformer vs. SSM performance gap while showing how to merge their strengths. We find that pretrained hybrid models, where SSMs are combined with a few attention layers, delegate the role of Aggregate Heads to attention. Similarly, replacing a single G&A head in a pretrained SSM with an attention variant boosts retrieval and benchmark scores.
Key facts. Cited by 15 papers in the map (data through 14 Oct 2026). Builds on 46 of its 54 references that are in the map. First posted on arXiv 22 Apr 2025.
Builds on
46 of 54 references.
References · 46
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
- 2022Locating and Editing Factual Associations in GPTcited by 2,691
- 2023Efficient Streaming Language Models with Attention Sinkscited by 2,186
- 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
- 2022Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Themcited by 1,807
34 more
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
- 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 992
- 2022In-context Learning and Induction Headscited by 964
- 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 613
- 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 532
- 2023Gated Linear Attention Transformers with Hardware-Efficient Trainingcited by 514
- 2024Retrieval Head Mechanistically Explains Long-Context Factualitycited by 232
- 2024Repeat After Me: Transformers are Better than State Space Models at Copyingcited by 216
- 2024Simple linear attention language models balance the recall-throughput tradeoffcited by 208
- 2023Zoology: Measuring and Improving Recall in Efficient Language Modelscited by 204
- 2024An Empirical Study of Mamba-based Language Modelscited by 186
- 2024Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modelingcited by 174
- 2023Birth of a Transformer: A Memory Viewpointcited by 161
- 2023Does Circuit Analysis Interpretability Scale? Evidence from Multiple Choice Capabilities in Chinchillacited by 149
- 2024When Benchmarks are Targets: Revealing the Sensitivity of Large Language Model Leaderboardscited by 137
- 2024HGRN2: Gated Linear RNNs with State Expansioncited by 127
- 2024Zamba: A Compact 7B SSM Hybrid Modelcited by 124
- 2024A Practical Review of Mechanistic Interpretability for Transformer-Based Language Modelscited by 118
- 2023Circuit Component Reuse Across Tasks in Transformer Language Modelscited by 108
- 2024Can Mamba Learn How to Learn? A Comparative Study on In-Context Learning Taskscited by 105
- 2024The Hidden Attention of Mamba Modelscited by 91
- 2024Gated Slot Attention for Efficient Linear-Time Sequence Modelingcited by 76
- 2024Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Modelscited by 67
- 2024Attention Heads of Large Language Models: A Surveycited by 67
- 2024RNNs are not Transformers (Yet): The Key Bottleneck on In-context Retrievalcited by 64
- 2024Falcon Mamba: The First Competitive Attention-free 7B Language Modelcited by 46
- 2024Interpreting and Improving Large Language Models in Arithmetic Calculationcited by 45
- 2024MMLU-CF: A Contamination-free Multi-task Language Understanding Benchmarkcited by 41
- 2024Answer, Assemble, Ace: Understanding How LMs Answer Multiple Choice Questionscited by 35
- 2025Llamba: Scaling Distilled Recurrent Models for Efficient Language Processingcited by 35
- 2025Thinking Slow, Fast: Scaling Inference Compute with Distilled Reasonerscited by 22
- 2024Correcting Negative Bias in Large Language Models through Negative Attention Score Alignmentcited by 12
- 2024MMLU-SR: A Benchmark for Stress-Testing Reasoning Capability of Large Language Modelscited by 2
- 2024Birdie: Advancing State Space Models with Reward-Driven Objectives and Curriculacited by 1
Built on by
Citing papers per year: 2025: 4; 2026: 11.
15 papers.
Sort:
By area:
- 2025Demystifying the Roles of LLM Layers in Retrieval, Knowledge, and Reasoningcited by 15
- 2025Mechanistic evaluation of Transformers and state space modelscited by 9
- 2026Retrieval-Aware Distillation for Transformer-SSM Hybridscited by 6
- 2026Raven: High-Recall Sequence Modeling with Sparse Memory Routingcited by 6
- 2025Local Linear Attention: An Optimal Interpolation of Linear and Softmax Attention For Test-Time Regressioncited by 4
- 2025StateX: Enhancing RNN Recall via Post-training State Expansioncited by 1
- 2026Hybrid Associative Memoriescited by 1
- 2026When Perplexity Lies: Generation-Focused Distillation of Hybrid Sequence Modelscited by 1
- 2026Parallax: Parameterized Local Linear Attention for Language Modelingcited by 1
- 2026Blurry Window Attentioncited by 1
- 2026On the Recall Scaling Laws in Mamba: A Theoretical and Mechanistic Study via Hashingcited by 1
- 2026Refit the Probe: Single-Direction Ablation Is Not a Necessity Testcited by 0
- 2026The Token Before the Value Is the Key: How Hybrid Architectures Organize Induction Circuits14 Sep 2026
- 2026On the Recall Scaling Laws in Mamba: A Theoretical and Mechanistic Study via Hashing7 Sep 2026
- 2026Raven: High-Recall Sequence Modeling with Sparse Memory Routing28 Jul 2026
- 2026Mechanistic Attention Guidance for Agent Memory Refinement20 Jul 2026
- 2026Blurry Window Attention31 May 2026
- 2026Refit the Probe: Single-Direction Ablation Is Not a Necessity Test30 May 2026
- 2026Contribution Weights: A Geometrical Analysis of Self-Attention Transformers29 May 2026
- 2026Parallax: Parameterized Local Linear Attention for Language Modeling27 May 2026
- 2026When Perplexity Lies: Generation-Focused Distillation of Hybrid Sequence Models27 Mar 2026
- 2026Hybrid Associative Memories20 Mar 2026
- 2026Retrieval-Aware Distillation for Transformer-SSM Hybrids11 Feb 2026
- 2025Demystifying the Roles of LLM Layers in Retrieval, Knowledge, and Reasoning2 Oct 2025
- 2026Blurry Window Attentioncited by 1
- 2026Blurry Window Attention31 May 2026
- 2026Retrieval-Aware Distillation for Transformer-SSM Hybridscited by 6
- 2026Raven: High-Recall Sequence Modeling with Sparse Memory Routingcited by 6
- 2025Local Linear Attention: An Optimal Interpolation of Linear and Softmax Attention For Test-Time Regressioncited by 4
- 2025StateX: Enhancing RNN Recall via Post-training State Expansioncited by 1
- 2026Hybrid Associative Memoriescited by 1
- 2026When Perplexity Lies: Generation-Focused Distillation of Hybrid Sequence Modelscited by 1
- 2026Parallax: Parameterized Local Linear Attention for Language Modelingcited by 1
- 2026Refit the Probe: Single-Direction Ablation Is Not a Necessity Testcited by 0
- 2026Contribution Weights: A Geometrical Analysis of Self-Attention Transformerscited by 0
- 2026Mechanistic Attention Guidance for Agent Memory Refinementcited by 0
- 2026The Token Before the Value Is the Key: How Hybrid Architectures Organize Induction Circuitscited by 0
- 2026The Token Before the Value Is the Key: How Hybrid Architectures Organize Induction Circuits14 Sep 2026
- 2026Raven: High-Recall Sequence Modeling with Sparse Memory Routing28 Jul 2026
- 2026Mechanistic Attention Guidance for Agent Memory Refinement20 Jul 2026
- 2026Refit the Probe: Single-Direction Ablation Is Not a Necessity Test30 May 2026
- 2026Contribution Weights: A Geometrical Analysis of Self-Attention Transformers29 May 2026
- 2026Parallax: Parameterized Local Linear Attention for Language Modeling27 May 2026
- 2026When Perplexity Lies: Generation-Focused Distillation of Hybrid Sequence Models27 Mar 2026
- 2026Hybrid Associative Memories20 Mar 2026
- 2026Retrieval-Aware Distillation for Transformer-SSM Hybrids11 Feb 2026
- 2025Local Linear Attention: An Optimal Interpolation of Linear and Softmax Attention For Test-Time Regression1 Oct 2025
- 2025StateX: Enhancing RNN Recall via Post-training State Expansion26 Sep 2025
Similar, not cited
Close in wording, no citation either way
- 2026From Interpretability to Performance: Optimizing Retrieval Heads for Long-Context Language Models
- 2026Retrievit: In-context Retrieval Capabilities of Transformers, State Space Models, and Hybrid Architectures
- 2025Beyond Semantics: How Temporal Biases Shape Retrieval in Transformer and State-Space Models
- 2025A Survey of Context Engineering for Large Language Models
- 2024State Space Models are Strong Text Rerankers
- 2025Overcoming Long-Context Limitations of State-Space Models via Context-Dependent Sparse Attention
- 2026Structured Memory for Edge Language Models: Persistent Context and Corpus Retrieval via O(1) SSM State Injection
- 2024On the Structural Memory of LLM Agents