Shandong Science ›› 2024, Vol. 37 ›› Issue (6): 116-124.doi: 10.3976/j.issn.1002-4026.20240010

• Environment and Ecology • Previous Articles     Next Articles

Prediction of effluent total nitrogen in wastewater treatment using LSTM neural network

YU Mingquan1(), SHI Haoming2,*()   

  1. 1. Power China Huadong Engineering Corporation Limited, Hangzhou 311122, China
    2. College of Civil Engineering and Architecture, Zhejiang University, Hangzhou 310058, China
  • Received:2024-01-16 Published:2024-12-20 Online:2024-12-05

Abstract:

The effluent total nitrogen (TN) is one of the key indicators for assessing the biological denitrification performance of wastewater treatment plants(WWTPs). To mitigate the prevalent issue of excessive TN discharges from WTTPs, we proposed a real-time prediction model based on long short-term memory (LSTM) networks. We performed Pearson correlation analysis to determine model inputs and used grid search algorithm to optimize model hyperparameters. Then, we used the proposed model to predict the actual effluent TN in a WWTP in Chongqing and compared its predictive performance with that of traditional time-series models. Results indicate that the proposed model can effectively predict effluent TN with an average absolute error of 0.911 mg/L, an average root mean square error of 1.074 mg/L, and an average absolute percentage error of 11.28%. All of these performance indicators surpass those of the recurrent neural network and ARIMA models. The proposed model can serve as the foundation for effective monitoring of effluent TN.

Key words: long short-term memory model, Pearson correlation, grid search algorithm, effluent total nitrogen

CLC Number: 

  • TU992

Open Access This article is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License (CC BY-NC 4.0), which permits third parties to freely share (i.e., copy and redistribute the material in any medium or format) and adapt (i.e., remix, transform, or build upon the material) the articles published in this journal, provided that appropriate credit is given, a link to the license is provided, and any changes made are indicated. The material may not be used for commercial purposes. For details of the CC BY-NC 4.0 license, please visit: https://creativecommons.org/licenses/by-nc/4.0