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.

Miss-gradient boosting regression tree: a novel approach to imputing water treatment data

Multi-step-ahead Traffic Speed Forecasting Using Multi-Output Gradient Boosting Regression Tree

A Boosting-Partial Least Squares Method For Ultraviolet Spectroscopic Analysis Of Water Quality

Basis expansion imputation network for incomplete multivariate sensor data in wastewater treatment system

A novel approach for multivariate time series interval prediction of water quality at wastewater treatment plants

Prediction of microbiological non-compliances using a Boosted Regression Trees model: application on the drinking water distribution system of a whole country

Pre-training enhanced spatio-temporal graph neural network for predicting influent water quality and flow rate of wastewater treatment plant: Improvement of forecast accuracy and analysis of related factors

Imputing missing data in a swat water quality modelling study using statistical methods

Transformer based Multi-output Regression Learning for Wastewater Treatment

Estimating Water Residence Time Distribution in River Networks by Boosted Regression Trees (BRT) Model

Exploring potential machine learning application based on big data for prediction of wastewater quality from different full-scale wastewater treatment plants

Stream water quality prediction using boosted regression tree and random forest models

Enhancing the effluent prediction accuracy with insufficient data based on transfer learning and LSTM algorithm in WWTPs

Beyond Tides and Time: Machine Learning Triumph in Water Quality

Leveraging Multi-Level Correlations for Imputing Monitoring Data in Water Supply Systems Using Graph Signal Sampling Theory

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

Data-driven approach for air pollutant concentrations forecasting: A window-based multi-output GBRT approach

Novel Methods for Imputing Missing Values in Water Level Monitoring Data

A Hybrid Model Based on Variational Mode Decomposition and Gradient Boosting Regression Tree for Monthly Runoff Forecasting

Efficiency Enhancement of A Process-Based Rainfall-Runoff Model Using A New Modified Adaboost.Rt Technique

RegCNN: A Deep Multi-output Regression Method for Wastewater Treatment.