HiPPO-Prophecy: State-Space Models can Provably Learn Dynamical Systems in Context
Not on the map. Why?
Abstract (arXiv)
This work explores the in-context learning capabilities of State Space Models (SSMs) and presents, to the best of our knowledge, the first theoretical explanation of a possible underlying mechanism. We introduce a novel weight construction for SSMs, enabling them to predict the next state of any dynamical system after observing previous states without parameter fine-tuning. This is accomplished by extending the HiPPO framework to demonstrate that continuous SSMs can approximate the derivative of any input signal. Specifically, we find an explicit weight construction for continuous SSMs and provide an asymptotic error bound on the derivative approximation. The discretization of this continuous SSM subsequently yields a discrete SSM that predicts the next state. Finally, we demonstrate the effectiveness of our parameterization empirically. This work should be an initial step toward understanding how sequence models based on SSMs learn in context.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 15 of its 20 references that are in the map. First posted on arXiv 12 Jul 2024.
Builds on
15 of 20 references.
References · 15
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,424
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 700
- 2022What learning algorithm is in-context learning? Investigations with linear modelscited by 671
- 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 600
- 2022Hungry Hungry Hippos: Towards Language Modeling with State Space Modelscited by 495
- 2023Resurrecting Recurrent Neural Networks for Long Sequencescited by 397
- 2023Larger language models do in-context learning differentlycited by 386
- 2023Transformers as Statisticians: Provable In-Context Learning with In-Context Algorithm Selectioncited by 291
- 2023Transformers learn to implement preconditioned gradient descent for in-context learningcited by 274
- 2024Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Modelscited by 232
- 2023One Step of Gradient Descent is Provably the Optimal In-Context Learner with One Layer of Linear Self-Attentioncited by 168
Built on by
Citing papers per year: 2025: 3.
3 papers.
- 2025From Markov to Laplace: How Mamba In-Context Learns Markov Chainscited by 9
- 2025Advancing Intelligent Sequence Modeling: Evolution, Trade-offs, and Applications of State-Space Architectures from S4 to Mambacited by 9
- 2025How Can Mamba Learn In Context with Outliers and Generalize Provably?cited by 2
- 2025How Can Mamba Learn In Context with Outliers and Generalize Provably?1 Oct 2025
- 2025Advancing Intelligent Sequence Modeling: Evolution, Trade-offs, and Applications of State-Space Architectures from S4 to Mamba22 Mar 2025
- 2025From Markov to Laplace: How Mamba In-Context Learns Markov Chains14 Feb 2025
Similar, not cited
Close in wording, no citation either way
- 2025UnHiPPO: Uncertainty-aware Initialization for State Space Models
- 2024Deep Learning-based Approaches for State Space Models: A Selective Review
- 2022Hidden Parameter Recurrent State Space Models For Changing Dynamics Scenarios
- 2025The Curious Case of In-Training Compression of State Space Models
- 2024Autocorrelation Matters: Understanding the Role of Initialization Schemes for State Space Models
- 2026MaRK: Markov-adapted Recurrent Kernels for Dynamic Operator Conditioning in State Space Models
- 2025Rethinking Selectivity in State Space Models: A Minimal Predictive Sufficiency Approach
- 2025Memory Determines Learning Direction: A Theory of Gradient-Based Optimization in State Space Models