Dreamitate: Real-World Visuomotor Policy Learning via Video Generation
Fuente:
arXiv
Enregistré dans:
| Auteurs principaux: | , , , , , , , |
|---|---|
| Format: | Preprint |
| Publié: |
2024
|
| Sujets: | |
| Accès en ligne: | |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866911930745094144 |
|---|---|
| author | Liang, Junbang Liu, Ruoshi Ozguroglu, Ege Sudhakar, Sruthi Dave, Achal Tokmakov, Pavel Song, Shuran Vondrick, Carl |
| author_facet | Liang, Junbang Liu, Ruoshi Ozguroglu, Ege Sudhakar, Sruthi Dave, Achal Tokmakov, Pavel Song, Shuran Vondrick, Carl |
| contents | A key challenge in manipulation is learning a policy that can robustly generalize to diverse visual environments. A promising mechanism for learning robust policies is to leverage video generative models, which are pretrained on large-scale datasets of internet videos. In this paper, we propose a visuomotor policy learning framework that fine-tunes a video diffusion model on human demonstrations of a given task. At test time, we generate an example of an execution of the task conditioned on images of a novel scene, and use this synthesized execution directly to control the robot. Our key insight is that using common tools allows us to effortlessly bridge the embodiment gap between the human hand and the robot manipulator. We evaluate our approach on four tasks of increasing complexity and demonstrate that harnessing internet-scale generative models allows the learned policy to achieve a significantly higher degree of generalization than existing behavior cloning approaches. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2406_16862 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Dreamitate: Real-World Visuomotor Policy Learning via Video Generation Liang, Junbang Liu, Ruoshi Ozguroglu, Ege Sudhakar, Sruthi Dave, Achal Tokmakov, Pavel Song, Shuran Vondrick, Carl Robotics Computer Vision and Pattern Recognition A key challenge in manipulation is learning a policy that can robustly generalize to diverse visual environments. A promising mechanism for learning robust policies is to leverage video generative models, which are pretrained on large-scale datasets of internet videos. In this paper, we propose a visuomotor policy learning framework that fine-tunes a video diffusion model on human demonstrations of a given task. At test time, we generate an example of an execution of the task conditioned on images of a novel scene, and use this synthesized execution directly to control the robot. Our key insight is that using common tools allows us to effortlessly bridge the embodiment gap between the human hand and the robot manipulator. We evaluate our approach on four tasks of increasing complexity and demonstrate that harnessing internet-scale generative models allows the learned policy to achieve a significantly higher degree of generalization than existing behavior cloning approaches. |
| title | Dreamitate: Real-World Visuomotor Policy Learning via Video Generation |
| topic | Robotics Computer Vision and Pattern Recognition |
| url | https://arxiv.org/abs/2406.16862 |