Application of Soft Actor-Critic Algorithms in Optimizing Wastewater Treatment with Time Delays Integration

TL;DR

This paper presents a novel deep reinforcement learning approach using Soft Actor-Critic algorithms, integrated with a custom simulator, to optimize wastewater treatment processes with stochastic delays, significantly improving phosphorus removal efficiency and reducing costs.

Contribution

It introduces a delay-aware reinforcement learning framework with a custom LSTM-based simulator to handle stochastic delays in wastewater treatment control.

Findings

36% reduction in phosphorus emissions

55% higher reward compared to traditional methods

77% lower deviation from regulatory limits

Abstract

Wastewater treatment plants face unique challenges for process control due to their complex dynamics, slow time constants, and stochastic delays in observations and actions. These characteristics make conventional control methods, such as Proportional-Integral-Derivative controllers, suboptimal for achieving efficient phosphorus removal, a critical component of wastewater treatment to ensure environmental sustainability. This study addresses these challenges using a novel deep reinforcement learning approach based on the Soft Actor-Critic algorithm, integrated with a custom simulator designed to model the delayed feedback inherent in wastewater treatment plants. The simulator incorporates Long Short-Term Memory networks for accurate multi-step state predictions, enabling realistic training scenarios. To account for the stochastic nature of delays, agents were trained under three delay scenarios: no delay, constant delay, and random delay. The results demonstrate that incorporating random delays into the reinforcement learning framework significantly improves phosphorus removal efficiency while reducing operational costs. Specifically, the delay-aware agent achieved 36% reduction in phosphorus emissions, 55% higher reward, 77% lower target deviation from the regulatory limit, and 9% lower total costs than traditional control methods in the simulated environment. These findings underscore the potential of reinforcement learning to overcome the limitations of conventional control strategies in wastewater treatment, providing an adaptive and cost-effective solution for phosphorus removal.