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.

A Stochastic Water Quality Forecasting System for the Yiluo River

Support Vector Machine―an Alternative to Artificial Neuron Network for Water Quality Forecasting in an Agricultural Nonpoint Source Polluted River?

Simulation of Water Quality in a River Network with Time-Varying Lateral Inflows and Pollutants

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

Dynamic pollutant concentration correction method for river sudden pollution

Water Quality Modeling for Load Reduction under Uncertainty: a Bayesian Approach.

A Water Quality Prediction Approach for the Downstream and Delta of Dongjiang River Basin under the Joint Effects of Water Intakes, Pollution Sources, and Climate Change

Spatial-temporal variation analysis and prediction of water quality in the Yellow River basin, China

Stochastic evolution of hydraulic geometry relations in the lower Yellow River of China under environmental uncertainties

Dynamic Water Quality Model for Looping River Network Based on the Framework of Uncertainty Analysis:A Case Study of Wenruitang River Basin

An Evaluation and Prediction Model of Water Quality

A new framework for water quality forecasting coupling causal inference, time-frequency analysis and uncertainty quantification

Water pollution risk simulation and prediction in a drinking water catchment

Simulation of Sudden Water Pollution Accidents in Hunhe River Basin Upstream of Dahuofang Reservoir

Water Quality Evaluation of the Yangtze River in China Using Machine Learning Techniques and Data Monitoring on Different Time Scales

Water quality assessment and pollution source apportionment using multivariate statistical and PMF receptor modeling techniques in a sub-watershed of the upper Yangtze River, Southwest China

Integrated Model of Nutrients for the Miyun Reservoir and Its Watershed

Model-Based Method to Establish Total Pollution Load Control Objective for Looped River Network

Dynamic Simulation of Water Resources in an Urban Wetland Based on Coupled Water Quantity and Water Quality Models

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

SELECTION AND SYSTEMATIC INTEGRATION OF NUMERICAL MODELS FOR WATER QUALITY FORECAST IN THREE GORGES RESERVOIR