Augmented Proximal Policy Optimization for Safe Reinforcement Learning

Juntao Dai,Jiaming Ji,Long Yang,Qian Zheng,Gang Pan
DOI: https://doi.org/10.1609/aaai.v37i6.25888
2023-06-26
Proceedings of the AAAI Conference on Artificial Intelligence
Abstract:Safe reinforcement learning considers practical scenarios that maximize the return while satisfying safety constraints. Current algorithms, which suffer from training oscillations or approximation errors, still struggle to update the policy efficiently with precise constraint satisfaction. In this article, we propose Augmented Proximal Policy Optimization (APPO), which augments the Lagrangian function of the primal constrained problem via attaching a quadratic deviation term. The constructed multiplier-penalty function dampens cost oscillation for stable convergence while being equivalent to the primal constrained problem to precisely control safety costs. APPO alternately updates the policy and the Lagrangian multiplier via solving the constructed augmented primal-dual problem, which can be easily implemented by any first-order optimizer. We apply our APPO methods in diverse safety-constrained tasks, setting a new state of the art compared with a comprehensive list of safe RL baselines. Extensive experiments verify the merits of our method in easy implementation, stable convergence, and precise cost control.
English Else
What problem does this paper attempt to address?