On the Expressive Power and Limitations of Multi-Layer SSMs
cs.LG, cs.AI, cs.CC
Submitted: 2026-04-16
Updated: 2026-09-02
Comments: 28 pages, 6 theorems
Journal ref: Transactions on Machine Learning Research (TMLR), 2026
License: http://creativecommons.org/licenses/by/4.0/
The gist: We study how depth, finite precision, state dimension, and chain-of-thought (CoT) affect the expressive power of multi-layer state-space models (SSMs).
Terminology
Abstract
We study how depth, finite precision, state dimension, and chain-of-thought (CoT) affect the expressive power of multi-layer state-space models (SSMs). For the explicit-table K-function-composition problem, a canonical benchmark for sequential information propagation, we prove that any L-layer SSM solving (L+3) -function composition must satisfy d 2p=Ω(N/L 3), where d is the state dimension and p is the per-scalar precision. Conversely, K-function composition is solved exactly by a (K+1) -layer generalized SSM with d=1 and p=Θ(N). This gives a worst-case depth hierarchy for this formal problem family. We then distinguish post-input reasoning, in which all thought tokens are generated after the input, from input-interleaved reasoning, in which thought tokens may be inserted while the input stream is being read. Post-input reasoning does not circumvent our communication-based lower-bound pipeline, whereas input-interleaved reasoning admits bidirectional simulations with general deterministic one-pass streaming algorithms at the granularity of persistent memory. Finally, width and precision are not interchangeable under exact step-preserving simulation in the base affine-state model, but become interchangeable through the streaming-memory characterization once input-interleaved reasoning is allowed.
Sources
- Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models
- S7: Selective and Simplified State Space Layers for Sequence Modeling
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: Local-Consistency Optimisation for Bayesian Neural Networks
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks