The invention discloses a complex
river network flow prediction method based on a
diffusion space-time diagram neural network guided by
physical information, and relates to the technical field of intelligent water conservancy and hydrological
time sequence prediction. According to the method, a
diffusion space-time diagram neural network (PI-
Diffusion STG) guided by
physical information is constructed aiming at the problems of lack of physical consistency and insufficient long-
distance space-time dependence capture capability of an existing data-driven model. According to the model, a bidirectional graph
diffusion convolution module is designed, forward-flow hydraulic conduction and reverse-flow jacking effects in a
river network are simulated through forward diffusion and backward diffusion respectively, and bidirectional spatial-temporal characteristics are effectively extracted; performing multi-step
traffic prediction by adopting a sequence-to-sequence architecture; meanwhile, a Saint-Venant equation set describing fluid
mechanics is used as a physical constraint regularization term to be deeply fused into a
loss function, and a physical residual error is calculated by utilizing a
hybrid strategy combining automatic differential and finite difference on a graph. According to the method, the prediction precision of the complex
river network under extreme conditions can be remarkably improved, the prediction result is ensured to accord with the law of conservation of
momentum and quality, and the method has relatively high robustness and physical
interpretability.