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.

A multi-source data-driven model of lake water level based on variational modal decomposition and external factors with optimized bi-directional long short-term memory neural network

Application of a Hybrid Model Based on Secondary Decomposition and ELM Neural Network in Water Level Prediction

Study on Wavelet Multi-Scale Analysis and Prediction of Landslide Groundwater

Application of a hybrid model based on GA–ELMAN neural networks and VMD double processing in water level prediction

A Data-Driven Method and Hybrid Deep Learning Model for Flood Risk Prediction

Floodplain Lake Water Level Prediction with Strong River-Lake Interaction Using the Ensemble Learning LightGBM

Lake level dynamics exploration using deep learning, artificial neural network, and multiple linear regression techniques

Hybrid metaheuristic machine learning approach for water level prediction: A case study in Dongting Lake

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

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

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

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

Enhancing long-term river water quality prediction: construction and validation of an improved hybrid model

Human-cognition-inspired Deep Model with Its Application to Ocean Wave Height Forecasting

A secondary modal decomposition ensemble deep learning model for groundwater level prediction using multi-data

An Empirical Modal Decomposition-Improved Whale Optimization Algorithm-Long Short-Term Memory Hybrid Model for Monitoring and Predicting Water Quality Parameters

Flood prediction based on feature selection and a hybrid deep learning network

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

A new stable and interpretable flood forecasting model combining multi-head attention mechanism and multiple linear regression

Combined Wavelet Transform With Long Short-Term Memory Neural Network for Water Table Depth Prediction in Baoding City, North China Plain

Runoff Forecasting using Convolutional Neural Networks and optimized Bi-directional Long Short-term Memory