TILDE: Teacher-in-the-Loop Distillation from Embodied Foundation Models
Organizations: The University of Hong Kong
Abstract
Large-scale action pretraining equips Embodied Foundation Models (EFMs) with strong manipulation capabilities, with particular strengths in generalization and closed-loop robustness, but reproducing such pretraining requires substantial robot data and computation. To transfer these capabilities from pretrained EFMs to policies without action pretraining, we introduce TILDE (Teacher-In-the-Loop Distillation from EFMs), which synergizes on-policy distillation at student-induced states with learning from teacher-executed recovery trajectories. We further propose AutoDAgger, an autonomous data aggregation pipeline in which a robotic reward model is utilized to monitor task progress and trigger teacher intervention, thereby collecting recovery experience without human monitoring or manual control. AutoDAgger accelerates recovery-trajectory collection by more than compared with the human-in-the-loop method. By unifying learning from demonstrations, on-policy teacher supervision, and teacher's corrective behavior, TILDE improves task performance by up to over behavior cloning across three real-world tasks and supports effective cross-architecture distillation.
Figures & tables
| Data Source | Distill | Fold Dishcloth | Pick Pen | Pick Beverage | |||
|---|---|---|---|---|---|---|---|
| #Episodes | Score | #Episodes | Score | #Episodes | Score | ||
| Human Demo | ✗ | 200 | 20.8 | 100 | 37.0 | 100 | 16.7 |
| +Extra Demo | ✗ | 230 | 27.8 (+ 7.0) | 130 | 45.4 (+ 8.4) | 130 | 44.2 (+ 27.5) |
| +AutoDAgger | ✗ | 230 | 48.6 (+ 27.8) | 130 | 63.0 (+ 26.0) | 130 | 50.0 (+ 33.3) |
| Human Demo | ✓ | 200 | 27.8 | 100 | 45.4 | 100 | 26.7 |
| +Extra Demo | ✓ | 230 | 29.2 (+ 1.4) | 130 | 54.6 (+ 9.2) | 130 | 48.3 (+ 21.6) |
| Method | Distill | Fold Dishcloth | Pick Pen | Pick Beverage | ||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Score | Rec. Time | Rec. SR(%) | Score | Rec. Time | Rec. SR(%) | Score | Rec. Time | Rec. SR(%) | ||
| HG-DAgger | ✗ | 34.7 | 54.4s | 76.7 | 75.9 | 29.4s | 76.7 | 60.0 | 32.2s | 90.0 |
| ✓ | 65.3 | 80.6 | 63.3 | |||||||
| AutoDAgger | ✗ | 48.6 | 21.5s | 73.3 | 63.0 | 13.2s | 66.7 | 50.0 | 12.4s | 90.0 |
| ✓ | 63.9 | (2.53 ) | 80.6 | (2.23 ) | 69.2 | (2.60 ) | ||||
Appendix figures & tables12 assets
Supplementary material from the paper’s appendix.
Appendix
| Task | (s) | (s) | ||
|---|---|---|---|---|
| Student | Teacher | Student | Teacher | |
| Fold Dishcloth | 6 | 14 | 5 | 5 |
| Pick Pen | 3 | 8 | 4 | 4 |
| Pick Beverage | 2 | 6 | 4 | 4 |
| Model | X-VLA | |||
|---|---|---|---|---|
| Task | AgileX | LIBERO | AgileX | LIBERO |
| Prediction Horizon | 50 | 10 | 30 | 30 |
| Action Space | Relative Joint | Delta EEF | Abs Joint | Abs EE6D |
| Global Batch Size | 128 | 256 | 128 | 128 |
| Training Steps | 50k | 30k | 50k | 60k |
| Optimizer | AdamW | AdamW | AdamW | AdamW |
| Model | VLA-Adapter-Pro | SmolVLA | |
|---|---|---|---|
| Task | AgileX | LIBERO | LIBERO |
| Prediction Horizon | 30 | 8 | 10 |
| Action Space | Abs Joint | Delta EEF | Delta EEF |
| Global Batch Size | 128 | 64 | 64 |
| Training Steps | 50k | 100k | 100k |
| Optimizer | AdamW | AdamW | AdamW |
| Method | |||
|---|---|---|---|
| BC Baseline | BC | – | – |
| Offline Distill | BC+KD | – | – |
| OPD | BC+KD | KD | – |
| Recovery BC | BC | – | BC |
| Recovery Distill | BC+KD | – | BC+KD |
| TILDE | BC+KD | KD | BC+KD |
| Prediction | Ground-truth outcome | Total | |
|---|---|---|---|
| True | False | ||
| Positive | 8 (TP) | 1 (FP) | 9 |
| Negative | 2 (FN) | 9 (TN) | 11 |
| Total | 10 | 10 | 20 |
| Method | Fold Dishcloth | Pick Pen | Pick Beverage | |||
|---|---|---|---|---|---|---|
| Mean | 96% CI | Mean | 96% CI | Mean | 96% CI | |
| Teacher | 87.5 | (69.4, 98.6) | 87.0 | (78.7, 94.4) | 79.2 | (65.0, 91.7) |
| BC Baseline | 20.8 | (2.8, 41.7) | 37.0 | (22.2, 51.9) | 16.7 | (8.3, 25.8) |
| Offline Distill | 27.8 | (6.9, 51.4) | 45.4 | (31.5, 60.2) | 26.7 | (14.2, 40.0) |
| OPD | 43.1 | (16.7, 69.4) | 58.3 | (45.4, 70.4) | 40.0 | (25.0, 55.0) |
| Recovery BC | 48.6 | (22.2, 75.0) | 63.0 | (48.1, 76.9) | 50.0 | (32.5, 67.5) |
| Data Source | Distill | Fold Dishcloth | Pick Pen | Pick Beverage | |||
|---|---|---|---|---|---|---|---|
| Mean | 96% CI | Mean | 96% CI | Mean | 96% CI | ||
| Human Demo | ✗ | 20.8 | (2.8, 41.7) | 37.0 | (22.2, 51.9) | 16.7 | (8.3, 25.8) |
| +Extra Demo | ✗ | 27.8 | (6.9, 51.4) | 45.4 | (27.8, 63.0) | 44.2 | (28.3, 60.0) |
| +AutoDAgger | ✗ | 48.6 | (22.2, 75.0) | 63.0 | (48.1, 76.9) | 50.0 | (32.5, 67.5) |
| Human Demo | ✓ | 27.8 | (6.9, 51.4) | 45.4 | (31.5, 60.2) | 26.7 | (14.2, 40.0) |
| +Extra Demo | ✓ | 29.2 | (9.7, 51.4) | 54.6 | (40.7, 68.5) | 48.3 | (33.3, 63.3) |
| Method | Distill | Fold Dishcloth | Pick Pen | Pick Beverage | |||
|---|---|---|---|---|---|---|---|
| Mean | 96% CI | Mean | 96% CI | Mean | 96% CI | ||
| HG-DAgger | ✗ | 34.7 | (11.1, 59.7) | 75.9 | (60.2, 89.8) | 60.0 | (46.7, 73.3) |
| ✓ | 65.3 | (33.3, 90.3) | 80.6 | (68.5, 90.7) | 63.3 | (48.3, 77.5) | |
| AutoDAgger | ✗ | 48.6 | (22.2, 75.0) | 63.0 | (48.1, 76.9) | 50.0 | (32.5, 67.5) |
| ✓ | 63.9 | (38.9, 87.5) | 80.6 | (72.2, 88.0) | 69.2 | (53.3, 83.3) | |
| Task | HG-DAgger | AutoDAgger | ||||
|---|---|---|---|---|---|---|
| Rec. Time | Rec. SR (%) | 96% CI | Rec. Time | Rec. SR (%) | 96% CI | |
| Fold Dishcloth | s | 76.7 | (60.0, 90.0) | s | 73.3 | (56.7, 90.0) |
| Pick Pen | s | 76.7 | (60.0, 90.0) | s | 66.7 | (50.0, 83.3) |
| Pick Beverage | s | 90.0 | (76.7, 100.0) | s | 90.0 | (76.7, 100.0) |
Knowledge Distillation
Knowledge distillation transfers useful behavior or representations from larger or specialized models into more compact or adaptable learners.
See all 814 papers