English

WaLRUS: Wavelets for Long-range Representation Using SSMs

Image and Video Processing 2025-05-20 v1 Machine Learning Systems and Control Audio and Speech Processing Signal Processing Systems and Control

Abstract

State-Space Models (SSMs) have proven to be powerful tools for modeling long-range dependencies in sequential data. While the recent method known as HiPPO has demonstrated strong performance, and formed the basis for machine learning models S4 and Mamba, it remains limited by its reliance on closed-form solutions for a few specific, well-behaved bases. The SaFARi framework generalized this approach, enabling the construction of SSMs from arbitrary frames, including non-orthogonal and redundant ones, thus allowing an infinite diversity of possible "species" within the SSM family. In this paper, we introduce WaLRUS (Wavelets for Long-range Representation Using SSMs), a new implementation of SaFARi built from Daubechies wavelets.

Cite

@article{arxiv.2505.12161,
  title  = {WaLRUS: Wavelets for Long-range Representation Using SSMs},
  author = {Hossein Babaei and Mel White and Sina Alemohammad and Richard G. Baraniuk},
  journal= {arXiv preprint arXiv:2505.12161},
  year   = {2025}
}

Comments

15 pages, 8 figures. Submitted to Neurips 2025

R2 v1 2026-07-01T02:19:00.485Z