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.

Medium and short term prediction of reservoir temperature and dissolved oxygen in deep lakes and reservoirs based on GPU and data assimilation

Surface water temperature prediction in large-deep reservoirs using a long short-term memory model

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

Study on Water Quality Prediction of Urban Reservoir by Coupled CEEMDAN Decomposition and LSTM Neural Network Model

PID4LaTe: a physics-informed deep learning model for lake multi-depth temperature prediction

Large-Scale Prediction of Stream Water Quality Using an Interpretable Deep Learning Approach

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

Prediction model of the outflow temperature from stratified reservoir regulated by stratified water intake facility based on machine learning algorithm

A hybrid water quality prediction model based on variational mode decomposition and bidirectional gated recursive unit

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

Intelligent Simulation of Water Temperature Stratification in the Reservoir

Strategies to alleviate shortages of nurses in adult social care.

Improved data assimilation for algal bloom dynamics simulation in the three gorges reservoir using particle filter

A novel hybrid model for water quality prediction based on VMD and IGOA optimized for LSTM

A novel hybrid model for long-term water quality prediction with the 'decomposition–inputs–prediction' hierarchical optimization framework

A Water Quality Prediction Method Based on the Deep LSTM Network Considering Correlation in Smart Mariculture

Lake eutrophication prediction based on improved MIMO-DD-3Q Learning

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

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

Enable high-resolution, real-time ensemble simulation and data assimilation of flood inundation using distributed GPU parallelization

Integrating ICESat-2 laser altimeter observations and hydrological modeling for enhanced prediction of climate-driven lake level change