mlsanityData to 14 Oct 2026

Rethinking Selectivity in State Space Models: A Minimal Predictive Sufficiency Approach

arXiv 2508.03158 ↗First posted 5 Aug 2025cs.LG cs.IT math.ITcited by 2

On the map: State-space sequence models (S4, Mamba) → Core SSM architectures for language

Abstract (arXiv)

State Space Models (SSMs), particularly recent selective variants like Mamba, have emerged as a leading architecture for sequence modeling, challenging the dominance of Transformers. However, the success of these state-of-the-art models largely relies on heuristically designed selective mechanisms, which lack a rigorous first-principle derivation. This theoretical gap raises questions about their optimality and robustness against spurious correlations. To address this, we introduce the Principle of Predictive Sufficiency, a novel information-theoretic criterion stipulating that an ideal hidden state should be a minimal sufficient statistic of the past for predicting the future. Based on this principle, we propose the Minimal Predictive Sufficiency State Space Model (MPS-SSM), a new framework where the selective mechanism is guided by optimizing an objective function derived from our principle. This approach encourages the model to maximally compress historical information without losing predictive power, thereby learning to ignore non-causal noise and spurious patterns. Extensive experiments on a wide range of benchmark datasets demonstrate that MPS-SSM not only achieves state-of-the-art performance, significantly outperforming existing models in long-term forecasting and noisy scenarios, but also exhibits superior robustness. Furthermore, we show that the MPS principle can be extended as a general regularization framework to enhance other popular architectures, highlighting its broad potential.

Key facts. Cited by 2 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 5 of its 16 references that are in the map. First posted on arXiv 5 Aug 2025.

Builds on

5 of 16 references.

Closest ancestors · 2

Same direction, newest first

  1. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spaces1 Dec
  2. 2021Efficiently Modeling Long Sequences with Structured State Spaces31 Oct

Other references · 3

  1. 2017Attention Is All You Needcited by 60,361
  2. 2016Deep Variational Information Bottleneckcited by 930
  3. 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 605

Built on by

Cited by year

Citing papers per year: 2025: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2024Theoretical Foundations of Deep Selective State-Space Models
  2. 2024Mathematical Formalism for Memory Compression in Selective State Space Models
  3. 2025SeRpEnt: Selective Resampling for Expressive State Space Models
  4. 2025KOSS: Kalman-Optimal Selective State Spaces for Long-Term Sequence Modeling
  5. 2024HiPPO-Prophecy: State-Space Models can Provably Learn Dynamical Systems in Context
  6. 2026A Theoretical Analysis of Mamba's Training Dynamics: Filtering Relevant Features for Generalization in State Space Models
  7. 2025Advancing Intelligent Sequence Modeling: Evolution, Trade-offs, and Applications of State-Space Architectures from S4 to Mamba
  8. 2026Mamba-3: Improved Sequence Modeling using State Space Principles