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.

Physics-guided spatio–temporal neural network for predicting dissolved oxygen concentration in rivers

Artificial Neural Network Modelling of Concentrations of Nitrogen, Phosphorus and Dissolved Oxygen in a Non‐point Source Polluted River in Zhejiang Province, Southeast China

An ensemble model for accurate prediction of key water quality parameters in river based on deep learning methods

Prediction of River Water Quality Considering Spatiotemporal Correlation and Meteorological Factors.

A deep learning model with spatio-temporal graph convolutional networks for river water quality prediction

A novel model for water quality prediction caused by non-point sources pollution based on deep learning and feature extraction methods

Real-time water quality prediction in water distribution networks using graph neural networks with sparse monitoring data

Physics-Guided Recurrent Graph Networks for Predicting Flow and Temperature in River Networks

A spatiotemporal dissolved oxygen prediction model based on graph attention networks suitable for missing data

River Water Quality Parameters Prediction Method Based on LSTM-RNN Model

Water Quality Prediction Based on an Innovated Physical and Data Driving Hybrid Model at Basin Scale

Optimized BP Neural Network for Dissolved Oxygen Prediction

From Hydrometeorology to River Water Quality: Can a Deep Learning Model Predict Dissolved Oxygen at the Continental Scale?

Water Quality Prediction Based on Hybrid Deep Learning Algorithm

Physics-Guided Fair Graph Sampling for Water Temperature Prediction in River Networks

Dissolved Oxygen Prediction Based on PCA-LSTM

Water quality prediction based on sparse dataset using enhanced machine learning

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

Beyond Tides and Time: Machine Learning Triumph in Water Quality

Prediction Of Dissolved Oxygen In Pond Culture Water Based On K-Means Clustering And Gated Recurrent Unit Neural Network

A deep learning interpretable model for river dissolved oxygen multi-step and interval prediction based on multi-source data fusion