mlsanityData to 14 Oct 2026

Mechanistic evaluation of Transformers and state space models

arXiv 2505.15105 ↗First posted 21 May 2025cs.CL cs.AIcited by 9

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

  1. 2024Mimetic Initialization Helps State Space Models Learn to Recall14 Oct
  2. 2024Longhorn: State Space Models are Amortized Online Learners19 Jul
  3. 2024An Empirical Study of Mamba-based Language Models12 Jun
  4. 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Duality31 May
  5. 2024Simple linear attention language models balance the recall-throughput tradeoff28 Feb
  6. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spaces1 Dec

Other references · 44

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020GLU Variants Improve Transformercited by 1,804
  3. 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
  4. 2014Neural Turing Machinescited by 644
  5. 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Modelscited by 499
  6. 2021Causal Abstractions of Neural Networkscited by 422
  7. 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 396
  8. 2023Hyena Hierarchy: Towards Larger Convolutional Language Modelscited by 393
  9. 2020On the Binding Problem in Artificial Neural Networkscited by 233
  10. 2024Repeat After Me: Transformers are Better than State Space Models at Copyingcited by 216
  11. 2023Causal Abstraction: A Theoretical Foundation for Mechanistic Interpretabilitycited by 205
  12. 2023Zoology: Measuring and Improving Recall in Efficient Language Modelscited by 204
32 more
  1. 2016Using Fast Weights to Attend to the Recent Pastcited by 170
  2. 2023Does Circuit Analysis Interpretability Scale? Evidence from Multiple Choice Capabilities in Chinchillacited by 149
  3. 2024Fine-Tuning Enhances Existing Mechanisms: A Case Study on Entity Trackingcited by 149
  4. 2023What Formal Languages Can Transformers Express? A Surveycited by 124
  5. 2023The mechanistic basis of data dependence and abrupt learning in an in-context classification taskcited by 118
  6. 2024The Evolution of Statistical Induction Heads: In-Context Learning Markov Chainscited by 118
  7. 2024In-Context Language Learning: Architectures and Algorithmscited by 89
  8. 2024What needs to go right for an induction head? A mechanistic study of in-context learning circuits and their formationcited by 89
  9. 2023A Theory of Emergent In-Context Learning as Implicit Structure Inductioncited by 86
  10. 2023Physics of Language Models: Part 1, Learning Hierarchical Language Structurescited by 83
  11. 2025Test-time regression: a unifying framework for designing sequence models with associative memorycited by 74
  12. 2024RNNs are not Transformers (Yet): The Key Bottleneck on In-context Retrievalcited by 64
  13. 2024LLM Circuit Analyses Are Consistent Across Training and Scalecited by 63
  14. 2025Which Attention Heads Matter for In-Context Learning?cited by 61
  15. 2024pyvene: A Library for Understanding and Improving PyTorch Models via Interventionscited by 54
  16. 2024A Mechanistic Analysis of a Transformer Trained on a Symbolic Multi-Step Reasoning Taskcited by 49
  17. 2023Entity Tracking in Language Modelscited by 44
  18. 2021Examining the Inductive Bias of Neural Language Models with Artificial Languagescited by 40
  19. 2025Language Models use Lookbacks to Track Beliefscited by 39
  20. 2016Associative Long Short-Term Memorycited by 37
  21. 2024Answer, Assemble, Ace: Understanding How LMs Answer Multiple Choice Questionscited by 35
  22. 2025(How) Do Language Models Track State?cited by 34
  23. 2024A Percolation Model of Emergence: Analyzing Transformers Trained on a Formal Languagecited by 26
  24. 2024gzip Predicts Data-dependent Scaling Lawscited by 16
  25. 2024Associative Recurrent Memory Transformercited by 15
  26. 2025Understanding the Skill Gap in Recurrent Language Models: The Role of the Gather-and-Aggregate Mechanismcited by 15
  27. 2022Benchmarking Compositionality with Formal Languagescited by 12
  28. 2023Focus Your Attention (with Adaptive IIR Filters)cited by 12
  29. 2024On the Power of Convolution Augmented Transformercited by 6
  30. 2017Learning to update Auto-associative Memory in Recurrent Neural Networks for Improving Sequence Memorizationcited by 5
  31. 2024Short-Long Convolutions Help Hardware-Efficient Linear Attention to Focus on Long Sequencescited by 5
  32. 2023A Fast Algorithm for Computing Prefix Probabilitiescited by 4

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 1; 2026: 8.

9 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Overcoming Long-Context Limitations of State-Space Models via Context-Dependent Sparse Attention
  2. 2025Beyond Semantics: How Temporal Biases Shape Retrieval in Transformer and State-Space Models
  3. 2025Understanding and Enhancing Mamba-Transformer Hybrids for Memory Recall and Language Modeling
  4. 2024Understanding and Mitigating Bottlenecks of State Space Models through the Lens of Recency and Over-smoothing
  5. 2026Retrieval-Aware Distillation for Transformer-SSM Hybrids
  6. 2026Retrievit: In-context Retrieval Capabilities of Transformers, State Space Models, and Hybrid Architectures
  7. 2025Characterizing Mamba's Selective Memory using Auto-Encoders
  8. 2025Revisiting associative recall in modern recurrent models