Machine learning for high-precision simulation of dissolved organic matter in sewer: Overcoming data restrictions with generative adversarial networks

Abstract

Understanding the transformation process of dissolved organic matter (DOM) in the sewer is imperative for comprehending material circulation and energy flow within the sewer. The machine learning (ML) model provides a feasible way to comprehend and simulate the DOM transformation process in the sewer. In contrast, the model accuracy is limited by data restriction. In this study, a novel framework by integrating generative adversarial network algorithm-machine learning models (GAN-ML) was established to overcome the drawbacks caused by the data restriction in the simulation of the DOM transformation process, and humification index (HIX) was selected as the output variable to evaluate the model performance. Results indicate that the GAN algorithm's virtual dataset could generally enhance the simulation performance of regression models, deep learning models, and ensemble models for the DOM transformation process. The highest prediction accuracy on HIX (R2 of 0.5389 and RMSE of 0.0273) was achieved by the adaptive boosting model which belongs to ensemble models trained by the virtual dataset of 1000 samples. Interpretability analysis revealed that dissolved oxygen (DO) and pH emerge as critical factors warranting attention for the future development of management strategies to regulate the DOM transformation process in sewers. The integrated framework proposed a potential approach for the comprehensive understanding and high-precision simulation of the DOM transformation process, paving the way for advancing sewer management strategy under data restriction.