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.

Simulating wastewater treatment plants for heavy metals using machine learning models

Prediction of Heavy Metal Removal Performance of Sulfate-Reducing Bacteria Using Machine Learning.

Prediction of effluent arsenic concentration of wastewater treatment plants using machine learning and kriging-based models

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

Heavy metals prediction in coastal marine sediments using hybridized machine learning models with metaheuristic optimization algorithm

Prediction of wastewater treatment plant performance through machine learning techniques

Data-driven water quality prediction for wastewater treatment plants

Machine learning modeling for the prediction of phosphorus and nitrogen removal efficiency and screening of crucial microorganisms in wastewater treatment plants

Prediction of weighted arithmetic water quality index for urban water quality using ensemble machine learning model

Analysis of Machine Learning Models for Wastewater Treatment Plant Sludge Output Prediction

Tracking the impact of heavy metals on human health and ecological environments in complex coastal aquifers using improved machine learning optimization

Machine Learning for the Prediction of Heavy Metal Removal by Chitosan-Based Flocculants

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

Optimizing Stabilization of Contaminated Mining Sludge: A Machine Learning Approach to Predict Strength and Heavy Metal Leaching

Development of artificial intelligence for modeling wastewater heavy metal removal: State of the art, application assessment and possible future research

Prediction and optimization of wastewater treatment process effluent chemical oxygen demand and energy consumption based on typical ensemble learning models

Interpretable Machine Learning for Predicting Heavy Metal Removal Efficiency in Electrokinetic Soil Remediation

Application of machine learning techniques to model a full-scale wastewater treatment plant with biological nutrient removal

Predicting biochemical oxygen demand in wastewater treatment plant using advance extreme learning machine optimized by Bat algorithm

Machine Learning Models with Innovative Outlier Detection Techniques for Predicting Heavy Metal Contamination in Soils.

A Comparative Analysis of Artificial Neural Networks and Wavelet Hybrid Approaches to Long-Term Toxic Heavy Metal Prediction