A Mechanistic Analysis of Sim-and-Real Co-Training in Generative Robot Policies
Organizations: Department of Computer Science, The University of Texas at Austin · Amazon FAR · NVIDIA
Abstract
Co-training, which combines limited in-domain real-world data with abundant surrogate data such as simulation or cross-embodiment robot data, is widely used for training generative robot policies. Despite its empirical success, the mechanisms that determine when and why co-training is effective remain poorly understood. We investigate the mechanism of sim-and-real co-training through theoretical analysis and empirical study, and identify two intrinsic effects governing performance. The first, structured representation alignment"**, reflects a balance between cross-domain representation alignment and domain discernibility, and plays a primary role in downstream performance. The second, the **importance reweighting effect", arises from domain-dependent modulation of action weighting and operates at a secondary level. We validate these effects with controlled experiments on a toy model and extensive sim-and-sim and sim-and-real robot manipulation experiments. Our analysis offers a unified interpretation of recent co-training techniques and motivates a simple method that consistently improves upon prior approaches. More broadly, our aim is to examine the inner workings of co-training and to facilitate research in this direction.
Figures & tables
| NutAssembly | MugCleanup | MugHang | Avg | |
|---|---|---|---|---|
| co-training | 0.85 | 0.44 | 0.27 | 0.52 |
| +discrimination | 0.79 | 0.415 | 0.215 | 0.47 |
| Method | Success Rate | Avg | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| NutAssembly | MugCleanup | MugHang | ||||||||
| Real-only | 11/30 | 8/30 | 6/30 | 8.6/30 | ||||||
| Mixing Ratio | 0.016 | 0.1 | 0.3 | 0.016 | 0.1 | 0.3 | 0.016 | 0.1 | 0.3 | - |
| Co-Training | 17/30 | 11/30 | 16/30 | 16/30 | 9/30 | 7/30 | 8/30 | 13/30 | 7/30 | 15.3/30 |
| + OT | 15/30 | 17/30 | 11/30 | 8/30 | 15/30 | 15/30 | 11/30 | 9/30 | 4/30 | 14.3/30 |
| + ADDA | 13/30 | 13/30 | 15/30 | 6/30 | 14/30 | 11/30 | 10/30 | 14/30 | 7/30 | 14.3/30 |
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
| Task | Pretrain+FT | Co-training |
|---|---|---|
| NutAssembly | 0.77 | 0.925 |
| MugCleanup | 0.215 | 0.495 |
| MugHang | 0.225 | 0.26 |
Robot Policy Learning
Robot policy learning develops controllers that map sensory observations, task goals, and interaction history to actions for embodied robots.
See all 372 papers