On the Strengths and Weaknesses of Data for Open-set Embodied Assistance
Organizations: Distyl AI · Toyota Research Institute
Abstract
Embodied foundation models are increasingly performant in real-world domains such as robotics or autonomous driving. These models are often deployed in interactive or assistive settings, where it is important that these assistive models generalize to new users and new tasks. Diverse interactive data generation offers a promising avenue for providing data-efficient generalization capabilities for interactive embodied foundation models. In this paper, we investigate the generalization capabilities of a multimodal foundation model fine-tuned on diverse interactive assistance data in a synthetic domain. We explore generalization along two axes: a) assistance with unseen categories of user behavior and b) providing guidance in new configurations not encountered during training. We study a broad capability called Open-Set Corrective Assistance, in which the model needs to inspect lengthy user behavior and provide assistance through either corrective actions or language-based feedback. This task remains unsolved in prior work, which typically assumes closed corrective categories or relies on external planners, making it a challenging testbed for evaluating the limits of assistive data. To support this task, we generate synthetic assistive datasets in Overcooked and fine-tune a LLaMA-based model to evaluate generalization to novel tasks and user behaviors. Our approach provides key insights into the nature of assistive datasets required to enable open-set assistive intelligence. In particular, we show that performant models benefit from datasets that cover different aspects of assistance, including multimodal grounding, defect inference, and exposure to diverse scenarios.
Figures & tables
| Held-Out Defects | Task Generalization | |||
| Baseline | Coaching | Corrections | Coaching | Corrections |
| Behavior Critic | 21.00 | 20.40 | 34.21 | 9.17 |
| Behavior Critic + Summaries | 55.70 | 19.80 | 71.05 | 15.83 |
| Ours 1B | 76.60 | 55.70 | 50.88 | 50.83 |
| Ours 8B | 77.80 | 54.60 | 85.96 | 56.67 |
| Held-out Defects | Task Generalization | |||
| Baseline | Coaching | Corrections | Coaching | Corrections |
| Behavior Critic | 21.00 | 20.40 | 34.21 | 9.17 |
| Ours 8B | 34.30 | 39.70 | 0.00 | 55.00 |
| Ours 8B + reasoning | 18.10 | 32.60 | 8.77 | 43.30 |
| Unseen Defects | Novel Tasks | |||
| Training Data | Coaching | Corrections | Coaching | Corrections |
| 1B ( ) | 76.60 | 55.70 | 50.88 | 50.83 |
| 1B ( ) | 38.10 | – | 54.39 | – |
| 1B ( ) | – | 29.40 | – | 33.34 |
| Baseline | Coaching | Corrections |
| 1B | 50.88 | 50.83 |
| 1B ( ) | 58.77 | 53.33 |
| 1B ( ) | 69.30 | 46.67 |
| 1B ( ) | 46.49 | 42.50 |
| 1B ( ) | 78.95 | 52.50 |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Model-name | Training Datasets | Data Aug | Learning rate | Training Steps | Per-device batch size | Weight Decay |
| Ours (1B) | No | 15000 | 1 | 0.01 | ||
| 1B + grounding datasets | Yes | 16500 | 1 | 0.01 | ||
| 1B Coaching Only | Yes | 5000 | 1 | 0.01 | ||
| 1B Correction Only | No | 5000 | 1 | 0.01 | ||
| Ours (1B) + I-QA | Yes | 16500 | 1 | 0.01 | ||
| Ours (1B) + T-QA | Yes | 16500 | 1 | 0.01 |
Assistive Robotics
Assistive robotics develops robots and wearable devices that support mobility, daily activities, rehabilitation, and care.
See all 207 papers