Safety Augmented Value Estimation From Demonstrations (SAVED): Safe Deep Model-Based RL for Sparse Cost Robotic Tasks
Safety Augmented Value Estimation From Demonstrations (SAVED): Safe Deep Model-Based RL for Sparse Cost Robotic Tasks
复制标题
DOI:
10.1109/lra.2020.2976272
复制
发表时间:
2019-05
影响因子:
5.2
通讯作者:
Brijen Thananjeyan;A. Balakrishna;Ugo Rosolia;Felix Li;R. McAllister;Joseph Gonzalez;S. Levine;F. Borrelli;Ken Goldberg
中科院分区:
文献类型:
--
作者:
Brijen Thananjeyan;A. Balakrishna;Ugo Rosolia;Felix Li;R. McAllister;Joseph Gonzalez;S. Levine;F. Borrelli;Ken Goldberg
Reinforcement learning (RL) for robotics is challenging due to the difficulty in hand-engineering a dense cost function, which can lead to unintended behavior, and dynamical uncertainty, which makes exploration and constraint satisfaction challenging. We address these issues with a new model-based reinforcement learning algorithm, Safety Augmented Value Estimation from Demonstrations (SAVED), which uses supervision that only identifies task completion and a modest set of suboptimal demonstrations to constrain exploration and learn efficiently while handling complex constraints. We then compare SAVED with 3 state-of-the-art model-based and model-free RL algorithms on 6 standard simulation benchmarks involving navigation and manipulation and a physical knot-tying task on the da Vinci surgical robot. Results suggest that SAVED outperforms prior methods in terms of success rate, constraint satisfaction, and sample efficiency, making it feasible to safely learn a control policy directly on a real robot in less than an hour. For tasks on the robot, baselines succeed less than $\text{5}\%$ of the time while SAVED has a success rate of over $\text{75}\%$ in the first 50 training iterations. Code and supplementary material is available at https://tinyurl.com/saved-rl.