An Improved PINN Algorithm for Shallow Water Equations Driven by Deep Learning

Abstract

Solving shallow water equations is crucial in science and engineering for understanding and predicting natural phenomena. To address the limitations of Physics-Informed Neural Network (PINN) in solving shallow water equations, we propose an improved PINN algorithm integrated with a deep learning framework. This algorithm introduces a regularization term as a penalty in the loss function, based on the PINN and Long Short-Term Memory (LSTM) models, and incorporates an attention mechanism to solve the original equation across the entire domain. Simulation experiments were conducted on one-dimensional and two-dimensional shallow water equations. The results indicate that, compared to the classical PINN algorithm, the improved algorithm shows significant advantages in handling discontinuities, such as sparse waves, in one-dimensional problems. It accurately captures sparse waves and avoids smoothing effects. In two-dimensional problems, the improved algorithm demonstrates good symmetry and effectively reduces non-physical oscillations. It also shows significant advantages in capturing details and handling complex phenomena, offering higher reliability and accuracy. The improved PINNs algorithm, which combines neural networks with physical mechanisms, can provide robust solutions and effectively avoid some of the shortcomings of classical PINNs methods. It also possesses high resolution and strong generalization capabilities, enabling accurate predictions at any given moment.