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.

Assessment for Surface Water Quality in Lake Taihu Tiaoxi River Basin China Based on Support Vector Machine

Support Vector Machine―an Alternative to Artificial Neuron Network for Water Quality Forecasting in an Agricultural Nonpoint Source Polluted River?

Turbidity Prediction of Lake-Type Raw Water Using Random Forest Model Based on Meteorological Data: A Case Study of Tai Lake, China.

Research on the Assessment for Air Environment Quality Based on Support Vector Machine

Application Of Least-Squares Support Vector Machines For Quantitative Evaluation Of Known Contaminant In Water Distribution System Using Online Water Quality Parameters

Prediction of dissolved oxygen concentration in hypoxic river systems using support vector machine: a case study of Wen-Rui Tang River, China

Online Classification Of Contaminants Based On Multi-Classification Support Vector Machine Using Conventional Water Quality Sensors

Rapid monitoring of heavy metal pollution in lake water using nitrogen and phosphorus nutrients and physicochemical indicators by support vector machine

A water quality prediction model based on variational mode decomposition and the least squares support vector machine optimized by the sparrow search algorithm (VMD-SSA-LSSVM) of the Yangtze River, China

Assessing and predicting water quality index with key water parameters by machine learning models in coastal cities, China

Prediction of water quality index (WQI) using support vector machine (SVM) and least square-support vector machine (LS-SVM)

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

Application of SVM on Chl-a Concentration Retrievals in Taihu Lake

Water Quality Prediction of Small-Micro Water Body Based on the Intelligent-Algorithm-Optimized Support Vector Machine Regression Method and Unmanned Aerial Vehicles Multispectral Data

Multivariate statistical methods for recognition of water quality feature in Meiliang Bay of Taihu Lake

Water Quality Evaluation of the Yangtze River in China Using Machine Learning Techniques and Data Monitoring on Different Time Scales

Estimating the water quality index based on interpretable machine learning models

A New Approach for Water Quality Assessment Based on Multivariate Statistical Analysis and Radial Basis Function Neural Networks

Modeling Chlorophyll-A in Taihu Lake with Machine Learning Models

Water Quality Evaluation and Pollution Source Apportionment of Surface Water in a Major City in Southeast China Using Multi-Statistical Analyses and Machine Learning Models

Research on the Application of Multimodal-Based Machine Learning Algorithms to Water Quality Classification