Learning to explore when mistakes are not allowed

TL;DR

This paper introduces a two-phase approach for goal-conditioned reinforcement learning that ensures safe exploration without making harmful mistakes, combining safety pretraining with a dynamic action arbitration mechanism.

Contribution

It presents a novel method that pretrains a safety policy and then uses it to guide safe exploration in GCRL, reducing errors and expanding goal coverage.

Findings

Significantly reduces mistakes compared to traditional GCRL.

Achieves broad goal space coverage in simulated environments.

Provides insights into failure modes and future research directions.

Abstract

Goal-Conditioned Reinforcement Learning (GCRL) provides a versatile framework for developing unified controllers capable of handling wide ranges of tasks, exploring environments, and adapting behaviors. However, its reliance on trial-and-error poses challenges for real-world applications, as errors can result in costly and potentially damaging consequences. To address the need for safer learning, we propose a method that enables agents to learn goal-conditioned behaviors that explore without the risk of making harmful mistakes. Exploration without risks can seem paradoxical, but environment dynamics are often uniform in space, therefore a policy trained for safety without exploration purposes can still be exploited globally. Our proposed approach involves two distinct phases. First, during a pretraining phase, we employ safe reinforcement learning and distributional techniques to train a safety policy that actively tries to avoid failures in various situations. In the subsequent safe exploration phase, a goal-conditioned (GC) policy is learned while ensuring safety. To achieve this, we implement an action-selection mechanism leveraging the previously learned distributional safety critics to arbitrate between the safety policy and the GC policy, ensuring safe exploration by switching to the safety policy when needed. We evaluate our method in simulated environments and demonstrate that it not only provides substantial coverage of the goal space but also reduces the occurrence of mistakes to a minimum, in stark contrast to traditional GCRL approaches. Additionally, we conduct an ablation study and analyze failure modes, offering insights for future research directions.