Lag Operator SSMs: A Geometric Framework for Structured State Space Modeling

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Tomonaga, Sutashu, Doya, Kenji, Murata, Noboru
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908727373725696
author Tomonaga, Sutashu
Doya, Kenji
Murata, Noboru
author_facet Tomonaga, Sutashu
Doya, Kenji
Murata, Noboru
contents Structured State Space Models (SSMs), which are at the heart of the recently popular Mamba architecture, are powerful tools for sequence modeling. However, their theoretical foundation relies on a complex, multi-stage process of continuous-time modeling and subsequent discretization, which can obscure intuition. We introduce a direct, first-principles framework for constructing discrete-time SSMs that is both flexible and modular. Our approach is based on a novel lag operator, which geometrically derives the discrete-time recurrence by measuring how the system's basis functions "slide" and change from one timestep to the next. The resulting state matrices are computed via a single inner product involving this operator, offering a modular design space for creating novel SSMs by flexibly combining different basis functions and time-warping schemes. To validate our approach, we demonstrate that a specific instance exactly recovers the recurrence of the influential HiPPO model. Numerical simulations confirm our derivation, providing new theoretical tools for designing flexible and robust sequence models.
format Preprint
id arxiv_https___arxiv_org_abs_2512_18965
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Lag Operator SSMs: A Geometric Framework for Structured State Space Modeling
Tomonaga, Sutashu
Doya, Kenji
Murata, Noboru
Machine Learning
Structured State Space Models (SSMs), which are at the heart of the recently popular Mamba architecture, are powerful tools for sequence modeling. However, their theoretical foundation relies on a complex, multi-stage process of continuous-time modeling and subsequent discretization, which can obscure intuition. We introduce a direct, first-principles framework for constructing discrete-time SSMs that is both flexible and modular. Our approach is based on a novel lag operator, which geometrically derives the discrete-time recurrence by measuring how the system's basis functions "slide" and change from one timestep to the next. The resulting state matrices are computed via a single inner product involving this operator, offering a modular design space for creating novel SSMs by flexibly combining different basis functions and time-warping schemes. To validate our approach, we demonstrate that a specific instance exactly recovers the recurrence of the influential HiPPO model. Numerical simulations confirm our derivation, providing new theoretical tools for designing flexible and robust sequence models.
title Lag Operator SSMs: A Geometric Framework for Structured State Space Modeling
topic Machine Learning
url https://arxiv.org/abs/2512.18965