Abstract:Effective identification and regulation of water quality impact factors is essential for water resource management and environmental protection. However, the complex coupling of water quality systems poses a significant challenge to this task. This study proposes coherent model for water quality prediction, classification and regulation based on interpretable machine learning. The decomposition-reconstruction module is used to transform non-stationary water quality series into stationary series while effectively reducing the feature dimensions. Spatiotemporal multi-source data is introduced by using the Maximum Information Coefficient (MIC) for feature selection. The Temporal Convolutional Network (TCN) is used to extract the temporal features of different variables, followed by the introduction of External Attention mechanism (EA) to construct the relationship between these features. Finally, the target water quality sequence is simulated using Gated Recurrent Unit (GRU). The proposed model was applied to Poyang Lake in China to predict six water quality indicators: ammonia nitrogen (NH 3 -N), dissolved oxygen (DO), pH, total nitrogen (TN), total phosphorus (TP), water temperature (WT). The water quality was then classified based on the prediction results using the XGBoost algorithm. The findings indicate that the proposed model's Nash-Sutcliff Efficiency (NSE) value ranges from 0.88 to 0.99, surpassing that of the benchmark model, and demonstrates strong interval prediction performance. The results highlight the superior performance of the XGBoost algorithm (with an accuracy of 0.89) in addressing water quality classification issues, particularly in cases of category imbalance. Subsequently, interpretability analysis using the SHapley Additive exPlanation (SHAP) method revealed that the model is capable of learning relationships between different variables and there exists a possibility of learning the physical laws. Ultimately, this study proposes a water quality regulation mechanism that improves TN and DO levels by stepwise changing the magnitude of water temperature, which significantly improves in the case of data limitations. In conclusion, this study presents an overall framework for integrating water quality prediction, classification and improvement for the first time, forming a complete set of water quality early warning and improvement management strategies. This framework provides new ideas and ways for lake water quality management.

Prediction and optimization of wastewater treatment process effluent chemical oxygen demand and energy consumption based on typical ensemble learning models

Prediction of effluent total nitrogen and energy consumption in wastewater treatment plants: Bayesian optimization machine learning methods

Machine learning-based prediction of biological oxygen demand and unit electricity consumption in different-scale wastewater treatment plants

Wastewater treatment process enhancement based on multi-objective optimization and interpretable machine learning

Optimized Deep Learning Models for Effluent Prediction in Wastewater Treatment Processes

Enhancing Real-Time Prediction of Effluent Water Quality of Wastewater Treatment Plant Based on Improved Feedforward Neural Network Coupled with Optimization Algorithm

Prediction of the effluent chemical oxygen demand and volatile fatty acids for anaerobic treatment based on different feature selections machine-learning methods from lab-scale to pilot-scale

A new approach to optimizing aeration using XGB-Bi-LSTM via the online monitoring of oxygen transfer efficiency and oxygen uptake rate

Learning a neural network-based soft sensor with double-errors parallel optimization towards effluent variable prediction in wastewater treatment plants

Hybrid modeling techniques for predicting chemical oxygen demand in wastewater treatment: a stacking ensemble learning approach with neural networks

Modeling Indirect Greenhouse Gas Emissions Sources from Urban Wastewater Treatment Plants: Integrating Machine Learning Models to Compensate for Sparse Parameters with Abundant Observations.

Simulating wastewater treatment plants for heavy metals using machine learning models

Predicting biochemical oxygen demand in wastewater treatment plant using advance extreme learning machine optimized by Bat algorithm

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

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

Dynamic multi-objective optimization control for wastewater treatment process based on modal decomposition and hybrid neural network

Real-time prediction of the chemical oxygen demand component parameters in activated sludge model using backpropagation neural network

Analysis of Machine Learning Models for Wastewater Treatment Plant Sludge Output Prediction

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

Deep learning-based prediction of effluent quality of a constructed wetland

Interpretable prediction, classification and regulation of water quality: A case study of Poyang Lake, China