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 (4)

• Language Models are Few-Shot Learners
• Understanding and mitigating gradient pathologies in physics-informed neural networks
• When and why PINNs fail to train: A neural tangent kernel perspective
• 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