Diffusion-EDFs: Bi-equivariant Denoising Generative Modeling on SE(3) for Visual Robotic Manipulation

Hyunwoo Ryu,Jiwoo Kim,Hyunseok An,Junwoo Chang,Joohwan Seo,Taehan Kim,Yubin Kim,Chaewon Hwang,Jongeun Choi,Roberto Horowitz
DOI: https://doi.org/10.48550/arXiv.2309.02685
2023-11-28
Abstract:Diffusion generative modeling has become a promising approach for learning robotic manipulation tasks from stochastic human demonstrations. In this paper, we present Diffusion-EDFs, a novel SE(3)-equivariant diffusion-based approach for visual robotic manipulation tasks. We show that our proposed method achieves remarkable data efficiency, requiring only 5 to 10 human demonstrations for effective end-to-end training in less than an hour. Furthermore, our benchmark experiments demonstrate that our approach has superior generalizability and robustness compared to state-of-the-art methods. Lastly, we validate our methods with real hardware experiments. Project Website: <a class="link-external link-https" href="https://sites.google.com/view/diffusion-edfs/home" rel="external noopener nofollow">this https URL</a>
Robotics,Artificial Intelligence,Machine Learning
What problem does this paper attempt to address?