An attempt to augment performance of machine learning models in a pilot-scale urban wastewater treatment system

The challenge we face is striking a balance between ensuring the standards for wastewater treatment processes and minimizing energy consumption. As such, accurate predictive models become critical. With machine learning algorithms having the ability to discern relationships between inputs and outputs, broad interest has been shown in their applications in wastewater treatment. A key research focus is on effectively adjusting these generic machine learning algorithms to adapt to frequent fluctuations of water quality, inflow volume, and environmental variations. This study developed a Nonlinear Autoregressive neural network (NARX) model having an adaptive real-time updating module to predict the ammonium concentration in the treated effluent. The model has exhibited high predictive accuracy on the testing data, achieving an R2 value of 0.997. By incorporating an adaptive real-time updating module, the model's capability to predict effluent ammonium concentration on novel data sets has been significantly improved, with the R2 value for these predictions increasing from an initial 0.918 to a more refined range of 0.966 to 0.984. Our findings confirm that integrating real-time updating capabilities with machine learning algorithms substantially improves the stability and predictive performance of the models. These outcomes will prompt a broader integration of artificial intelligence in the domain of environmental engineering, which shall foster the implementation of more accurate and eco-friendly wastewater treatment strategies. [Display omitted] •A NARX model was developed for a pilot-scale wastewater treatment process.•The model was enhanced with adaptive real-time learning.•Machine learning analyzed the impact of indicators on ammonia concentration