On the Expressive Power and Limitations of Multi-Layer SSMs

arXiv:2604.14501 · cs.LG, cs.AI, cs.CC · Submitted 2026-04-16 · Read on arXiv

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

Related papers