Related Experiment Video
Updated: Oct 14, 2025

Procedure to Evaluate the Efficiency of Flocculants for the Removal of Dispersed Particles from Plant Extracts
Published on: April 9, 2016
Model construction and application for effluent prediction in wastewater treatment plant: Data processing method
Rui Wang1, Yadan Yu2, Yangwu Chen2
1Faculty of Geosciences and Environmental Engineering, Southwest Jiaotong University, 611756, Chengdu, China; Key Laboratory of Environmental and Applied Microbiology, Chengdu Institute of Biology, Chinese Academy of Sciences, 610041, Chengdu, China; Environmental Microbiology Key Laboratory of Sichuan Province, Chengdu Institute of Biology, Chinese Academy of Sciences, 610041, Chengdu, China.
Machine learning models predict wastewater effluent quality, enhancing treatment plant operations. Optimized K-Nearest Neighbour and Gradient Boosting Decision Tree algorithms achieved high accuracy for Chemical Oxygen Demand, Total Phosphorus, Total Nitrogen, and pH prediction.
Area of Science:
- Environmental Engineering
- Water Treatment Technologies
- Machine Learning Applications
Background:
- Activated sludge wastewater treatment is complex and sensitive to operational variables, leading to effluent instability.
- Stringent environmental regulations necessitate accurate effluent quality prediction for early warning systems.
- Existing models often lack the precision required for real-world wastewater treatment plant (WWTP) management.
Purpose of the Study:
- To develop and optimize machine learning models for predicting effluent quality in WWTPs.
- To evaluate the performance of nine different machine learning algorithms for Chemical Oxygen Demand (COD) prediction.
- To assess the applicability of optimized models for predicting Total Phosphorus (TP), Total Nitrogen (TN), and pH.
Main Methods:
- Employed nine machine learning algorithms to predict effluent COD.
- Optimized models using Hydraulic Retention Time (HRT), K-FOLD data processing, and parameters like dissolved oxygen (DO), sludge return ratio (SRR), and mixed liquid suspended solids (MLSS).
- Validated model performance using Mean Absolute Percentage Error (MAPE), Root Mean Square Error (RMSE), and R-squared (R²).
Main Results:
- K-Nearest Neighbour (KNN) and Gradient Boosting Decision Tree (GBDT) showed superior prediction accuracy for COD (MAPE 7.34%, R² 0.92).
- Optimized models demonstrated high accuracy for TN (MAPE 7.43%, R² 0.93, GBDT), TP (MAPE 17.81%, R² 0.99, KNN), and pH (MAPE 0.53%, R² 0.99, KNN).
- Model optimization significantly improved prediction capabilities for various effluent quality parameters.
Conclusions:
- Machine learning, particularly KNN and GBDT with optimized parameters, offers a robust solution for predicting WWTP effluent quality.
- The study provides insights into optimizing modeling conditions for improved wastewater treatment prediction.
- Findings support practical WWTP operation, management, and energy-saving initiatives.
Related Concept Videos
Typical Model Studies
Mechanistic Models: Compartment Models in Individual and Population Analysis
Design Example: Creating a Hydraulic Model of a Dam Spillway

