Abstract:Accurately predicting the state of surface water quality is crucial for ensuring the sustainable use of water resources and environmental protection. This often requires a focus on the range of factors affecting water quality, such as physical and chemical parameters. Tree models, with their flexible tree-like structure and strong capability for partitioning and selecting influential water quality features, offer clear decision-making rules, making them suitable for this task. However, an individual decision tree model has limitations and cannot fully capture the complex relationships between all influencing parameters and water quality. Therefore, this study proposes a method combining ensemble tree models with voting algorithms to predict water quality classification. This study was conducted using five surface water monitoring sites in Qingdao, representing a portion of many municipal water environment monitoring stations in China, employing a single-factor determination method with stringent surface water standards. The soft voting algorithm achieved the highest accuracy of 99.91%, and the model addressed the imbalance in original water quality categories, reaching a Matthews Correlation Coefficient (MCC) of 99.88%. In contrast, conventional machine learning algorithms, such as logistic regression and K-nearest neighbors, achieved lower accuracies of 75.90% and 91.33%, respectively. Additionally, the model's supervision of misclassified data demonstrated its good learning of water quality determination rules. The trained model was also transferred directly to predict water quality at 13 monitoring stations in Beijing, where it performed robustly, achieving an ensemble hard voting accuracy of 97.73% and an MCC of 96.81%. In many countries' water environment systems, different water qualities correspond to different uses, and the magnitude of influencing parameters is directly related to water quality categories; critical parameters can even directly determine the quality category. Tree models are highly capable of handling nonlinear relationships and selecting important water quality features, allowing them to identify and exploit interactions between water quality parameters, which is especially important when multiple parameters together determine the water quality category. Therefore, there is significant motivation to develop tree model-based water quality prediction models.

Decision Tree Classification Model in Water Supply Network

Classification Algorithm of Decision Tree Applied in Water Supply Network

Discussion on Status Modeling Method for Municipal Water Distribution Network

Better Water Demand and Pipe Description Improve the Distribution Network Modeling Results

Demand-Driven Spatiotemporal Variations of Flow Hydraulics and Water Age by Comparative Modeling Analysis of Distribution Network

A Novel Statistical Model for Water Age Estimation in Water Distribution Networks

An Improved Genetic Algorithm For Optimal Layout Of Flow Meters And Valves In Water Network Partitioning

Optimal Operation of Residual Chlorine Concentration in Multi-Source Water Distribution System

Topology-distance-based Clustering Method for Water Distribution Network Partitioning

Rule-based Model for Aging-induced Leakage from Water Supply Pipe Network in Beijing City

Machine learning based water pipe failure prediction: The effects of engineering, geology, climate and socio-economic factors

Pipe Burst Detection and Valve-turnoff Algorithm on Water Supply Pipe Network Based on GIS

Water Pipe Failure Prediction Using AutoML

Leak Detection in Water Distribution Systems by Classifying Vibration Signals

Application Of Modeling And Decision Tree In Railway System

Failure Prediction of Municipal Water Pipes Using Machine Learning Algorithms

Comparison of Statistical and Machine Learning Models for Pipe Failure Modeling in Water Distribution Networks

Contamination Event Detection Method Using Multi-Stations Temporal-Spatial Information Based on Bayesian Network in Water Distribution Systems

Evaluation of Tree-Based Voting Algorithms in Water Quality Classification Prediction

An Integrated Bottom-Up Approach for Leak Detection in Water Distribution Networks Based on Assessing Parameters of Water Balance Model

RS-SVM machine learning approach driven by case data for selecting urban drainage network restoration scheme