Deep Learning Based Simulators for the Phosphorus Removal Process Control in Wastewater Treatment via Deep Reinforcement Learning Algorithms

Esmaeel Mohammadi,Mikkel Stokholm-Bjerregaard,Aviaja Anna Hansen,Per Halkjær Nielsen,Daniel Ortiz-Arroyo,Petar Durdevic
DOI: https://doi.org/10.1016/j.engappai.2024.107992
2024-01-23
Abstract:Phosphorus removal is vital in wastewater treatment to reduce reliance on limited resources. Deep reinforcement learning (DRL) is a machine learning technique that can optimize complex and nonlinear systems, including the processes in wastewater treatment plants, by learning control policies through trial and error. However, applying DRL to chemical and biological processes is challenging due to the need for accurate simulators. This study trained six models to identify the phosphorus removal process and used them to create a simulator for the DRL environment. Although the models achieved high accuracy (>97%), uncertainty and incorrect prediction behavior limited their performance as simulators over longer horizons. Compounding errors in the models' predictions were identified as one of the causes of this problem. This approach for improving process control involves creating simulation environments for DRL algorithms, using data from supervisory control and data acquisition (SCADA) systems with a sufficient historical horizon without complex system modeling or parameter estimation.
Systems and Control,Artificial Intelligence,Machine Learning
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the challenges in controlling the phosphorus removal process in wastewater treatment plants through deep reinforcement learning algorithms. Specifically, phosphorus removal is crucial for reducing dependence on limited resources, while existing control strategies such as fuzzy control and model predictive control (MPC) have limitations when dealing with the complexity and uncertainty in the wastewater treatment process. Therefore, researchers try to use deep reinforcement learning (DRL) to optimize this process. ### Background and Problem Description 1. **Importance of Phosphorus and Removal Requirements** - Phosphorus (P) is an essential element for plant growth and human nutrition, but its available amount is limited. - Phosphorus emissions from wastewater treatment plants (WWTP) into surface water can lead to eutrophication, affecting the environment and human health. - Efficient phosphorus removal can prevent eutrophication and its consequences. 2. **Limitations of Existing Control Methods** - Traditional control methods such as dissolved oxygen (DO) are not the best indicators. - Changes in phosphate concentration in incoming wastewater and process conditions (pH value, temperature, etc.) increase the uncertainty of control. - Existing benchmark simulation models (BSM) cannot well represent the dynamic characteristics of the phosphorus removal process. 3. **Application Potential of Deep Reinforcement Learning** - Deep reinforcement learning (DRL) is a machine - learning technique that can learn control strategies through trial and error and optimize complex nonlinear systems. - Although DRL has been successful in fields such as games, robotics, and control systems, its application in wastewater treatment is still very limited. ### Main Contributions of the Paper 1. **Creation of Simulation Environment** - To overcome the limitations of existing control methods, researchers trained six different models to identify the phosphorus removal process and used these models to create a simulation environment. - These models include linear models (LTSF Linear), recurrent neural networks (LSTM), attention - based models (Transformer, Informer), and autocorrelation mechanisms (Autoformer). 2. **Model Performance Evaluation** - Although these models perform well in terms of prediction accuracy (> 97%), there are still problems of uncertainty and error accumulation in long - term prediction. - By comparing the performance of different models, researchers aim to find the most effective time - series prediction model to support the implementation of DRL algorithms. 3. **Practical Application and Verification** - This study uses actual data from the Agtrup wastewater treatment plant in Denmark for model training and verification. - By optimizing hyperparameters and using the Adam optimization method, the model achieved the lowest mean - square error (MSE) on the validation set. ### Conclusion This paper lays the foundation for the application of deep reinforcement learning in the phosphorus removal process in wastewater treatment plants by creating an accurate simulation environment. Although there are some challenges, such as error accumulation in long - term prediction, the research results show that DRL has the potential to optimize control strategies in complex and uncertain environments. This provides an important reference for future research and practical applications.