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.

Enhancing machine learning models for total organic carbon prediction by integrating geospatial parameters in river watersheds

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

Deep learning-based mapping of total suspended solids in rivers across South Korea using high resolution satellite imagery

Multiple remotely sensed datasets and machine learning models to predict chlorophyll-a concentration in the Nakdong River, South Korea

Machine learning regression-based prediction model for the autonomous control of coagulant dosing in smart water purification plants

Beyond Tides and Time: Machine Learning Triumph in Water Quality

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

Machine learning-based prediction of harmful algal blooms in water supply reservoirs

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

Prediction of long-term water quality using machine learning enhanced by Bayesian optimisation.

Classifying eutrophication spatio-temporal dynamics in river systems using deep learning technique

Water Quality Prediction Based on Machine Learning and Comprehensive Weighting Methods

Dynamic patterns and potential drivers of river water quality in a coastal city: Insights from a machine-learning-based framework and water management

Water quality prediction based on sparse dataset using enhanced machine learning

Spatial prediction of groundwater salinity in multiple aquifers of the Mekong Delta region using explainable machine learning models

Interpretable CEEMDAN-FE-LSTM-Transformer hybrid model for predicting total phosphorus concentrations in surface water

Groundwater Potential Mapping Using Remote Sensing and GIS-Based Machine Learning Techniques

Stream water quality prediction using boosted regression tree and random forest models

Machine Learning and Multiple Imputation Approach to Predict Chlorophyll-a Concentration in the Coastal Zone of Korea

An innovative approach for predicting groundwater TDS using optimized ensemble machine learning algorithms at two levels of modeling strategy

Decoding river pollution trends and their landscape determinants in an ecologically fragile karst basin using a machine learning model