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.

Classification Algorithm of Decision Tree Applied in Water Supply Network

Decision Tree Classification Model in Water Supply Network

Preprocessing Methods for Data of Neural Network Model of Water Distribution System

Daily water supply prediction method based on integrated learning and deep learning

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

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

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

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

A Real-Time Method to Detect the Leakage Location in Urban Water Distribution Networks

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

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

Dealing with Data Missing and Outlier to Calibrate Nodal Water Demands in Water Distribution Systems

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

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

Failure Prediction of Municipal Water Pipes Using Machine Learning Algorithms

The use of decision tree based predictive models for improving the culvert inspection process

Water Pipe Failure Prediction Using AutoML

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

Application Of Modeling And Decision Tree In Railway System

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

Leak Detection in Water Distribution Systems by Classifying Vibration Signals