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.

Transformer based Multi-output Regression Learning for Wastewater Treatment

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

Hidformer: Hierarchical Dual-Tower Transformer Using Multi-Scale Mergence for Long-Term Time Series Forecasting

Prediction of Urban Groundwater Level Based on CNN-Transformer

Dynamic Real-Time Prediction of Reclaimed Water Volumes Using the Improved Transformer Model and Decomposition Integration Technology

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

Long-term prediction of multiple river water quality indexes based on hybrid deep learning models

Optimized Deep Learning Models for Effluent Prediction in Wastewater Treatment Processes

A Ladder Water Level Prediction Model for the Yangtze River Based on Transfer Learning and Transformer

A multi-step water quality prediction model based on the Savitzky-Golay filter and Transformer optimized network

Research on Runoff Prediction Based on Time2Vec-TCN-Transformer Driven by Multi-Source Data

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

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

Enhancing sewage flow prediction using an integrated improved SSA-CNN-Transformer-BiLSTM model

A New Method for Transformer Fault Prediction Based on Multifeature Enhancement and Refined Long Short-Term Memory

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

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

Enhancing hydrological modeling with transformers: a case study for 24-h streamflow prediction

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