An improved graph convolutional network with feature and temporal attention for multivariate water quality prediction.

Abstract

The analysis and prediction of water quality are of great significance to water quality management and pollution control. In general, current water quality prediction methods are often aimed at single indicator, while the prediction effect is not ideal for multivariate water quality data. At the same time, there may be some correlations between multiple indicators which the conventional prediction models cannot capture. To resolve these problems, this paper proposes a deep learning model: Graph Convolutional Network with Feature and Temporal Attention (FTGCN), realizing the prediction for multivariable water quality data. Firstly, a feature attention mechanism based on multi-head self-attention is designed to capture the potential correlations between water indicators. Then, a temporal prediction module including temporal convolution and bidirectional GRU with a temporal attention mechanism is designed to deal with temporal dependencies of time series. Moreover, an adaptive graph learning mechanism is introduced to extract hidden associations between water quality indicators. An auto-regression module is also added to solve the disadvantage of non-linear nature of neural networks. Finally, an evolutionary algorithm is adopted to optimize the parameters of the proposed model. Our model is applied on four real-world water quality datasets, compared with other models for multivariate time series forecasting. Experimental results demonstrate that the proposed model has a better performance in water quality prediction than others by two indices.