Juan Huan, Wenjie Liao, Yongchun Zheng, Xiangen Xu, Hao Zhang, Bing Shi

A deep learning model with spatio-temporal graph convolutional networks for river water quality prediction

  • Water Science and Technology
Save to CiteDrive Show PDF

Abstract High-precision water quality prediction plays a vital role in preventing and controlling river pollution. However, river water's highly nonlinear and complex spatio-temporal dependencies pose significant challenges to water quality prediction tasks. In order to capture the spatial and temporal characteristics of water quality data simultaneously, this paper combines deep learning algorithms for river water quality prediction in the river network area of Jiangnan Plain, China. A water quality prediction method based on graph convolutional network (GCN) and long short-term memory neural network (LSTM), namely spatio-temporal graph convolutional network model (ST-GCN), is proposed. Specifically, the spatio-temporal graph is constructed based on the spatio-temporal correlation between river stations, the spatial features in the river network are extracted using GCN, and the temporal correlation of water quality data is obtained by integrating LSTM. The model was evaluated using R2, MAE, and RMSE, and the experimental results were 0.977, 0.238, and 0.291, respectively. Compared with traditional water quality prediction models, the ST-GCN model has significantly improved prediction accuracy, better stability, and generalization ability.

Need a simple solution for managing your BibTeX entries? Explore CiteDrive!

  • Web-based, modern reference management
  • Collaborate and share with fellow researchers
  • Integration with Overleaf
  • Comprehensive BibTeX/BibLaTeX support
  • Save articles and websites directly from your browser
  • Search for new articles from a database of tens of millions of references
Try out CiteDrive

More from our Archive