EgoMimic: Scaling Imitation Learning via Egocentric Video

Simar Kareer, Dhruv Patel, Ryan Punamiya, Pranay Mathur, Shuo Cheng +3 more
10/31/2024
cs.ROcs.CV

No official code found. Best implementation: SimarKareer/EgoMimic (218 stars, MIT, last commit 2024-11-10).

Abstract

The scale and diversity of demonstration data required for imitation learning is a significant challenge. We present EgoMimic, a full-stack framework which scales manipulation via human embodiment data, specifically egocentric human videos paired with 3D hand tracking. EgoMimic achieves this through: (1) a system to capture human embodiment data using the ergonomic Project Aria glasses, (2) a low-cost bimanual manipulator that minimizes the kinematic gap to human data, (3) cross-domain data alignment techniques, and (4) an imitation learning architecture that co-trains on human and robot data. Compared to prior works that only extract high-level intent from human videos, our approach treats human and robot data equally as embodied demonstration data and learns a unified policy from both data sources. EgoMimic achieves significant improvement on a diverse set of long-horizon, single-arm and bimanual manipulation tasks over state-of-the-art imitation learning methods and enables generalization to entirely new scenes. Finally, we show a favorable scaling trend for EgoMimic, where adding 1 hour of additional hand data is significantly more valuable than 1 hour of additional robot data. Videos and additional information can be found at https://egomimic.github.io/

View on arXivView PDF

Code Implementations(1)

MIT
21821Python, Jupyter NotebookOct 31, 20241 years agoMIT

Similar Papers

Cite this paper

@article{kareer2024egomimic,
  title  = {EgoMimic: Scaling Imitation Learning via Egocentric Video},
  author = {Simar Kareer and Dhruv Patel and Ryan Punamiya and Pranay Mathur and Shuo Cheng and Chen Wang and Judy Hoffman and Danfei Xu},
  year   = {2024},
  eprint = {2410.24221},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2410.24221v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2410.24221.svg)](https://codeofpaper.com/paper/2410.24221)

Discussion