English

The Expressive Limits of Diagonal SSMs for State-Tracking

Machine Learning 2026-03-03 v1

Abstract

State-Space Models (SSMs) have recently been shown to achieve strong empirical performance on a variety of long-range sequence modeling tasks while remaining efficient and highly-parallelizable. However, the theoretical understanding of their expressive power remains limited. In this work, we study the expressivity of input-Dependent Complex-valued Diagonal (DCD) SSMs on sequential state-tracking tasks. We show that single-layer DCD SSMs cannot express state-tracking of any non-Abelian group at finite precision. More generally, we show that kk-layer DCD SSMs can express state-tracking of a group if and only if that group has a subnormal series of length kk, with Abelian factors. That is, we identify the precise expressivity range of kk-layer DCD SSMs within the solvable groups. Empirically, we find that multi-layer models often fail to learn state-tracking for non-Abelian groups, highlighting a gap between expressivity and learnability.

Cite

@article{arxiv.2603.01959,
  title  = {The Expressive Limits of Diagonal SSMs for State-Tracking},
  author = {Mehran Shakerinava and Behnoush Khavari and Siamak Ravanbakhsh and Sarath Chandar},
  journal= {arXiv preprint arXiv:2603.01959},
  year   = {2026}
}

Comments

18 pages, 5 figures, 4 tables. Accepted at ICLR 2026

R2 v1 2026-07-01T10:59:22.931Z