mlsanityData to 14 Oct 2026

To Infinity and Beyond: Tool-Use Unlocks Length Generalization in State Space Models

arXiv 2510.14826 ↗First posted 16 Oct 2025cs.LGcited by 4

Not on the map. Why?

Abstract (arXiv)

State Space Models (SSMs) have become the leading alternative to Transformers for sequence modeling. Their primary advantage is efficiency in long-context and long-form generation, enabled by fixed-size memory and linear scaling of computational complexity. We begin this work by showing a simple theoretical result stating that SSMs cannot accurately solve any ``truly long-form'' generation problem (in a sense we formally define), undermining their main competitive advantage. However, we show that this limitation can be mitigated by allowing SSMs interactive access to external tools. In fact, we show that given the right choice of tool access and problem-dependent training data, SSMs can learn to solve any tractable problem and generalize to arbitrary problem length/complexity (i.e., achieve length generalization). Following our theoretical finding, we demonstrate that tool-augmented SSMs achieve remarkable length generalization on a variety of arithmetic, reasoning, and coding tasks. These findings highlight SSMs as a potential efficient alternative to Transformers in interactive tool-based and agentic settings.

Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 59 of its 71 references that are in the map. First posted on arXiv 16 Oct 2025.

Builds on

59 of 71 references.

References · 59

  1. 2017Attention Is All You Needcited by 60,361
  2. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
  3. 2021Training Verifiers to Solve Math Word Problemscited by 10,109
  4. 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,695
  5. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
  6. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
  7. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
  8. 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
  9. 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,411
  10. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
  11. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
  12. 2024SWE-agent: Agent-Computer Interfaces Enable Automated Software Engineeringcited by 1,737
38 more
  1. 2020Rethinking Attention with Performerscited by 1,601
  2. 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
  3. 2014On the Properties of Neural Machine Translation: Encoder-Decoder Approachescited by 1,304
  4. 2021Show Your Work: Scratchpads for Intermediate Computation with Language Modelscited by 886
  5. 2014Neural Turing Machinescited by 644
  6. 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 613
  7. 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 532
  8. 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Modelscited by 499
  9. 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 396
  10. 2023The Impact of Positional Encoding on Length Generalization in Transformerscited by 327
  11. 2025The Illusion of Thinking: Understanding the Strengths and Limitations of Reasoning Models via the Lens of Problem Complexitycited by 302
  12. 2025SWE-smith: Scaling Data for Software Engineering Agentscited by 285
  13. 2022Neural Networks and the Chomsky Hierarchycited by 217
  14. 2024Repeat After Me: Transformers are Better than State Space Models at Copyingcited by 216
  15. 2025Large Language Model Agent: A Survey on Methodology, Applications and Challengescited by 205
  16. 2025Survey on Evaluation of LLM-based Agentscited by 204
  17. 2024OpenMathInstruct-2: Accelerating AI for Math with Massive Open-Source Instruction Datacited by 200
  18. 2023What Algorithms can Transformers Learn? A Study in Length Generalizationcited by 199
  19. 2024An Empirical Study of Mamba-based Language Modelscited by 186
  20. 2024OpenMathInstruct-1: A 1.8 Million Math Instruction Tuning Datasetcited by 143
  21. 2021Investigating the Limitations of Transformers with Simple Arithmetic Taskscited by 139
  22. 2023Teaching Arithmetic to Small Transformerscited by 123
  23. 2015Neural GPUs Learn Algorithmscited by 121
  24. 2023Randomized Positional Encodings Boost Length Generalization of Transformerscited by 117
  25. 2022Block-Recurrent Transformerscited by 110
  26. 2024Can Mamba Learn How to Learn? A Comparative Study on In-Context Learning Taskscited by 105
  27. 2025Nemotron-H: A Family of Accurate and Efficient Hybrid Mamba-Transformer Modelscited by 100
  28. 2024Looped Transformers for Length Generalizationcited by 99
  29. 2024In-Context Language Learning: Architectures and Algorithmscited by 89
  30. 2024Transformers Can Achieve Length Generalization But Not Robustlycited by 86
  31. 2024Transformers Can Do Arithmetic with the Right Embeddingscited by 78
  32. 2015Inferring Algorithmic Patterns with Stack-Augmented Recurrent Netscited by 76
  33. 2021Making Transformers Solve Compositional Taskscited by 72
  34. 2022Simplicity Bias in Transformers and their Ability to Learn Sparse Boolean Functionscited by 70
  35. 2023Auto-Regressive Next-Token Predictors are Universal Learnerscited by 67
  36. 2023Length Generalization in Arithmetic Transformerscited by 66
  37. 2024DeciMamba: Exploring the Length Extrapolation Potential of Mambacited by 49
  38. 2024A Formal Framework for Understanding Length Generalization in Transformerscited by 45

50 most cited of 59 All 59 (JSON)

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 4.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2024Limits of Deep Learning: Sequence Modeling through the Lens of Complexity Theory
  2. 2024On the Expressiveness and Length Generalization of Selective State-Space Models on Regular Languages
  3. 2026When Does Tool Use Increase the Expressive Power of Finite-Precision Recurrent Models?
  4. 2026Towards Understanding What State Space Models Learn About Code
  5. 2024LongSSM: On the Length Extension of State-space Models in Language Modelling
  6. 2024Longhorn: State Space Models are Amortized Online Learners
  7. 2025The Imitation Game: Turing Machine Imitator is Length Generalizable Reasoner
  8. 2026Where Induction Runs Out: Description-Length Difficulty and the Memorisation Gap in Integer-Sequence Benchmarks