Hugging Face Daily PapersTrung Dao, Sankalp Yamsani, Jaden Park1 min readpaperadvanced
Think Like a World Model, Act Like a VLA: Distilling World-Model Representations into Compact Robot Policies
Summary
Distill frozen world‑model features into a Vision‑Language‑Action policy via a single alignment loss; no teacher at train time, no extra runtime cost. 0.8 B student runs 32 ms / 1.86 GB on RTX 5090, hits 97.9 % on LIBERO and improves RoboCasa‑GR1 from 48.2 % to 50.5 %, with transfer to real single‑arm and bimanual robots.
- Add a feature‑alignment term: run a frozen world model on training frames once, cache its internal features, and train the VLA student to match them.
- The alignment projector is discarded after training; the deployed policy is identical to the baseline VLA, so speed and memory gains come solely from better representations.
- Runtime: 32 ms per decision, 1.86 GB VRAM on a consumer RTX 5090 (no extra test‑time compute).
- Performance: 0.8 B parameter student reaches 97.9 % on LIBERO; RoboCasa‑GR1 manipulation improves from 48.2 % to 50.5 %; same gains on real single‑arm and bimanual hardware.
World‑model grounding is usually too expensive for closed‑loop control. This work shows you can inherit the grounding via static feature alignment, keeping inference cheap while still reaping the robustness benefits—directly applicable to any VLA‑style robot policy.
8/10