English

Generalization Error Analysis for Selective State-Space Models Through the Lens of Attention

Machine Learning 2025-11-05 v3

Abstract

State-space models (SSMs) have recently emerged as a compelling alternative to Transformers for sequence modeling tasks. This paper presents a theoretical generalization analysis of selective SSMs, the core architectural component behind the Mamba model. We derive a novel covering number-based generalization bound for selective SSMs, building upon recent theoretical advances in the analysis of Transformer models. Using this result, we analyze how the spectral abscissa of the continuous-time state matrix influences the model's stability during training and its ability to generalize across sequence lengths. We empirically validate our findings on a synthetic majority task, the IMDb sentiment classification benchmark, and the ListOps task, demonstrating how our theoretical insights translate into practical model behavior.

Keywords

Cite

@article{arxiv.2502.01473,
  title  = {Generalization Error Analysis for Selective State-Space Models Through the Lens of Attention},
  author = {Arya Honarpisheh and Mustafa Bozdag and Octavia Camps and Mario Sznaier},
  journal= {arXiv preprint arXiv:2502.01473},
  year   = {2025}
}

Comments

Accepted at NeurIPS 2025

R2 v1 2026-06-28T21:30:46.916Z