Spatiotemporal prediction of water quality and ecological risk assessment in a river basin using T-GCN based on river network topology constraints.

Existing water quality prediction methods often ignore the true flow direction and flow variations of river networks and employ static topological structures. This leads to distorted depictions of pollutant migration processes, resulting in significant bias in prediction results. This in turn leads to misjudgments in ecological risk assessments, hindering scientific decision-making for targeted pollution control and risk early warning. To address this issue, this paper proposes a Temporal Graph Convolutional Network (T-GCN) model that incorporates river network topological constraints to improve the accuracy of watershed water quality predictions and the reliability of ecological risk assessment. The model constructs a directed graph based on the river network structure and introduces a flow-driven dynamic connection mechanism to adaptively reflect the impact of changing hydrological conditions on pollutant transport paths and velocities. It captures water quality evolution through joint spatiotemporal modeling and embeds hydrophysical constraints to ensure the rationality of prediction results. Experiments show that T-GCN outperforms spatiotemporal graph comparative model such as DCRNN, Graph WaveNet, and AGCRN in predicting four water quality indicators: DO (Dissolved Oxygen), Ammonia Nitrogen (NH