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.

LSTM-based Modelling for Coagulant Dosage Prediction in Wastewater Treatment Plant

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

Quantitative Modeling and Predictive Analysis of Chemical Oxygen Demand in Wastewater Treatment Systems Utilizing Long Short-Term Memory Neural Network

Enhancing the effluent prediction accuracy with insufficient data based on transfer learning and LSTM algorithm in WWTPs

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

Adaptive prediction for effluent quality of wastewater treatment plant: Improvement with a dual-stage attention-based LSTM network

A Probabilistic Deep Learning Approach to Enhance the Prediction of Wastewater Treatment Plant Effluent Quality under Shocking Load Events

Prediction of Influent Wastewater Quality Based on Wavelet Transform and Residual LSTM

Using LSTM Neural Network Based on Improved PSO and Attention Mechanism for Predicting the Effluent COD in a Wastewater Treatment Plant

A hybrid deep learning approach to improve real-time effluent quality prediction in wastewater treatment plant

Comparative analysis of water quality prediction performance based on LSTM in the Haihe River Basin, China

Machine learning approaches to predict coagulant dosage in water treatment plants

Integrating water quality and streamflow into prediction of chemical dosage in a drinking water treatment plant using machine learning algorithms

Intelligent Aeration Amount Prediction Control for Wastewater Treatment Process Based on Recurrent Neural Network

Water Quality Prediction in Urban Waterways Based on Wavelet Packet Denoising and LSTM

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

Simulate the Forecast Capacity of a Complicated Water Quality Model Using the Long Short-Term Memory Approach

A deep learning based dynamic COD prediction model for urban sewage

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

Accurate prediction of water quality in urban drainage network with integrated EMD-LSTM model

River Water Quality Parameters Prediction Method Based on LSTM-RNN Model