English

Actionable Models: Unsupervised Offline Reinforcement Learning of Robotic Skills

Robotics 2021-06-14 v3 Machine Learning

Abstract

We consider the problem of learning useful robotic skills from previously collected offline data without access to manually specified rewards or additional online exploration, a setting that is becoming increasingly important for scaling robot learning by reusing past robotic data. In particular, we propose the objective of learning a functional understanding of the environment by learning to reach any goal state in a given dataset. We employ goal-conditioned Q-learning with hindsight relabeling and develop several techniques that enable training in a particularly challenging offline setting. We find that our method can operate on high-dimensional camera images and learn a variety of skills on real robots that generalize to previously unseen scenes and objects. We also show that our method can learn to reach long-horizon goals across multiple episodes through goal chaining, and learn rich representations that can help with downstream tasks through pre-training or auxiliary objectives. The videos of our experiments can be found at https://actionable-models.github.io

Keywords

Cite

@article{arxiv.2104.07749,
  title  = {Actionable Models: Unsupervised Offline Reinforcement Learning of Robotic Skills},
  author = {Yevgen Chebotar and Karol Hausman and Yao Lu and Ted Xiao and Dmitry Kalashnikov and Jake Varley and Alex Irpan and Benjamin Eysenbach and Ryan Julian and Chelsea Finn and Sergey Levine},
  journal= {arXiv preprint arXiv:2104.07749},
  year   = {2021}
}