IROS 2021poster135 citations

State-Only Imitation Learning for Dexterous Manipulation

Ilija Radosavovic, Xiaolong Wang, Lerrel Pinto, Jitendra Malik

Abstract

Modern model-free reinforcement learning methods have recently demonstrated impressive results on a number of problems. However, complex domains like dexterous manipulation remain a challenge due to the high sample complexity. To address this, current approaches employ expert demonstrations in the form of state-action pairs, which are difficult to obtain for real-world settings such as learning from videos. In this paper, we move toward a more realistic setting and explore state-only imitation learning. To tackle this setting, we train an inverse dynamics model and use it to predict actions for state-only demonstrations. The inverse dynamics model and the policy are trained jointly. Our method performs on par with state-action approaches and considerably outperforms RL alone. By not relying on expert actions, we are able to learn from demonstrations with different dynamics, morphologies, and objects. Videos available on the {\text{project page}}{\text{project page}}.

BibTeX
@inproceedings{iros2021_stateonlyimitati,
  title = {State-Only Imitation Learning for Dexterous Manipulation},
  author = {Ilija Radosavovic and Xiaolong Wang and Lerrel Pinto and Jitendra Malik},
  booktitle = {IROS 2021},
  year = {2021}
}
State-Only Imitation Learning for Dexterous Manipulation · IROS 2021