mlsanityData to 14 Oct 2026

W4S4: WaLRUS Meets S4 for Long-Range Sequence Modeling

arXiv 2506.07920 ↗First posted 9 Jun 2025cs.LG eess.AS eess.IVcited by 0

Not on the map. Why?

Abstract (arXiv)

State Space Models (SSMs) have emerged as powerful components for sequence modeling, enabling efficient handling of long-range dependencies via linear recurrence and convolutional computation. However, their effectiveness depends heavily on the choice and initialization of the state matrix. In this work, we build on the SaFARi framework and existing WaLRUS SSMs to introduce a new variant, W4S4 (WaLRUS for S4), a new class of SSMs constructed from redundant wavelet frames. WaLRUS admits a stable diagonalization and supports fast kernel computation without requiring low-rank approximations, making it both theoretically grounded and computationally efficient. We show that WaLRUS retains information over long horizons significantly better than HiPPO-based SSMs, both in isolation and when integrated into deep architectures such as S4. Our experiments demonstrate consistent improvements across delay reconstruction tasks, classification benchmarks, and long-range sequence modeling, confirming that high-quality, structured initialization enabled by wavelet-based state dynamic offers substantial advantages over existing alternatives. WaLRUS provides a scalable and versatile foundation for the next generation of deep SSM-based models.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 18 of its 26 references that are in the map. First posted on arXiv 9 Jun 2025.

Builds on

18 of 26 references.

References · 18

  1. 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
  2. 2021Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecastingcited by 1,676
  3. 2017Modeling Long- and Short-Term Temporal Patterns with Deep Neural Networkscited by 879
  4. 2018Speech Commands: A Dataset for Limited-Vocabulary Speech Recognitioncited by 752
  5. 2020Long Range Arena: A Benchmark for Efficient Transformerscited by 700
  6. 2021Combining Recurrent, Convolutional, and Continuous-time Models with Linear State-Space Layerscited by 608
  7. 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 605
  8. 2017Understanding the Effective Receptive Field in Deep Convolutional Neural Networkscited by 553
  9. 2022Transformers in Time Series: A Surveycited by 497
  10. 2022On the Parameterization and Initialization of Diagonal State Space Modelscited by 424
  11. 2021SCINet: Time Series Modeling and Forecasting with Sample Convolution and Interactioncited by 338
  12. 2023TSMixer: An All-MLP Architecture for Time Series Forecastingcited by 240
6 more
  1. 2022How to Train Your HiPPO: State Space Models with Generalized Orthogonal Basis Projectionscited by 122
  2. 2024Mamba-360: Survey of State Space Models as Transformer Alternative for Long Sequence Modelling: Methods, Applications, and Challengescited by 64
  3. 2020Machine Learning Advances for Time Series Forecastingcited by 49
  4. 2019Rethinking Full Connectivity in Recurrent Neural Networkscited by 6
  5. 2020Wearing a MASK: Compressed Representations of Variable-Length Sequences Using Recurrent Neural Tangent Kernelscited by 3
  6. 2025SaFARi: State-Space Models for Frame-Agnostic Representationcited by 3

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025WaLRUS: Wavelets for Long-range Representation Using SSMs
  2. 2022Simplified State Space Layers for Sequence Modeling
  3. 2025MS-SSM: A Multi-Scale State Space Model for Efficient Sequence Modeling
  4. 2025Advancing Intelligent Sequence Modeling: Evolution, Trade-offs, and Applications of State-Space Architectures from S4 to Mamba
  5. 2024Incorporating Exponential Smoothing into MLP: A Simple but Effective Sequence Model
  6. 2024S7: Selective and Simplified State Space Layers for Sequence Modeling
  7. 2026A Simple State Space Model Excels at Multivariate Time Series Classification
  8. 2022Simplifying and Understanding State Space Models with Diagonal Linear RNNs