Multi-Teacher On-Policy Distillation for Flow-Based Vision-Language-Action Models
Organizations: Huawei Technologies Co., Ltd. · Tsinghua University
Abstract
Specialized vision-language-action (VLA) policies acquire strong domain-specific skills, yet consolidating these skills into one shared policy remains challenging. On-policy distillation (OPD) transfers teacher knowledge on inputs induced by the student, addressing the distribution mismatch encountered during deployment. For flow-based VLAs, this mismatch arises at two levels: environment observations and intermediate action-generation states. We propose a two-level on-policy distillation framework that couples student-driven environment interaction with teacher velocity supervision along student-generated action flow trajectories. Frozen domain specialists jointly supervise one student. After each update, the student collects new environment experience and generates fresh action-flow trajectories. Starting from a joint supervised fine-tuning (SFT) initialization, Velocity Loss OPD yields absolute success-rate gains of 22.55% on LIBERO-40, 33.4% on CALVIN five-instruction sequences, and 22.6% on Meta-World MT50. Compared with Action Loss OPD and Teacher-rollout SFT, it achieves absolute success-rate gains of 6.4-9.5% on LIBERO-40, 9.7-18.3% on CALVIN, and 3.2-4.2% on Meta-World MT50. Analysis by sequence length further shows larger success-rate gains on longer CALVIN sequences. These results demonstrate effective multi-domain capability consolidation through two-level on-policy supervision, with a single student deployed across domains.
Figures & tables
| LIBERO | MT50 | CALVIN | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Method | All-40 | Spatial | Object | Goal | LIBERO-10 | SR | Length | |||||
| Domain specialists | 98.00 | 98.8 | 98.4 | 97.8 | 97.0 | 76.2 | 99.7 | 97.4 | 94.4 | 91.5 | 86.8 | 4.698 |
| Shared SFT | 53.15 | 74.2 | 76.2 | 53.2 | 9.0 | 38.0 | 93.4 | 77.0 | 62.4 | 51.2 | 39.5 | 3.235 |
| Teacher-rollout SFT | 69.30 | 88.4 | 94.0 | 62.2 | 32.6 | 57.4 | 98.1 | 92.2 | 81.3 | 72.6 | 63.2 | 4.074 |
| Action Loss OPD | 66.20 | 94.6 | 77.2 | 60.0 | 33.0 | 56.4 | 95.6 | 85.6 | 74.6 | 64.0 | 54.6 | 3.744 |
| Velocity Loss OPD | 75.70 | 94.2 | 98.4 | 63.8 | 46.4 | 60.6 | 99.0 | 94.7 | 88.4 | 81.2 | 72.9 | 4.362 |
Offline Semantic Guidance for Efficient Vision-Language-Action Policy Distillation
libero\_goal. Additional analysis indicates that phase-level supervision and multi-frame directional cues make the student less sensitive to noisy teacher actions, such as erroneous high-frequency gripper changes. Overall, VLA-AD demonstrates that offline semantic guidance from VLMs can substantially improve the efficiency, robustness, and deployability of VLA policy distillation.