Physics-Informed Neural Networks for Solving the Two-Dimensional Shallow Water Equations with Terrain Topography and Rainfall Source Terms

Abstract: Solving the two-dimensional shallow water equations is a fundamental problem in flood simulation technology. In recent years, physics-informed neural networks (PINNs) have emerged as a novel methodology for addressing this problem. Given their advantages in parallel computing, the potential for data assimilation and parameter calibration, and the rapid advancement of artificial intelligence, it is crucial to investigate both the capabilities and limitations of PINNs. While current research has demonstrated the significant potential of PINNs, many aspects of this new approach remain to be explored. In this study, we employ PINNs enhanced by dimensional transformation and N-LAAF techniques to validate their effectiveness in solving two-dimensional free surface flow with rainfall on terrain topography. The shallow water equations primarily exist in two forms: the variables form and the conservative form. Through theoretical analysis and experimental validation, we demonstrate that a hybrid variable-conservation form offers superior performance. Additionally, we find that incorporating the energy conservation law, specifically the entropy condition, does not yield substantial improvements and may even lead to training failure. Furthermore, we have developed an open-source module on the PINNacle platform for solving shallow water equations using PINNs, which includes over ten case studies and various equation forms, to promote research and application in this field.