Physics-informed neural networks for the shallow-water equations on the sphere

Abstract

We propose the use of physics-informed neural networks for solving the shallow-water equations on the sphere in the meteorological context. Physics-informed neural networks are trained to satisfy the differential equations along with the prescribed initial and boundary data, and thus can be seen as an alternative approach to solving differential equations compared to traditional numerical approaches such as finite difference, finite volume or spectral methods. We discuss the training difficulties of physics-informed neural networks for the shallow-water equations on the sphere and propose a simple multi-model approach to tackle test cases of comparatively long time intervals. Here we train a sequence of neural networks instead of a single neural network for the entire integration interval. We also avoid the use of a boundary value loss by encoding the boundary conditions in a custom neural network layer. We illustrate the abilities of the method by solving the most prominent test cases proposed by Williamson et al. [J. Comput. Phys. 102 (1992), 211-224].

24 pages, 9 figures, 1 tables, minor extensions

References in corpus (18)

• Adam: A Method for Stochastic Optimization
• Convolutional LSTM Network: A Machine Learning Approach for Precipitation Nowcasting
• Playing Atari with Deep Reinforcement Learning
• Language Models are Few-Shot Learners
• NSFnets (Navier-Stokes Flow nets): Physics-informed neural networks for the incompressible Navier-Stokes equations
• Adaptive activation functions accelerate convergence in deep and physics-informed neural networks
• PPINN: Parareal Physics-Informed Neural Network for time-dependent PDEs
• Machine learning in cardiovascular flows modeling: Predicting arterial blood pressure from non-invasive 4D flow MRI data using physics-informed neural networks
• Lookahead Optimizer: k steps forward, 1 step back
• Multi-Task Learning as Multi-Objective Optimization
• Understanding and mitigating gradient pathologies in physics-informed neural networks
• On the Spectral Bias of Neural Networks
• When and why PINNs fail to train: A neural tangent kernel perspective
• Gradient Surgery for Multi-Task Learning
• A primal-dual mimetic finite element scheme for the rotating shallow water equations on polygonal spherical meshes
• Invariant discretization schemes for the shallow-water equations
• Variational integrator for the rotating shallow-water equations on the sphere
• Precipitation nowcasting using a stochastic variational frame predictor with learned prior distribution

Cited by in corpus (7)

• A unified scalable framework for causal sweeping strategies for Physics-Informed Neural Networks (PINNs) and their temporal decompositions
• Lagrangian PINNs: A causality-conforming solution to failure modes of physics-informed neural networks
• Towards physically consistent data-driven weather forecasting: Integrating data assimilation with equivariance-preserving deep spatial transformers
• Deep learning for improved global precipitation in numerical weather prediction systems
• Compatible finite element interpolated neural networks
• Scientific Machine Learning of Flow Resistance Using Universal Shallow Water Equations with Differentiable Programming
• Computing the ensemble spread from deterministic weather predictions using conditional generative adversarial networks