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.

Predicting the Temporal-Spatial Distribution of Chlorophyll-a in the Yellow River Estuary Using Explainable Machine Learning

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

Modeling Chlorophyll $a$ Concentration Affected by Artificial Upwelling in Qiandao Lake

Predicting the Short-Time-scale Variability of Chlorophyll a in the Elbe River Using a Lagrangian-based Multi-Criterion Analog Model

Prediction of Sea Surface Chlorophyll-a Concentrations Based on Deep Learning and Time-Series Remote Sensing Data

A novel method to estimate the 3D chlorophyll a distribution in the South China Sea surface waters using hydrometeorological parameters

Time-Series Forecasting of Chlorophyll-a in Coastal Areas Using LSTM, GRU and Attention-Based RNN Models

Exploring the response and prediction of phytoplankton to environmental factors in eutrophic marine areas using interpretable machine learning methods

Application of Extreme Learning Machine for Predicting Chlorophyll-a Concentration Inartificial Upwelling Processes

Modeling Spatially Anisotropic Nonstationary Processes in Coastal Environments Based on a Directional Geographically Neural Network Weighted Regression.

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

Evaluating Traditional Empirical Models and BPNN Models in Monitoring the Concentrations of Chlorophyll-A and Total Suspended Particulate of Eutrophic and Turbid Waters

Mapping Multi-Depth Soil Salinity Using Remote Sensing-Enabled Machine Learning in the Yellow River Delta, China

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

The effect of red light on the flavonoid composition of etiolated pea plumules.

Prediction and sensitivity analysis of chlorophyll a based on a support vector machine regression algorithm

Incorporating marine particulate carbon into machine learning for accurate estimation of coastal chlorophyll -a

Spatiotemporal distribution and prediction of chlorophyll-a in Ulansuhai lake from an arid area of China

Exploring the Influence Mechanism of Meteorological Conditions on the Concentration of Suspended Solids and Chlorophyll-a in Large Estuaries Based on MODIS Imagery

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

Applying Deep Learning in the Prediction of Chlorophyll-a in the East China Sea