Adaptive prediction for effluent quality of wastewater treatment plant: Improvement with a dual-stage attention-based LSTM network

Abstract

The accurate effluent prediction plays a crucial role in providing early warning for abnormal effluent and achieving the adjustment of feedforward control parameters during wastewater treatment. This study applied a dual-staged attention mechanism based on long short-term memory network (DA-LSTM) to improve the accuracy of effluent quality prediction. The results showed that input attention (IA) and temporal attention (TA) significantly enhanced the prediction performance of LSTM. Specially, IA could adaptively adjust feature weights to enhance the robustness against input noise, with R2 increased by 13.18%. To promote its long-term memory ability, TA was used to increase the memory span from 96 h to 168 h. Compared to a single LSTM model, the DA-LSTM model showed an improvement in prediction accuracy by 5.10%, 2.11%, 14.47% for COD, TP, and TN. Additionally, DA-LSTM demonstrated excellent generalization performance in new scenarios, with the R2 values for COD, TP, and TN increasing by 22.67%, 20.06%, and 17.14% respectively, while the MAPE values decreased by 56.46%, 63.08%, and 42.79%. In conclusion, the DA-LSTM model demonstrated excellent prediction performance and generalization ability due to its advantages of feature-adaptive weighting and long-term memory focusing. This has forward-looking significance for achieving efficient early warning of abnormal operating conditions and timely management of control parameters.