Mechanistic evaluation of Transformers and state space models
On the map: State-space sequence models (S4, Mamba) → Core SSM architectures for language
Abstract (arXiv)
State space models (SSMs) for language modelling promise an efficient and performant alternative to quadratic-attention Transformers, yet show variable performance on recalling basic information from the context. While performance on synthetic tasks like Associative Recall (AR) can point to this deficiency, behavioural metrics provide little information as to textitwhy -- on a mechanistic level -- certain architectures fail and others succeed. To address this, we conduct experiments on AR, and find that only Transformers and Based SSM models fully succeed at AR, with Mamba and DeltaNet close behind, while the other SSMs (H3, Hyena) fail. We then use causal interventions to explain why. We find that Transformers and Based learn to store key-value associations in-context using induction. By contrast, the SSMs seem to compute these associations only at the last state using a single layer. We further investigate the mechanism underlying the success of Mamba, and find novel evidence that Mamba textitdoes implement induction: not via the SSM, but instead via short convolutions. Further experiments on a new hierarchical retrieval task, Associative Treecall (ATR), show that all architectures learn the same mechanism as they did for AR. Furthermore, we show that Mamba can learn Attention-like induction on ATR when short convolutions are removed. These results reveal that architectures with similar accuracy may still have substantive differences, motivating the adoption of mechanistic evaluations.
Key facts. Cited by 9 papers in the map (data through 14 Oct 2026). Placed in Core SSM architectures for language, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 50 of its 59 references that are in the map. First posted on arXiv 21 May 2025.
Builds on
50 of 59 references.
Closest ancestors · 6
Same direction, newest first
- 2024Mimetic Initialization Helps State Space Models Learn to Recall14 Oct
- 2024Longhorn: State Space Models are Amortized Online Learners19 Jul
- 2024An Empirical Study of Mamba-based Language Models12 Jun
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Duality31 May
- 2024Simple linear attention language models balance the recall-throughput tradeoff28 Feb
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spaces1 Dec
Other references · 44
- 2017Attention Is All You Needcited by 60,361
- 2020GLU Variants Improve Transformercited by 1,804
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
- 2014Neural Turing Machinescited by 644
- 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Modelscited by 499
- 2021Causal Abstractions of Neural Networkscited by 422
- 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 396
- 2023Hyena Hierarchy: Towards Larger Convolutional Language Modelscited by 393
- 2020On the Binding Problem in Artificial Neural Networkscited by 233
- 2024Repeat After Me: Transformers are Better than State Space Models at Copyingcited by 216
- 2023Causal Abstraction: A Theoretical Foundation for Mechanistic Interpretabilitycited by 205
- 2023Zoology: Measuring and Improving Recall in Efficient Language Modelscited by 204
32 more
- 2016Using Fast Weights to Attend to the Recent Pastcited by 170
- 2023Does Circuit Analysis Interpretability Scale? Evidence from Multiple Choice Capabilities in Chinchillacited by 149
- 2024Fine-Tuning Enhances Existing Mechanisms: A Case Study on Entity Trackingcited by 149
- 2023What Formal Languages Can Transformers Express? A Surveycited by 124
- 2023The mechanistic basis of data dependence and abrupt learning in an in-context classification taskcited by 118
- 2024The Evolution of Statistical Induction Heads: In-Context Learning Markov Chainscited by 118
- 2024In-Context Language Learning: Architectures and Algorithmscited by 89
- 2024What needs to go right for an induction head? A mechanistic study of in-context learning circuits and their formationcited by 89
- 2023A Theory of Emergent In-Context Learning as Implicit Structure Inductioncited by 86
- 2023Physics of Language Models: Part 1, Learning Hierarchical Language Structurescited by 83
- 2025Test-time regression: a unifying framework for designing sequence models with associative memorycited by 74
- 2024RNNs are not Transformers (Yet): The Key Bottleneck on In-context Retrievalcited by 64
- 2024LLM Circuit Analyses Are Consistent Across Training and Scalecited by 63
- 2025Which Attention Heads Matter for In-Context Learning?cited by 61
- 2024pyvene: A Library for Understanding and Improving PyTorch Models via Interventionscited by 54
- 2024A Mechanistic Analysis of a Transformer Trained on a Symbolic Multi-Step Reasoning Taskcited by 49
- 2023Entity Tracking in Language Modelscited by 44
- 2021Examining the Inductive Bias of Neural Language Models with Artificial Languagescited by 40
- 2025Language Models use Lookbacks to Track Beliefscited by 39
- 2016Associative Long Short-Term Memorycited by 37
- 2024Answer, Assemble, Ace: Understanding How LMs Answer Multiple Choice Questionscited by 35
- 2025(How) Do Language Models Track State?cited by 34
- 2024A Percolation Model of Emergence: Analyzing Transformers Trained on a Formal Languagecited by 26
- 2024gzip Predicts Data-dependent Scaling Lawscited by 16
- 2024Associative Recurrent Memory Transformercited by 15
- 2025Understanding the Skill Gap in Recurrent Language Models: The Role of the Gather-and-Aggregate Mechanismcited by 15
- 2022Benchmarking Compositionality with Formal Languagescited by 12
- 2023Focus Your Attention (with Adaptive IIR Filters)cited by 12
- 2024On the Power of Convolution Augmented Transformercited by 6
- 2017Learning to update Auto-associative Memory in Recurrent Neural Networks for Improving Sequence Memorizationcited by 5
- 2024Short-Long Convolutions Help Hardware-Efficient Linear Attention to Focus on Long Sequencescited by 5
- 2023A Fast Algorithm for Computing Prefix Probabilitiescited by 4
Built on by
Citing papers per year: 2025: 1; 2026: 8.
9 papers.
- 2026Mamba-3: Improved Sequence Modeling using State Space Principlescited by 89
- 2025The Mechanistic Emergence of Symbol Grounding in Language Modelscited by 6
- 2026WriteSAE: Sparse Autoencoders for Recurrent Statecited by 1
- 2026On the Recall Scaling Laws in Mamba: A Theoretical and Mechanistic Study via Hashingcited by 1
- 2026Anatomy of Associative Recall in Fixed-State Recurrences: A Matched-State Decomposition, an Interference Wall, and a Curriculum That Breaks Itcited 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 Circuitscited by 0
- 2026The Key Handoff: Retrieval in Hybrid Language Modelscited by 0
- 2026AMemOps: Evaluating Memory Updates in Long Audio Streamscited by 0
- 2026AMemOps: Evaluating Memory Updates in Long Audio Streams13 Oct 2026
- 2026The Key Handoff: Retrieval in Hybrid Language Models26 Sep 2026
- 2026Anatomy of Associative Recall in Fixed-State Recurrences: A Matched-State Decomposition, an Interference Wall, and a Curriculum That Breaks It14 Sep 2026
- 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
- 2026Refit the Probe: Single-Direction Ablation Is Not a Necessity Test30 May 2026
- 2026WriteSAE: Sparse Autoencoders for Recurrent State12 May 2026
- 2026Mamba-3: Improved Sequence Modeling using State Space Principles16 Mar 2026
- 2025The Mechanistic Emergence of Symbol Grounding in Language Models15 Oct 2025
Similar, not cited
Close in wording, no citation either way
- 2025Overcoming Long-Context Limitations of State-Space Models via Context-Dependent Sparse Attention
- 2025Beyond Semantics: How Temporal Biases Shape Retrieval in Transformer and State-Space Models
- 2025Understanding and Enhancing Mamba-Transformer Hybrids for Memory Recall and Language Modeling
- 2024Understanding and Mitigating Bottlenecks of State Space Models through the Lens of Recency and Over-smoothing
- 2026Retrieval-Aware Distillation for Transformer-SSM Hybrids
- 2026Retrievit: In-context Retrieval Capabilities of Transformers, State Space Models, and Hybrid Architectures
- 2025Characterizing Mamba's Selective Memory using Auto-Encoders
- 2025Revisiting associative recall in modern recurrent models