Optimized Deep Learning Models for Effluent Prediction in Wastewater Treatment Processes

Canyun Yang,Zhuoyue Guo,Yun Geng,Fengshan Zhang,Wenguang Wei,Hongbin Liu
DOI: https://doi.org/10.1039/d3ew00875d
2024-03-28
Environmental Science Water Research & Technology
Abstract:Monitoring wastewater holds significant importance in minimizing pollution discharge and enhancing resource utilization. This study aims to develop a soft sensor for the prediction and optimization of critical wastewater treatment indicators. To address the nonlinear characteristics of the wastewater process, this work introduces a novel model incorporating convolutional neural network (CNN) and gated recursive unit (GRU). Considering the high-dimensional data in practical production processes, feature selection based on Lasso regression is employed to identify key features for the training model. In addition, the proposed CNN-GRU model is enhanced using parameters determined by Bayesian optimization algorithm in this work. The model is trained using urban wastewater data to predict effluent chemical oxygen demand (COD) and suspended solids (SS). For COD prediction, the test set achieves an R2 of 0.97, and it attains an R2 of 0.90 for SS prediction, demonstrating superior generalization ability. Furthermore, comparative analysis with random forest, long short-term memory, CNN and GRU models reveals improved prediction performance of the CNN-GRU model after Bayesian optimization.
environmental sciences,engineering, environmental,water resources
What problem does this paper attempt to address?