Think Like a World Model, Act Like a VLA: Distilling World-Model Representations into Compact Robot Policies
This research proposes a method to distill world-model representations into compact robot policies by aligning student policies with a frozen world model during training. The approach achieves performance improvements on tasks like LIBERO and RoboCasa-GR1, with a 0.8B student reaching 97.9% on LIBERO and improving from 48.2% to 50.5% on RoboCasa-GR1. The method is also effective on real hardware, demonstrating broad representational prior rather than fragile network alignment.