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.

Dongting Lake Algal Bloom Forecasting: Robustness and Accuracy Analysis of Deep Learning Models

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

Algal bloom forecasting with time-frequency analysis: A hybrid deep learning approach.

Deep-Learning-Based Approach for Prediction of Algal Blooms

Employing hybrid deep learning for near-real-time forecasts of sensor-based algal parameters in a Microcystis bloom-dominated lake

Cyanobacterial blooms prediction in China's large hypereutrophic lakes based on MODIS observations and Bayesian theory

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

Machine Learning in Extreme Value Analysis, an Approach to Detecting Harmful Algal Blooms with Long-Term Multisource Satellite Data.

An Intelligent Early Warning System for Harmful Algal Blooms: Harnessing the Power of Big Data and Deep Learning

Improved Predictive Performance of Cyanobacterial Blooms Using a Hybrid Statistical and Deep-Learning Method

Prediction of Chlorophyll-a as an Indicator of Harmful Algal Blooms using Deep Learning with Bayesian Approximation for Uncertainty Assessment

Deep-learning and data-resampling: A novel approach to predict cyanobacterial alert levels in a reservoir

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

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

Prediction of harmful algal blooms in large water bodies using the combined EFDC and LSTM models

Predicting coastal algal blooms with environmental factors by machine learning methods

Ensemble Machine Learning of Gradient Boosting (XGBoost, LightGBM, CatBoost) and Attention-Based CNN-LSTM for Harmful Algal Blooms Forecasting

A Comparative Analysis of Machine Learning Methods for Algal Bloom Detection Using Remote Sensing Images

Causality Analysis and Prediction of Riverine Algal Blooms by Combining Empirical Dynamic Modeling and Machine Learning Techniques

Inland harmful algal blooms (HABs) modeling using internet of things (IoT) system and deep learning

Deep Learning-Based Water Quality Retrieval in an Impounded Lake Using Landsat 8 Imagery: An Application in Dongping Lake