Merging Memory and Space: A State Space Neural Operator
Not on the map. Why?
Abstract (arXiv)
We propose the *State Space Neural Operator* (SS-NO), a compact architecture for learning solution operators of time-dependent partial differential equations (PDEs). Our formulation extends structured state space models (SSMs) to joint spatiotemporal modeling, introducing two key mechanisms: *adaptive damping*, which stabilizes learning by localizing receptive fields, and *learnable frequency modulation*, which enables data-driven spectral selection. These components provide a unified framework for capturing long-range dependencies with parameter efficiency. Theoretically, we establish connections between SSMs and neural operators, proving a universality theorem for convolutional architectures with full field-of-view. Empirically, SS-NO achieves state-of-the-art performance across diverse PDE benchmarks-including 1D Burgers' and Kuramoto-Sivashinsky equations, and 2D Navier-Stokes and compressible Euler flows-while using significantly fewer parameters than competing approaches. A factorized variant of SS-NO further demonstrates scalable performance on challenging 2D problems. Our results highlight the effectiveness of damping and frequency learning in operator modeling, while showing that lightweight factorization provides a complementary path toward efficient large-scale PDE learning.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 21 of its 32 references that are in the map. First posted on arXiv 31 Jul 2025.
Builds on
21 of 32 references.
References · 21
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,908
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
- 2019DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operatorscited by 2,018
- 2024Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Modelcited by 1,056
- 2022On the Parameterization and Initialization of Diagonal State Space Modelscited by 424
- 2022PDEBENCH: An Extensive Benchmark for Scientific Machine Learningcited by 413
- 2021Choose a Transformer: Fourier or Galerkincited by 343
- 2020Model Reduction and Neural Networks for Parametric PDEscited by 258
- 2021Factorized Fourier Neural Operatorscited by 249
- 2022U-NO: U-shaped Neural Operatorscited by 230
- 2023PDE-Refiner: Achieving Accurate Long Rollouts with Neural PDE Solverscited by 216
9 more
- 2023Convolutional Neural Operators for robust and accurate learning of PDEscited by 214
- 2024Poseidon: Efficient Foundation Models for PDEscited by 200
- 2022Towards Multi-spatiotemporal-scale Generalized PDE Modelingcited by 197
- 2023Scalable Transformer for PDE Surrogate Modelingcited by 128
- 2024Operator Learning: Algorithms and Analysiscited by 85
- 2022Lie Point Symmetry Data Augmentation for Neural PDE Solverscited by 77
- 2024On the Benefits of Memory for Modeling Time-Dependent PDEscited by 31
- 2024State-space models are accurate and efficient neural operators for dynamical systemscited by 28
- 2024Mamba Neural Operator: Who Wins? Transformers vs. State-Space Models for PDEscited by 16
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025CFO: Learning Continuous-Time PDE Dynamics via Flow-Matched Neural Operators
- 2026Autoregression-Free Neural Operators for Time-Dependent PDEs
- 2026Learning Physical Operators using Neural Operators
- 2026Oscillatory State-Space Models as Inductive Biases for Physics-Informed Neural PDE Solvers
- 2026Stability and Discretization Error of State Space Model Neural Operators
- 2025Towards a Foundation Model for Partial Differential Equations Across Physics Domains
- 2025Latent Mamba Operator for Partial Differential Equations
- 2026Adaptive Mamba Neural Operators