Variational OOD State Correction for Offline Reinforcement Learning
This addresses a key challenge in offline RL for improving agent safety and performance, though it appears incremental as it builds on existing OOD correction approaches.
The paper tackled the problem of state distributional shift in offline reinforcement learning by proposing a novel method for out-of-distribution state correction, resulting in validated effectiveness on MuJoCo and AntMaze benchmarks.
The performance of Offline reinforcement learning is significantly impacted by the issue of state distributional shift, and out-of-distribution (OOD) state correction is a popular approach to address this problem. In this paper, we propose a novel method named Density-Aware Safety Perception (DASP) for OOD state correction. Specifically, our method encourages the agent to prioritize actions that lead to outcomes with higher data density, thereby promoting its operation within or the return to in-distribution (safe) regions. To achieve this, we optimize the objective within a variational framework that concurrently considers both the potential outcomes of decision-making and their density, thus providing crucial contextual information for safe decision-making. Finally, we validate the effectiveness and feasibility of our proposed method through extensive experimental evaluations on the offline MuJoCo and AntMaze suites.