To Infinity and Beyond: Tool-Use Unlocks Length Generalization in State Space Models
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
- 2017Attention Is All You Needcited by 60,361
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,695
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,411
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
- 2024SWE-agent: Agent-Computer Interfaces Enable Automated Software Engineeringcited by 1,737
38 more
- 2020Rethinking Attention with Performerscited by 1,601
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
- 2014On the Properties of Neural Machine Translation: Encoder-Decoder Approachescited by 1,304
- 2021Show Your Work: Scratchpads for Intermediate Computation with Language Modelscited by 886
- 2014Neural Turing Machinescited by 644
- 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 613
- 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 532
- 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Modelscited by 499
- 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 396
- 2023The Impact of Positional Encoding on Length Generalization in Transformerscited by 327
- 2025The Illusion of Thinking: Understanding the Strengths and Limitations of Reasoning Models via the Lens of Problem Complexitycited by 302
- 2025SWE-smith: Scaling Data for Software Engineering Agentscited by 285
- 2022Neural Networks and the Chomsky Hierarchycited by 217
- 2024Repeat After Me: Transformers are Better than State Space Models at Copyingcited by 216
- 2025Large Language Model Agent: A Survey on Methodology, Applications and Challengescited by 205
- 2025Survey on Evaluation of LLM-based Agentscited by 204
- 2024OpenMathInstruct-2: Accelerating AI for Math with Massive Open-Source Instruction Datacited by 200
- 2023What Algorithms can Transformers Learn? A Study in Length Generalizationcited by 199
- 2024An Empirical Study of Mamba-based Language Modelscited by 186
- 2024OpenMathInstruct-1: A 1.8 Million Math Instruction Tuning Datasetcited by 143
- 2021Investigating the Limitations of Transformers with Simple Arithmetic Taskscited by 139
- 2023Teaching Arithmetic to Small Transformerscited by 123
- 2015Neural GPUs Learn Algorithmscited by 121
- 2023Randomized Positional Encodings Boost Length Generalization of Transformerscited by 117
- 2022Block-Recurrent Transformerscited by 110
- 2024Can Mamba Learn How to Learn? A Comparative Study on In-Context Learning Taskscited by 105
- 2025Nemotron-H: A Family of Accurate and Efficient Hybrid Mamba-Transformer Modelscited by 100
- 2024Looped Transformers for Length Generalizationcited by 99
- 2024In-Context Language Learning: Architectures and Algorithmscited by 89
- 2024Transformers Can Achieve Length Generalization But Not Robustlycited by 86
- 2024Transformers Can Do Arithmetic with the Right Embeddingscited by 78
- 2015Inferring Algorithmic Patterns with Stack-Augmented Recurrent Netscited by 76
- 2021Making Transformers Solve Compositional Taskscited by 72
- 2022Simplicity Bias in Transformers and their Ability to Learn Sparse Boolean Functionscited by 70
- 2023Auto-Regressive Next-Token Predictors are Universal Learnerscited by 67
- 2023Length Generalization in Arithmetic Transformerscited by 66
- 2024DeciMamba: Exploring the Length Extrapolation Potential of Mambacited by 49
- 2024A Formal Framework for Understanding Length Generalization in Transformerscited by 45
50 most cited of 59 All 59 (JSON)
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026On the Spatiotemporal Dynamics of Generalization in Neural Networkscited by 3
- 2026Planning with Transformers: Chain of Computation and Structured Context Windowscited by 0
- 2026On the Importance of Gating: Memorization vs. In-Context Learning in State Space Modelscited by 0
- 2026Length Generalization Needs Proper Regularizationcited by 0
- 2026Length Generalization Needs Proper Regularization3 Oct 2026
- 2026On the Importance of Gating: Memorization vs. In-Context Learning in State Space Models15 Sep 2026
- 2026Planning with Transformers: Chain of Computation and Structured Context Windows20 Jul 2026
- 2026On the Spatiotemporal Dynamics of Generalization in Neural Networks2 Feb 2026
Similar, not cited
Close in wording, no citation either way
- 2024Limits of Deep Learning: Sequence Modeling through the Lens of Complexity Theory
- 2024On the Expressiveness and Length Generalization of Selective State-Space Models on Regular Languages
- 2026When Does Tool Use Increase the Expressive Power of Finite-Precision Recurrent Models?
- 2026Towards Understanding What State Space Models Learn About Code
- 2024LongSSM: On the Length Extension of State-space Models in Language Modelling
- 2024Longhorn: State Space Models are Amortized Online Learners
- 2025The Imitation Game: Turing Machine Imitator is Length Generalizable Reasoner
- 2026Where Induction Runs Out: Description-Length Difficulty and the Memorisation Gap in Integer-Sequence Benchmarks