Physics-informed neural network for diffusive wave model

Abstract

The diffusive wave model (DWM), a nonlinear second-order simplified form of the shallow water equation, has been widely used in hydraulic, hydrologic and irrigation engineering. Solving the forward problem of the DWM can be utilized to predict evolution in water levels and discharge. Solving its inverse problem allows for the identification of crucial parameters (such as Manning's coefficient, rainfall intensity, etc.) based on observations. This paper applies the physics-informed neural network (PINN) with novel improvements to solve the DWM for both forward and inverse problems. In the forward problem, compared to traditional numerical methods, PINN can predict the evolution at any location. In the inverse problem, PINN provides a simple and efficient solution process. In order to overcome the gradient explosion in the training process caused by the characteristics of the DWM, the stop-gradient technique was adopted to train the neural network. To improve the estimation of DWM parameters, the concept of time division was developed, and a new network structure was then proposed. To verify the effectiveness of PINN and its improved algorithm for DWM, seven examples were simulated. The PINN solutions for forward problems were compared with the results obtained by classical numerical methods, while the correct rainfall pattern was learnt for the inverse problem.