Constrained Model-based Reinforcement Learning with Robust Cross-Entropy Method

Zuxin Liu,Hongyi Zhou,Baiming Chen,Sicheng Zhong,Martial Hebert,Ding Zhao
DOI: https://doi.org/10.48550/arXiv.2010.07968
2021-03-06
Abstract:This paper studies the constrained/safe reinforcement learning (RL) problem with sparse indicator signals for constraint violations. We propose a model-based approach to enable RL agents to effectively explore the environment with unknown system dynamics and environment constraints given a significantly small number of violation budgets. We employ the neural network ensemble model to estimate the prediction uncertainty and use model predictive control as the basic control framework. We propose the robust cross-entropy method to optimize the control sequence considering the model uncertainty and constraints. We evaluate our methods in the Safety Gym environment. The results show that our approach learns to complete the tasks with a much smaller number of constraint violations than state-of-the-art baselines. Additionally, we are able to achieve several orders of magnitude better sample efficiency when compared with constrained model-free RL approaches. The code is available at \url{<a class="link-external link-https" href="https://github.com/liuzuxin/safe-mbrl" rel="external noopener nofollow">this https URL</a>}.
Artificial Intelligence,Machine Learning,Robotics
What problem does this paper attempt to address?