AtomEgo: Exploring Ego-Robot Integration for Embodied Foundation Model Pretraining
Organizations: INFIFORCE · Tongji University · HKU · Lionrock Artificial Intelligence Laboratory ·
Abstract
Embodied foundation models are constrained by the limited scale and diversity of robot demonstrations, motivating the use of large-scale egocentric human interaction data. However, how to effectively incorporate such data into embodied-model pre-training remains unclear because of substantial embodiment and action-space gaps between humans and robots. We present AtomEgo, a systematic study of ego--robot co-training supported by a curated corpus of approximately 2,659 hours and a scalable data processing pipeline. Across vision--language--action and world--action model architectures, we investigate three representative paradigms: joint co-training with domain-specific action heads, progressive ego-to-robot transfer through embodiment alignment, and joint video--action modeling. We evaluate these paradigms through multi-task real-robot experiments and language-conditioned cross-embodiment representation analysis. Our results reveal a simple principle: Data Scale * Alignment Quality --> Capability Gain; egocentric data can improve generalization, but their value depends on how effectively they are aligned and utilized. This principle can provide practical guidance for scalable ego--robot pre-training.
Figures & tables
| Group | Source | Episodes | Hours | Proportion |
|---|---|---|---|---|
| Robot data | Piper (in-house) | 5,829 | 25.5 | 1.745% |
| AgiBot World Beta [ 2 ] | 21,837 | 314.0 | 6.537% | |
| DROID [ 12 ] | 64,124 | 343.0 | 19.196% | |
| RoboCOIN [ 29 ] | 93,352 | 655.7 | 27.945% | |
| RoboMIND [ 28 ] | 83,152 | 221.7 | 24.892% | |
| Ego data | EgoVerse [ 21 ] | 64,464 | 1,079.5 | 19.297% |
| Range | Width | Physical meaning | Action semantics |
|---|---|---|---|
| 7 | Left arm joints | Relative | |
| 6 | Left end-effector position and Euler rotation | Absolute | |
| 3 | Reserved | Masked | |
| 1 | Left gripper | Absolute | |
| 12 | Left hand joints | Absolute | |
| 7 | Right arm joints | Relative |
| Stage | Objective and training data | Initialization | Updates | Trainable components |
|---|---|---|---|---|
| 1 | Egocentric prior learning on Ego data | Base PaliGemma VLM | 100,000 | Full model |
| 2 | Embodiment alignment on task-aligned human and robot data | Stage 1 checkpoint | 50,000 | Vision encoder and action expert |
| 3 | Robot pre-training on robot datasets | Stage 2 checkpoint | 97,728 | Full model |
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
| Task | ID setup | OOD type | OOD perturbation | OOD specification and controls | Success criterion |
|---|---|---|---|---|---|
| Cook Bread | Standard pot and bread poses | Spatial | Rotate pot | Pot yaw CCW; pot center and other object poses unchanged | Bread is placed in the pot, and the pot is placed on the cloth |
| Hang Cup | Standard cup mass | Physical | Increase cup mass | Mass ; cup appearance and initial pose unchanged | Handle is inserted and the cup hangs stably for the predefined duration |
| Place Fruits | Standard target set | Appearance | Add visually similar non-target fruit as a distractor | Distractor identity and pose fixed; target identities and initial poses unchanged | All three target fruits are placed in the basket while the distractor remains outside |
| Place Plate | Standard rack pose | Spatial | Translate rack | Rack displacement fixed; plate pose and rack orientation unchanged | Plate is aligned, inserted, released, and remains stable in the rack |
| Place Pot | Standard pot pose | Spatial | Rotate pot | Pot yaw ; pot center and cloth pose unchanged | Pot is grasped by the handle and placed on the cloth |
| Press Button | Standard illumination | Appearance | Reduce illumination | Illuminance ; camera and button poses unchanged | Button activation is registered and the end effector reaches the predefined retreat region |
| Hyperparameter | Robot pre-training | Multi-task post-training |
|---|---|---|
| Training data | 34 robot datasets; 150.1M source frames; EgoVerse excluded | Two in-house Piper multi-task datasets |
| Action representation | Unified80 | Unified80 |
| Hardware | NVIDIA B200 GPUs | NVIDIA B200 GPUs |
| FSDP devices | 4 | 4 |
| Global batch size (train / validation) | 1,536 / 96 | 512 / 96 |
| Training steps | 97,728 | 20,000 |
| Variant | ID success | OOD success | Combined success |
|---|---|---|---|
| Atom-DH (+OT) | 38.57% (27/70) | 31.43% (22/70) | 35.00% (49/140) |
| Atom-DH (w/o OT) | 44.29% (31/70) | 42.86% (30/70) | 43.57% (61/140) |
Robot Learning
Robot learning develops methods for robots to acquire skills and adapt from demonstrations, interaction, data, and physical models.
See all 321 papers