Search papers, labs, and topics across Lattice.
This paper introduces Dyna-style Safety Augmented Reinforcement Learning (Dyna-SAuR), an algorithm that learns a safety filter and control policy using a learned uncertainty-aware dynamics model. The key idea is to use a learned dynamics model to predict uncertainty and avoid high-uncertainty regions, thereby improving safety during exploration. Experiments on CartPole and MuJoCo Walker show Dyna-SAuR reduces failures by two orders of magnitude compared to existing methods.
RL agents can learn to stay safe in complex environments without excessive conservatism by explicitly modeling and avoiding regions of high uncertainty.
Safety remains an open problem in reinforcement learning (RL), especially during training. While safety filters are promising to address safe exploration, they are generally poorly suited for high-dimensional systems with unknown dynamics. We propose Dyna-style Safety Augmented Reinforcement Learning (Dyna-SAuR), a novel algorithm that learns both a scalable safety filter and a control policy using a learned uncertainty-aware dynamics model, while requiring minimal domain knowledge. The filter avoids failures and high uncertainty regions. Thus, better models expand the set of safe and certain states, reducing filter conservatism. We present the effectiveness of Dyna-SAuR on goal-reaching CartPole as well as MuJoCo Walker, reducing failures compared to state-of-the-art methods by 2 orders of magnitude.
