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.

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

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

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

Novel Simulation of Aqueous Total Nitrogen and Phosphorus Concentrations in Taihu Lake with Machine Learning

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

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

Beyond Tides and Time: Machine Learning Triumph in Water Quality

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

Multi-machine learning methods to predict spatial variation characteristics of total nitrogen at watershed scale: Evidences from the largest watershed (Yangtze River Watershed), Asian

Development of remote sensing algorithm for total phosphorus concentration in eutrophic lakes: Conventional or machine learning?

Pollution loads in the middle-lower Yangtze river by coupling water quality models with machine learning

Performance of ensemble‐learning models for predicting eutrophication in Zhuyi Bay, Three Gorges Reservoir

An ensemble model for accurate prediction of key water quality parameters in river based on deep learning methods

The magnitude and drivers of harmful algal blooms in China’s lakes and reservoirs: A national-scale characterization

Predicting and analyzing the algal population dynamics of a grass-type lake with explainable machine learning

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

Comparative Analysis of Surface Water Quality Prediction Performance and Identification of Key Water Parameters Using Different Machine Learning Models Based on Big Data.

Temporal continuous monitoring of cyanobacterial blooms in Lake Taihu at an hourly scale using machine learning

Leveraging explainable machine learning for enhanced management of lake water quality

Prediction of Total Phosphorus Based on Distance Correlation and Machine Learning Methods—a Case Study of Dongjiang River, China

Spatiotemporal Dynamic Analysis of Eutrophication Status Based on Machine Learning-Based Retrieval Algorithm: Case Study in Liangzi Lake, Hubei, China