Abstract:Water pollution generated from intensive anthropogenic activities has emerged as a critical issue concerning ecosystem balance and livelihoods worldwide. Although optimizing wastewater treatment efficiency is widely regarded as the foremost step to minimize pollutants released into the environment, this widespread application has encountered two major problems: firstly, the significant variation of influent wastewater constituents; secondly, complex treatment processes within wastewater treatment plants (WWTPs). Based on the data collected hourly using real-time sensors in three different full-scale WWTPs (24 h × 365 days × 3 WWTPs × 10 wastewater parameters), this work introduced the potential application of Machine Learning (ML) to predict wastewater quality. In this work, six different ML algorithms were examined and compared, varying from shallow to deep learning architectures including Seasonal Autoregressive Integrated Moving Average (SARIMAX), Random Forest (RF), Support Vector Machine (SVM), Gradient Tree Boosting (GTB), Adaptive Neuro-Fuzzy Inference System (ANFIS) and Long Short-Term Memory (LSTM). These models were developed to detect total phosphorus in the outlet (Outlet-TP), which served as an output variable due to the rising concerns about the eutrophication problem. Irrespective of WWTPs, SARIMAX consistently demonstrated the best performance for regression estimation as evidenced by the lowest values of Mean Square Error (MSE), Mean Absolute Error (MAE), Mean Absolute Percentage Error (MAPE) and the highest coefficient of determination (R2). In terms of computation efficiency, SARIMAX exhibited acceptable time computation, acknowledging the successful application of this algorithm for Outlet-TP modeling. In contrast, the complex structure of LSTM made it time-consuming and unstable coupled with noise, while other shallower architectures, i.e., RF, SVM, GTB, and ANFIS were unable to address large datasets with nonlinear and nonstationary behavior. Consequently, this study provides a reliable and accurate approach to forecast wastewater effluent quality, which is pivotal in terms of the socio-economic aspects of wastewater management.

Machine learning approaches to predict coagulant dosage in water treatment plants

Machine learning regression-based prediction model for the autonomous control of coagulant dosing in smart water purification plants

Integrating water quality and streamflow into prediction of chemical dosage in a drinking water treatment plant using machine learning algorithms

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

Water COD Prediction Based on Machine Learning

Modeling the optimal dosage of coagulants in water treatment plants using various machine learning models

Optimizing coagulant dosage using deep learning models with large-scale data

Prediction of wastewater treatment plant performance through machine learning techniques

Machine learning-based prediction of biological oxygen demand and unit electricity consumption in different-scale wastewater treatment plants

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

Time series-based machine learning for forecasting multivariate water quality in full-scale drinking water treatment with various reagent dosages

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

An efficient neural network model for aiding the coagulation process of water treatment plants

Data-driven water quality prediction for wastewater treatment plants

An Auto-Adjustable and Time-Consistent Model for Determining Coagulant Dosage Based on Operators’ Experience

Modelling coagulant dosage in drinking water treatment plant using advance machine learning model: Hybrid extreme learning machine optimized by Bat algorithm

Prediction of the Optimal Dosage of Poly Aluminum Chloride for Coagulation in Drinking Water Treatment using Artificial Neural Networks

Improved neural network with least square support vector machine for wastewater treatment process

Forecasting effluent and performance of wastewater treatment plant using different machine learning techniques

Water Quality Prediction Based on Machine Learning and Comprehensive Weighting Methods

Predicting effluent quality parameters for wastewater treatment plant: A machine learning-based methodology