Abstract:With the rapid development of economy, the problem of river pollution is becoming more and more serious. It is very important and challenging to build a high-precision water quality prediction model for the comprehensive management of water environment and prevention of water pollution. At present, some achievements have been made in data-driven river water quality prediction, mainly including grey model, time series model, support vector machine model and neural network model. However, these existing prediction methods are limited to single station prediction. The influence of the upstream water quality on the downstream water quality is ignored and the upstream stations at different locations will have different effects on the downstream water quality. In order to solve the problem that the traditional model of water quality prediction does not consider the upstream influence and difficulties in long-term prediction, a new water quality prediction model based on spatiotemporal attention mechanism and long-short-term memory neural network (TS-Attention-LSTM) was proposed, which considers the spatiotemporal correlation and meteorological factors. Firstly, the spatial attention module was embedded in the encoder to extract the significant spatial correlation between upstream and downstream. The interaction among the water quality indicators was also extracted. Then, the temporal attention module was embedded in the decoder to extract the important time series features. Moreover, the meteorological factors and spatial characteristics were fused in the decoder. Finally, the multi-step prediction of water quality was carried out by using LSTM model. In this paper, the research area located in Jinjiang River Watershed in Fujian Province, China. The nonpoint source pollution in this river basin mainly interrelated with livestock discharge, agricultural field runoff and rural domestic sewage. The results show that the TS-Attention-LSTM model can effectively capture the spatial and temporal characteristics of water quality index (such as dissolved oxygen, DO and total phosphorus, TP) and the influence of meteorological factors in Jinjiang River Basin. The mean absolute error (MAE) of the TS-Attention-LSTM model was 0.24, the mean absolute percent error (MAPE) was 3.36%, and the root mean square error (RMSE) was 0.32, which performed best in all comparison models. The determinable coefficient R2 was 0.5062, performed second best in all comparison models.

An ensemble model for accurate prediction of key water quality parameters in river based on deep learning methods

Prediction of River Water Quality Considering Spatiotemporal Correlation and Meteorological Factors.

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

Water quality prediction based on sparse dataset using enhanced machine learning

Which riverine water quality parameters can be predicted by meteorologically-driven deep learning?

Hybrid deep learning based prediction for water quality of plain watershed

Predicting Water Quality Based on EEMD and LSTM Networks

A novel model for water quality prediction caused by non-point sources pollution based on deep learning and feature extraction methods

Long-term prediction of multiple river water quality indexes based on hybrid deep learning models

Physics-guided spatio–temporal neural network for predicting dissolved oxygen concentration in rivers

Large-Scale Prediction of Stream Water Quality Using an Interpretable Deep Learning Approach

Ensemble Empirical Mode Decomposition and a Long Short-Term Memory Neural Network for Surface Water Quality Prediction of the Xiaofu River, China

A Unified Deep Learning Framework for Water Quality Prediction Based on Time-Frequency Feature Extraction and Data Feature Enhancement

W-WaveNet: A multi-site water quality prediction model incorporating adaptive graph convolution and CNN-LSTM

River Water Quality Parameters Prediction Method Based on LSTM-RNN Model

Comparison of Forecasting Models for Real-Time Monitoring of Water Quality Parameters Based on Hybrid Deep Learning Neural Networks

Predict water quality using an improved deep learning method based on spatiotemporal feature correlated: a case study of the Tanghe Reservoir in China

Enhanced predictive modeling of dissolved oxygen concentrations in riverine systems using novel hybrid temporal pattern attention deep neural networks

A Water Quality Prediction Model Based on Multi-Task Deep Learning: A Case Study of the Yellow River, China

Water Quality Prediction Method Based on OVMD and Spatio-Temporal Dependence

A novel hybrid model for water quality prediction based on VMD and IGOA optimized for LSTM