Constraints Penalized Q-Learning for Safe Offline Reinforcement Learning

Published in Thirty-Sixth AAAI Conference on Artificial Intelligence (AAAI2022), 2021

Recommended citation: Xu, H., Zhan, X., and Zhu, X. Constraints Penalized Q-Learning for Safe Offline Reinforcement Learning. In the Thirty-Sixth AAAI Conference on Artificial Intelligence (AAAI2022).

Abstract

We study the problem of safe offline reinforcement learning (RL), the goal is to learn a policy that maximizes long-term reward while satisfying safety constraints given only offline data, without further interaction with the environment. This problem is more appealing for real world RL applications, in which data collection is costly or dangerous. Enforcing constraint satisfaction is non-trivial, especially in offline settings, as there is a potential large discrepancy between the policy distribution and the data distribution, causing errors in estimating the value of safe constraints. We show that naive approaches that combine techniques from safe RL and offline RL can only learn sub-optimal solution. We thus develop a simple yet effective algorithm, Constraints Penalized Q-Learning (CPQ), to solve the problem. Our method admits the use of data generated by mixed behavior policies. We present a theoretical analysis and demonstrate empirically that our approach can learn robustly across a variety of benchmark control tasks, outperforming several baselines.

Arxiv version of the paper