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.

Development and Evaluation of a Real-Time Forecasting Framework for Daily Water Quality Forecasts for Lake Chaohu to Lead Time of Six Days

Assessing Different Roles of Baseflow and Surface Runoff for Long-Term Streamflow Forecasting in Southeastern China

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

Adaptive Assessment of Reservoir Scheduling to Hydrometeorological Comprehensive Dry and Wet Condition Evolution in a Multi-Reservoir Region of Southeastern China

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

Calibration and quantifying uncertainty of daily water quality forecasts for large lakes with a Bayesian joint probability modelling approach

Long-term Probabilistic Streamflow Forecast Model with “inputs–structure–parameters” Hierarchical Optimization Framework

Assessment of impacts of water transfer on lake flow and water quality in Lake Chaohu using a three-dimensional hydrodynamic-ecological model

Forecasting environmental water availability of lakes using temporal fusion transformer: case studies of China's two largest freshwater lakes

A Framework for Forecasting the Hourly Nodal Water Demand and Improving the Performance of Real-Time Hydraulic Models Considering Model Uncertainty

Spatio‐temporal heterogeneities in water quality and their potential drivers in Lake Chaohu (China) from 2001 to 2017

Predicting Lake Water Quality Responses to Load Reduction: a Three-Dimensional Modeling Approach for Total Maximum Daily Load

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

Predicting abrupt depletion of dissolved oxygen in Chaohu lake using CNN-BiLSTM with improved attention mechanism

Water Quality Forecast for a Gradually Changeable Water Source Based on Time Series Analysis

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

A real-time simulation model of water quality with the impact of internal pollution for water source reservoir

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

CART and PSO+KNN Algorithms to Estimate the Impact of Water Level Change on Water Quality in Poyang Lake, China

Evaluating impacts of climate and management on reservoir water quality using environmental fluid dynamics code

Combining the Swat Model with Sequential Uncertainty Fitting Algorithm for Streamflow Prediction and Uncertainty Analysis for the Lake Dianchi Basin, China