A novel deep learning ensemble model based on two-stage feature selection and intelligent optimization for water quality prediction

Abstract

Accurate prediction of effluent total nitrogen (E-TN) can assist in feed-forward control of wastewater treatment plants (WWTPs) to ensure effluent compliance with standards while reducing energy consumption. However, multivariate time series prediction of E-TN is a challenge due to the complex nonlinearity of WWTPs. This paper proposes a novel prediction framework that combines a two-stage feature selection model, the Golden Jackal Optimization (GJO) algorithm, and a hybrid deep learning model, CNN-LSTM-TCN (CLT), aiming to effectively capture the nonlinear relationships of multivariate time series in WWTPs. Specifically, convolutional neural network (CNN), long short-term memory (LSTM), and temporal convolutional network (TCN) combined to build a hybrid deep learning model CNN-LSTM-TCN (CLT). A two-stage feature selection method is utilized to determine the optimal feature subset to reduce the complexity and improve the accuracy of the prediction model, and then, the feature subset is input into the CLT. The hyperparameters of the CLT are optimized using GJO to further improve the prediction performance. Experiments indicate that the two-stage feature selection model learns the optimal feature subset to predict best, and the GJO-CLT achieves the best performance for different backtracking windows and prediction steps. These results demonstrate that the prediction system excels in the task of multivariate water quality time series prediction of WWTPs.