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.

Exploring the Application of Artificial Intelligence Technology for Identification of Water Pollution Characteristics and Tracing the Source of Water Quality Pollutants

Identification of Pollution Source and Prediction of Water Quality Based on Deep Learning Techniques

Survey on machine vision-based intelligent water quality monitoring techniques in water treatment plant: Fish activity behavior recognition-based schemes and applications

Identification of River Water Pollution Characteristics Based on Projection Pursuit and Factor Analysis

Spatio-temporal Patterns and Source Apportionment of Pollution in Qiantang River (china) Using Neural-Based Modeling and Multivariate Statistical Techniques

Innovations of Water Pollution Traceability Technology with Artificial Intelligence

Application and innovation of artificial intelligence models in wastewater treatment

Artificial intelligence model for water resources management

Water quality assessment of deep learning-improved comprehensive pollution index: a case study of Dagu River, Jiaozhou Bay, China

Temporal trend and source apportionment of water pollution in different functional zones of Qiantang River, China.

Application of Artificial Intelligence in the Process of Ecological Water Environment Governance and Its Impact on Economic Growth

Identifying Potential Pollution Sources in River Basin Via Water Quality Reasoning Based Expert System

Analysis of Water Quality Monitoring Data Based on LSTM

RESEARCH ON THE WATER POLLUTION MONITORING AND RAPID DECISION-MAKING SYSTEM BASED ON ARTIFICIAL INTELLIGENCE AGENT

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

Artificial Intelligence in Wastewater Treatment: A Data-Driven Analysis of Status and Trends.

Artificial Intelligence-Based Microfluidic Platform for Detecting Contaminants in Water: A Review

A Review on Applications of Artificial Intelligence in Wastewater Treatment

Overview of the Application of Artificial Intelligence in Several Key Issues of Water Conservancy

Water-Quality Assessment and Pollution-Risk Early-Warning System Based on Web Crawler Technology and LSTM

Deep-Learning-Based Water Quality Monitoring and Early Warning Methods: A Case Study of Ammonia Nitrogen Prediction in Rivers