mlsanityData to 14 Oct 2026

Fast convolution algorithm for state space models

arXiv 2411.17729 ↗First posted 22 Nov 2024math.NA cs.AI cs.NAcited by 1

Not on the map. Why?

Abstract (arXiv)

We present an unconditionally stable algorithm for applying matrix transfer function of a linear time invariant system (LTI) in time domain. The state matrix of an LTI system used for modeling long range dependencies in state space models (SSMs) has eigenvalues close to 1. The standard recursion defining LTI system becomes unstable if the m× m state matrix has just one eigenvalue with absolute value even slightly greater than 1. This may occur when approximating a state matrix by a structured matrix to reduce the cost of matrix-vector multiplication from Oleft(m2right) to Oleft(mright) or Oleft(mlog mright). We introduce an unconditionally stable algorithm that uses an approximation of the rational transfer function in the z-domain by a matrix polynomial of degree 2N+1-1, where N is chosen to achieve any user-selected accuracy. Using a cascade implementation in time domain, applying such transfer function to compute L states requires no more than 2L matrix-vector multiplications (whereas the standard recursion requires L matrix-vector multiplications). However, using unconditionally stable algorithm, it is not necessary to assure that an approximate state matrix has all eigenvalues with absolute values strictly less than 1 i.e., within the desired accuracy, the absolute value of some eigenvalues may possibly exceed 1. Consequently, this algorithm allows one to use a wider variety of structured approximations to reduce the cost of matrix-vector multiplication and we briefly describe several of them to be used for this purpose.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 6 of its 13 references that are in the map. First posted on arXiv 22 Nov 2024.

Builds on

6 of 13 references.

References · 6

  1. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
  2. 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
  3. 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 605
  4. 2022How to Train Your HiPPO: State Space Models with Generalized Orthogonal Basis Projectionscited by 122
  5. 2023Simple Hardware-Efficient Long Convolutions for Sequence Modelingcited by 62
  6. 2023Robustifying State-space Models for Long Sequences via Approximate Diagonalizationcited by 18

Built on by

Cited by year

Citing papers per year: 2025: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2024Efficient State Space Model via Fast Tensor Convolution and Block Diagonalization
  2. 2022On the Parameterization and Initialization of Diagonal State Space Models
  3. 2023Spectral State Space Models
  4. 2024HOPE for a Robust Parameterization of Long-memory State Space Models
  5. 2022Efficient learning of hidden state LTI state space models of unknown order
  6. 2025L2RU: a Structured State Space Model with prescribed L2-bound
  7. 2023FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores
  8. 2024Deep Learning-based Approaches for State Space Models: A Selective Review