SAMBA: safe model-based & active reinforcement learning

Alexander I. Cowen-Rivers,Daniel Palenicek,Vincent Moens,Mohammed Amin Abdullah,Aivar Sootla,Jun Wang,Haitham Bou-Ammar
DOI: https://doi.org/10.1007/s10994-021-06103-6
IF: 5.414
2022-01-01
Machine Learning
Abstract:In this paper, we propose SAMBA, a novel framework for safe reinforcement learning that combines aspects from probabilistic modelling, information theory, and statistics. Our method builds upon PILCO to enable active exploration using novel acquisition functions for out-of-sample Gaussian process evaluation optimised through a multi-objective problem that supports conditional-value-at-risk constraints. We evaluate our algorithm on a variety of safe dynamical system benchmarks involving both low and high-dimensional state representations. Our results show orders of magnitude reductions in samples and violations compared to state-of-the-art methods. Lastly, we provide intuition as to the effectiveness of the framework by a detailed analysis of our acquisition functions and safety constraints.
computer science, artificial intelligence
What problem does this paper attempt to address?