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.

Water Resources Quality Indicators Monitoring by Nonlinear Programming and Simulated Annealing Optimization with Ensemble Learning Approaches

Comparative Assessment of Individual and Ensemble Machine Learning Models for Efficient Analysis of River Water Quality

Incorporation of Water Quality Index Models with Machine Learning-Based Techniques for Real-Time Assessment of Aquatic Ecosystems

Monitoring the Industrial waste polluted stream - Integrated analytics and machine learning for water quality index assessment

Study of water resources parameters using artificial intelligence techniques and learning algorithms: a survey

Evaluation of water quality indexes with novel machine learning and SHapley Additive ExPlanation (SHAP) approaches

Optimization of water quality index models using machine learning approaches

Enhancing Water Quality Monitoring: an Integrated Machine Learning Framework with Physical Constraints for Imputation and Time Series Downscaling

A novel interval decomposition correlation particle swarm optimization-extreme learning machine model for short-term and long-term water quality prediction

Enhancing groundwater quality prediction through ensemble machine learning techniques

A Hybrid Evolutionary Data Driven Model for River Water Quality Early Warning.

Reservoir water quality simulation with data mining models

Comparative analysis of ensemble learning algorithms in water quality prediction

Developing an ensembled machine learning model for predicting water quality index in Johor River Basin

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

Modeling of wetlands storage instabilities using an optimized slffn machine learning using evolutionary computation considering the RROC analysis and PDF techniques

Evaluation of total dissolved solids in rivers by improved neuro fuzzy approaches using metaheuristic algorithms

A stacking ANN ensemble model of ML models for stream water quality prediction of Godavari River Basin, India

Mapping reservoir water quality from Sentinel-2 satellite data based on a new approach of weighted averaging: Application of Bayesian maximum entropy

Innovative interpretable AI-guided water quality evaluation with risk adversarial analysis in river streams considering spatial-temporal effects

Forecasting water quality variable using deep learning and weighted averaging ensemble models