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.

Method to predict key factors affecting lake eutrophication – A new approach based on Support Vector Regression model

Comparative Analysis of Machine Learning Methods for Prediction of Chlorophyll-a in a River with Different Hydrology Characteristics: A Case Study in Fuchun River, China

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

Turbidity Prediction of Lake-Type Raw Water Using Random Forest Model Based on Meteorological Data: A Case Study of Tai Lake, China.

A Comparative Study of Statistical Learning Methods to Predict Eutriphication Tendency in a Reservoir, Northeast China

Using Support Vector Regression Coupled with the Genetic Algorithm for Predicting Acute Toxicity to the Fathead Minnow

A hybrid approach of support vector regression with genetic algorithm optimization for aquaculture water quality prediction.

Prediction of chlorophyll a and risk assessment of water blooms in Poyang Lake based on a machine learning method

Identification of driving factors for chlorophyll‐a in multi‐stable shallow lakes of China employing machine learning methods

A Prediction Model for Phytoplankton Abundance Based on Relevance Vector Machine

Early warning of algal blooms based on the optimization support vector machine regression in a typical tributary bay of the Three Gorges Reservoir, China

Prediction and sensitivity analysis of chlorophyll a based on a support vector machine regression algorithm

Source Identification and Prediction of Nitrogen and Phosphorus Pollution of Lake Taihu by an Ensemble Machine Learning Technique

Research on SVR Water Quality Prediction Model Based on Improved Sparrow Search Algorithm

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

Modeling Chlorophyll-A in Taihu Lake with Machine Learning Models

Improved prediction of chlorophyll-a concentrations in reservoirs by GRU neural network based on particle swarm algorithm optimized variational modal decomposition

Water Quality Prediction of Small-Micro Water Body Based on the Intelligent-Algorithm-Optimized Support Vector Machine Regression Method and Unmanned Aerial Vehicles Multispectral Data

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 of River Water Quality Considering Spatiotemporal Correlation and Meteorological Factors.

Examination of Multiple Linear Regression (MLR) and Neural Network (NN) Models to Predict Eutrophication Levels in Lake Champlain