Several prior works use probabilistic dynamical models to avoid unsafe behaviour [1]}, [2]}, [3]} during learning through trial-and-error. For instance, in [3]}, the agent first uses a model-based RL approach to learn a probabilistic model to capture uncertainty about transition dynamics and catastrophic states. The model is then used in the real world for predicting and avoiding potentially unsafe states.
