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.

Regression Models (SVR, EMD and FastICA) in Forecasting Water Quality of the Haihe River of China

Support Vector Machine―an Alternative to Artificial Neuron Network for Water Quality Forecasting in an Agricultural Nonpoint Source Polluted River?

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

Prediction Modelling Framework Comparative Analysis of Dissolved Oxygen Concentration Variations Using Support Vector Regression Coupled with Multiple Feature Engineering and Optimization Methods: A Case Study in China

Analysis of dissolved oxygen influencing factors and concentration prediction using input variable selection technique: A hybrid machine learning approach

Forecasting model for DO of pond water quality based on PCA-SVR

Prediction of dissolved oxygen concentration in hypoxic river systems using support vector machine: a case study of Wen-Rui Tang River, China

Forecasting DO of the river-type reservoirs using input variable selection and machine learning techniques - taking Shuikou reservoir in the Minjiang River as an example

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

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

Modeling Daily and Monthly Water Quality Indicators in a Canal Using a Hybrid Wavelet-Based Support Vector Regression Structure

Urban river ammonia nitrogen prediction model based on improved whale optimization support vector regression mixed synchronous compression wavelet transform

A hybrid water quality prediction model based on variational mode decomposition and bidirectional gated recursive unit

Improved monthly runoff time series prediction using the SOA–SVM model based on ICEEMDAN–WD decomposition

Forecasting riverine total nitrogen loads using wavelet analysis and support vector regression combination model in an agricultural watershed

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

A deep learning interpretable model for river dissolved oxygen multi-step and interval prediction based on multi-source data fusion

Surface water quality index forecasting using multivariate complementing approach reinforced with locally weighted linear regression model

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

A support vector regression model to predict nitrate-nitrogen isotopic composition using hydro-chemical variables

A Data-Driven Model Based on Fourier Transform and Support Vector Regression for Monthly Reservoir Inflow Forecasting