mlsanityData to 14 Oct 2026

SPDEBench: An Extensive Benchmark for Learning Stochastic PDEs

arXiv 2505.18511 ↗First posted 24 May 2025cs.LG math.AP physics.comp-phcited by 1

Not on the map. Why?

Abstract (arXiv)

Stochastic Partial Differential Equations (SPDEs) driven by random noise play a central role in modeling physical processes with rough spatio-temporal dynamics, such as turbulence flows, superconductors, and quantum dynamics. Although machine learning (ML)-based surrogate models have shown promise for efficiently approximating such dynamics, progress remains limited by the lack of a unified benchmark with controlled data generation and comprehensive evaluation. This gap is particularly significant for singular SPDEs, for which benchmark datasets are largely unavailable and reliable simulation requires numerically delicate schemes based on renormalization. Moreover, subtle differences in data-generation procedures, such as noise approximation, basis choice, and the inclusion of renormalization, can significantly affect the resulting datasets and, consequently, model evaluation. We introduce SPDEBench, the first unified benchmark for ML-based SPDE learning. SPDEBench provides ready-to-use datasets for physically and mathematically significant SPDEs on 1-3D domains with periodic or Dirichlet boundary condition. Both regular and singular SPDEs are taken into consideration. SPDEBench also incorporates representative ML baselines in operator learning, together with 7 evaluation metrics, including Sobolev and distributional metrics beyond the standard L2-error. Supported by SPDEBench, we conduct systematic evaluations of model accuracy, robustness, and out-of-distribution generalization under controlled data variations. Our numerical results show that SPDE-aware architectures generally achieve stronger performance than generic operator-learning baselines. These findings establish SPDEBench as a reproducible and extensible resource, paving pathway for principled benchmarking and architecture design for stochastic spatio-temporal dynamics.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 16 of its 46 references that are in the map. First posted on arXiv 24 May 2025.

Builds on

16 of 46 references.

References · 16

  1. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  2. 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,908
  3. 2019DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operatorscited by 2,018
  4. 2017The Deep Ritz method: A deep learning-based numerical algorithm for solving variational problemscited by 583
  5. 2017Solving high-dimensional partial differential equations using deep learningcited by 556
  6. 2020Neural Controlled Differential Equations for Irregular Time Seriescited by 506
  7. 2021Choose a Transformer: Fourier or Galerkincited by 343
  8. 2020Scalable Gradients for Stochastic Differential Equationscited by 271
  9. 2020Neural Rough Differential Equations for Long Time Seriescited by 143
  10. 2022Wavelet neural operator: a neural operator for parametric partial differential equationscited by 78
  11. 2021Sig-Wasserstein GANs for Time Series Generationcited by 60
  12. 2018Signature moments to characterize laws of stochastic processescited by 56
4 more
  1. 2021Neural Stochastic PDEs: Resolution-Invariant Learning of Continuous Spatiotemporal Dynamicscited by 47
  2. 2022Neural Operator with Regularity Structure for Modeling Dynamics Driven by SPDEscited by 11
  3. 2021Feature Engineering with Regularity Structurescited by 7
  4. 2024Solving stochastic partial differential equations using neural networks in the Wiener chaos expansioncited by 6

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.