A multi-parameter prediction algorithm for water quality based on spatial-temporal graph convolutional networks
CSTR:
Author:
Affiliation:

1. School of Microelectronics and Communication Engineering; Chongqing University, , Chongqing 400044, P. R. China;2. Key Laboratory of Optoelectronic Technology and Systems of Ministry of Education, Chongqing University, , Chongqing 400044, P. R. China;3. School of Artificial Intelligence, Chongqing University of Technology, Chongqing 400054, P. R. China

Clc Number:

TP399;X832

Fund Project:

Supported by the Fundamental Research Funds for the Central Universities (2023CDJKYJH085),the Science and Technology Planning Project of Jiulongpo District, Chongqing (2023TJ2001), and the Chongqing Innovative Talent Training Program for Adolescents (CY240903).

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    Accurate water quality prediction is essential for effective ecological water management. However, water quality exhibits complex non-stationary dynamics and multi-dimensional nonlinear relationships driven by temporal evolution and environmental variability. In multi-parameter river water quality prediction, intricate spatial-temporal dependencies make it difficult for traditional models to effectively integrate dynamic topology and long-period features. To address this challenge, we propose a spatial-temporal graph convolutional network (STGCN) model. In the temporal dimension, a masked sub-series transformer module is employed to extract long-term trends through self-supervised pretraining. Combined with dilated causal convolution, it captures cumulative water quality effects and alleviates the response lag common in traditional models when facing abrupt changes. In the spatial dimension, a dynamic graph learning module integrates a predefined station-distance adjacency matrix with a dynamic residual map to generate adaptive graph structures. Experimental results demonstrate that the proposed model outperforms existing methods in water quality prediction, achieving an R2 greater than 0.93 across all water quality indicators.

    Reference
    Related
    Cited by
Get Citation

叶彬强,曹雪杰,李东,陈昶宏,刘宏,汤斌,冯鹏.基于时空图卷积网络的水质多参数预测算法[J].重庆大学学报,2025,48(11):92~105

Copy
Related Videos

Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:May 27,2025
  • Revised:
  • Adopted:
  • Online: December 15,2025
  • Published:
Article QR Code