Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

COptiDICE: Offline Constrained Reinforcement Learning via Stationary Distribution Correction Estimation

About

We consider the offline constrained reinforcement learning (RL) problem, in which the agent aims to compute a policy that maximizes expected return while satisfying given cost constraints, learning only from a pre-collected dataset. This problem setting is appealing in many real-world scenarios, where direct interaction with the environment is costly or risky, and where the resulting policy should comply with safety constraints. However, it is challenging to compute a policy that guarantees satisfying the cost constraints in the offline RL setting, since the off-policy evaluation inherently has an estimation error. In this paper, we present an offline constrained RL algorithm that optimizes the policy in the space of the stationary distribution. Our algorithm, COptiDICE, directly estimates the stationary distribution corrections of the optimal policy with respect to returns, while constraining the cost upper bound, with the goal of yielding a cost-conservative policy for actual constraint satisfaction. Experimental results show that COptiDICE attains better policies in terms of constraint satisfaction and return-maximization, outperforming baseline algorithms.

Jongmin Lee, Cosmin Paduraru, Daniel J. Mankowitz, Nicolas Heess, Doina Precup, Kee-Eung Kim, Arthur Guez• 2022

Related benchmarks

TaskDatasetResultRank
Safe Reinforcement LearningMetaDrive
Normalized Reward0.58
10
Safe Reinforcement LearningBullet Safety Gym
Normalized Reward0.54
10
BallRunBullet-Safety-Gym OSRL
Reward0.63
9
DroneRunBullet-Safety-Gym OSRL
Reward0.71
9
BallCircleBullet-Safety-Gym OSRL
Reward0.73
9
CarRunBullet-Safety-Gym OSRL
Reward0.9
9
DroneCircleBullet-Safety-Gym OSRL
Reward0.35
9
CarCircleBullet-Safety-Gym OSRL
Reward0.52
9
Collision AvoidanceAGV Collision Avoidance
Safe Set Volume (%)62.32
8
Constrained Offline Reinforcement LearningDSRL PointCircle1
Normalized Return0.86
7
Showing 10 of 28 rows

Other info

Follow for update