Prediction of biological nutrients removal in full-scale wastewater treatment plants using H2O automated machine learning and back propagation artificial neural network model: Optimization and comparison

Abstract

The effective control of total nitrogen (ETN) and total phosphorus (ETP) in effluent is challenging for wastewater treatment plants (WWTPs). In this work, automated machine learning (AutoML) (mean square error = 0.4200 ∼ 3.8245, R2 = 0.5699 ∼ 0.6219) and back propagation artificial neural network (BPANN) model (mean square error = 0.0012 ∼ 6.9067, R2 = 0.4326 ∼ 0.8908) were used to predict and analyze biological nutrients removal in full-scale WWTPs. Interestingly, BPANN model presented high prediction performance and general applicability for WWTPs with different biological treatment units. However, the AutoML candidate models were more interpretable, and the results showed that electricity carbon emission dominated the prediction. Meanwhile, increasing data volume and types of WWTP hardly affected the interpretable results, demonstrating its wide applicability. This study demonstrated the validity and the specific advantages of predicting ETN and ETP using H2O AutoML and BPANN model, which provided guidance on the prediction and improvement of biological nutrients removal in WWTPs.