2021

Efficiently Modeling Long Sequences with Structured State Spaces

Gu, Albert, Goel, Karan, Ré, Christopher

Understand

A central goal of sequence modeling is designing a single principled model that can address sequence data across a range of modalities and tasks, particularly on long-range dependencies.

  • Although conventional models including RNNs, CNNs, and Transformers have specialized variants for capturing long dependencies, they still struggle to scale to very long sequences of $10000$ or more steps.
  • A promising recent approach proposed modeling sequences by simulating the fundamental state space model (SSM) \( x'(t) = Ax(t) + Bu(t), y(t) = Cx(t) + Du(t) \), and showed that for appropriate choices of the state matrix \( A \), this system could handle long-range dependencies mathematically and empirically.
  • However, this method has prohibitive computation and memory requirements, rendering it infeasible as a general sequence modeling solution.

Reading the bibliography…