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.

Prediction of Influent Wastewater Quality Based on Wavelet Transform and Residual LSTM

Daily water supply prediction method based on integrated learning and deep learning

Water Quality Prediction in Urban Waterways Based on Wavelet Packet Denoising and LSTM

Real-time Water Quality Detection Based on Fluctuation Feature Analysis with the LSTM Model

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

Dynamic Real-Time Forecasting Technique for Reclaimed Water Volumes in Urban River Environmental Management

Adaptive prediction for effluent quality of wastewater treatment plant: Improvement with a dual-stage attention-based LSTM network

A hybrid deep learning approach to improve real-time effluent quality prediction in wastewater treatment plant

Deep learning based data-driven model for detecting time-delay water quality indicators of wastewater treatment plant influent

Predicting Water Quality Based on EEMD and LSTM Networks

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

Deep learning-based prediction of effluent quality of a constructed wetland

Quantitative Modeling and Predictive Analysis of Chemical Oxygen Demand in Wastewater Treatment Systems Utilizing Long Short-Term Memory Neural Network

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

Water Quality Prediction Based on an Innovated Physical and Data Driving Hybrid Model at Basin Scale

Study on Wavelet Multi-Scale Analysis and Prediction of Landslide Groundwater

Accurate prediction of water quality in urban drainage network with integrated EMD-LSTM model

Study on Water Quality Prediction of Urban Reservoir by Coupled CEEMDAN Decomposition and LSTM Neural Network Model

A Hybrid Model for Water Quality Prediction Based on an Artificial Neural Network, Wavelet Transform, and Long Short-Term Memory

A watershed water quality prediction model based on attention mechanism and Bi-LSTM

Ensemble water quality forecasting based on decomposition, sub-model selection, and adaptive interval