A Neural Operator Emulator for Coastal and Riverine Shallow Water Dynamics
cs.CE, cs.LG, physics.comp-ph, physics.geo-ph
Submitted: 2025-02-20
Updated: 2026-02-02
Journal ref: Journal of Geophysical Research: Machine Learning and Computation, 3 (2026): e2025JH000697
DOI: 10.1029/2025JH000697
Code: https://github.com/erdc/SW
License: http://creativecommons.org/licenses/by/4.0/
The gist: Coastal regions and river floodplains are particularly vulnerable to the impacts of extreme weather events.
Terminology
Abstract
Coastal regions and river floodplains are particularly vulnerable to the impacts of extreme weather events. Accurate real-time forecasting of hydrodynamic processes in these areas is essential for infrastructure planning and climate adaptation. Yet high-fidelity numerical models are often too computationally expensive for real-time use, and lower-cost approaches, such as traditional model order reduction algorithms or conventional neural networks, typically struggle to generalize to out-of-distribution conditions. In this study, we present the Multiple-Input Temporal Operator Network (MITONet), a novel autoregressive neural emulator that employs latent-space operator learning to efficiently approximate high-dimensional numerical solvers for complex, nonlinear problems that are governed by time-dependent, parameterized partial differential equations. We showcase MITONet's predictive capabilities by forecasting regional tide-driven dynamics in the Shinnecock Inlet in New York and riverine flow in a section of the Red River in Louisiana, both described by the two-dimensional shallow-water equations (2D SWE), while incorporating initial conditions, time-varying boundary conditions, and domain parameters such as the bottom friction coefficient. Despite the distinct flow regimes, the complex geometries and meshes, and the wide range of bottom friction coefficients studied, MITONet displays consistently high predictive skill, with anomaly correlation coefficients above 0.9, a maximum normalized root mean square error of 0.011, and computational speedups between 100x-1,250x, even for 175 days of autoregressive rollout forecast from random initial conditions and with unseen parameter values.
Sources
- Universal Physics Transformers: A Framework For Efficiently Scaling Neural Operators
- A Foundation Model for the Earth System
- Message Passing Neural PDE Solvers
- Data-driven reduced order modeling of environmental hydrodynamics using deep autoencoders and neural ODEs
- Accelerating HEC-RAS: A Recurrent Neural Operator for Rapid River Forecasting
- Learning in latent spaces improves the predictive accuracy of deep neural operators
- GraphCast: Learning skillful medium-range global weather forecasting
- Fourier Neural Operator for Parametric Partial Differential Equations
- Neural Operator: Graph Kernel Network for Partial Differential Equations
- Decoupled Weight Decay Regularization
- Real-time Inference and Extrapolation via a Diffusion-inspired Temporal Transformer Operator (DiTTO)
- Nonlinear integro-differential operator regression with neural networks
Related papers
- Constrained Sensing and Reliable State Estimation with Shallow Recurrent Decoders on a TRIGA Mark II Reactor
- Evidence-Unit Fairness and the Limits of Query-Adaptive Sparse-Dense Fusion in Financial Document Retrieval
- Chemical Chain-of-Thought Functions as a Hallucination-Prone Molecular Scratchpad
- Lightweight Adaptation of EEG Foundation Models for Stroke Motor Imagery Decoding: Domain Shift and Subject-Level Robustness
- RetroDFM-R: Reasoning-Driven Retrosynthesis Prediction with Large Language Models via Reinforcement Learning
- Wildfire Suppression: Complexity, Models, and Instances