Merging Memory and Space: A State Space Neural Operator
About
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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Operator learning | 1D Kuramoto-Sivashinsky ν = 0.075 (test) | Time (ms)0.8 | 25 | |
| Forecasting | Spherical Shallow Water Equations (SWE) long time horizons | Relative L2 Error1.28 | 6 | |
| Solving 2D Navier-Stokes equations | TorusLi | Relative L2 Error3.45 | 6 | |
| Solving 2D Navier-Stokes equations | TorusVis | Relative L2 Error2.18 | 6 | |
| Solving 2D Navier-Stokes equations | TorusVisForce | Relative L2 Error0.0263 | 6 | |
| Solving Compressible Euler equations | CE-RM | Relative L2 Error0.0583 | 6 | |
| Solving Compressible Euler equations | GCE-RT | Relative L2 Error1.38 | 6 | |
| Data-Driven Aerodynamics | AirfRANS | Volume Error0.0017 | 2 |