Abstract:Intelligent prediction of water quality plays a pivotal role in water pollution control, water resource protection, emergency decision-making for sudden water pollution incidents, tracking and evaluation of water quality changes in river basins, and is crucial to ensuring water security. The primary methodology employed in this paper for water quality prediction is as follows: (1) utilizing the comprehensive pollution index method and Mann-Kendall (MK) trend analysis method, an assessment is made of the pollution status and change trend within the basin, while simultaneously extracting the principal water quality parameters based on their respective pollution share rates; (2) employing the spearman method, an analysis is conducted to identify the influential factors impacting each key parameter; (3) subsequently, a water quality parameter prediction model, based on Long Short-Term Memory (LSTM) analysis, is constructed using the aforementioned driving factor analysis outcomes. The developed LSTM model in this study showed good prediction performance. The average coefficient of determination (R-2) of the prediction of crucial water quality parameters such as total nitrogen (TN) and dissolved oxygen (DO) reached 0.82 and 0.86 respectively. Additionally, the error analysis of WQI prediction results showed that >75% of the prediction errors were in the range of 0-0.15. The comparative analysis revealed that the LSTM model outperforms both the random forest (RF) model in time series prediction and demonstrates superior robustness and applicability compared to the AutoRegressive Moving Average with eXogenous inputs model (ARMAX). Hence, the model developed in this study offers valuable technical assistance for water quality prediction and early warning systems, particularly in economically disadvantaged regions with limited monitoring capabilities. This contribution facilitates resource optimization and promotes sustainable development.

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

Daily water supply prediction method based on integrated learning and deep learning

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

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

Predicting Water Quality Based on EEMD and LSTM Networks

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

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

Water Quality Prediction Based on an Innovated Physical and Data Driving Hybrid Model at Basin Scale

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

A Method for Predicting Water Quality of River Basin Based on OVMD-GAT-GRU

A Hybrid Model for Water Quality Prediction Based on an Artificial Neural Network, Wavelet Transform, and Long Short-Term Memory

A Novel Multivariate Time Series Prediction of Crucial Water Quality Parameters with Long Short-Term Memory (LSTM) Networks

An Empirical Modal Decomposition-Improved Whale Optimization Algorithm-Long Short-Term Memory Hybrid Model for Monitoring and Predicting Water Quality Parameters

Hybrid Water Quality Prediction with Bidirectional Long Short-Term Memory and Encoder-Decoder

A novel hybrid model for long-term water quality prediction with the 'decomposition–inputs–prediction' hierarchical optimization framework

Water Quality Prediction Based on Hybrid Deep Learning Algorithm

Application of VMD-LSTM in Water Quality Prediction

A Water Quality Prediction Method Based on the Multi-Time Scale Bidirectional Long Short-Term Memory Network

Study on Water Quality Prediction of Urban Reservoir by Coupled CEEMDAN Decomposition and LSTM Neural Network Model

Water quality prediction based on IGRA-ISSA-LSTM model

Application of a Hybrid Model Based on Secondary Decomposition and ELM Neural Network in Water Level Prediction