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.

Prediction Studies of River Water Quality Based on Moving Average Weighted Markov Model -- A Case Study of Shiwei Port, Jingjiang City

What will the water quality of the Yangtze River be in the future?

Temporal Variations and Trends Prediction of Water Quality During 2010–2019 in the Middle Yangtze River, China

Simulation of Water Quality in a River Network with Time-Varying Lateral Inflows and Pollutants

An Evaluation and Prediction Model of Water Quality

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

A Water Quality Prediction Approach for the Downstream and Delta of Dongjiang River Basin under the Joint Effects of Water Intakes, Pollution Sources, and Climate Change

Assessing and predicting water quality index with key water parameters by machine learning models in coastal cities, China

Three-Phase-Based Approach to Develop a River Health Prediction and Early Warning System to Guide River Management

Spatio-temporal Patterns and Source Apportionment of Pollution in Qiantang River (china) Using Neural-Based Modeling and Multivariate Statistical Techniques

Analysis of Water Quality Prediction in the Yangtze River Delta under the River Chief System

Spatiotemporal analysis and prediction of water quality in Pearl River, China, using multivariate statistical techniques and data-driven model

Integrated prediction of water pollution and risk assessment of water system connectivity based on dynamic model average and model selection criteria

Water quality assessment and pollution source apportionment using multivariate statistical and PMF receptor modeling techniques in a sub-watershed of the upper Yangtze River, Southwest China

Design and Implementation of Water Quality Prediction System of Shaying River

Simulation and Prediction of Sudden Water Pollution in the Hanjiang River to Weihe River Water Transfer Area

Spatial-temporal variation analysis and prediction of water quality in the Yellow River basin, China

Water Quality Evaluation of the Yangtze River in China Using Machine Learning Techniques and Data Monitoring on Different Time Scales

Water Quality Evaluation and Pollution Source Apportionment of Surface Water in a Major City in Southeast China Using Multi-Statistical Analyses and Machine Learning Models

A water quality prediction model based on variational mode decomposition and the least squares support vector machine optimized by the sparrow search algorithm (VMD-SSA-LSSVM) of the Yangtze River, China

Based on Hydroinformatics and Machine Learning: Evolutionary Characteristics of Water Quality in the Yangtze River Basin over the Past Two Decades