DriftWorld: Fast World Modeling through Drifting
DriftWorld replaces diffusion with drifting generative models for single-step world model inference at 30+ FPS — 17x faster than diffusion baselines — achieving state-of-the-art decision-making across multiple robotic manipulation benchmarks.
Paper · arXiv:2607.15065Background
Predictive world models enable robots to plan by imagining action outcomes, but diffusion-based world models require expensive multistep sampling for each rollout, severely limiting the scale of action search at inference time. This bottleneck has prevented diffusion world models from achieving practical real-time performance in robotic control.
Core Innovation
DriftWorld introduces drifting generative models as an alternative to diffusion:
- Training: Learns an action-conditioned drift process rather than a denoising process
- Inference: Generates future frames from observation + action sequence in a single forward pass — no iterative sampling
- Achieves 30+ FPS inference, 17× faster on average than diffusion baselines
Results
Evaluated across standard vision-based robotic manipulation benchmarks:
| Benchmark | Result |
|---|---|
| Bridge-V2, RT-1, Language Table, Push-T, Robomimic | State-of-the-art decision performance |
| Inference speed | 30+ FPS (17× faster than diffusion baselines) |
| Policy ranking correlation | Up to 0.99 with ground-truth rollouts |
Key finding: DriftWorld serves both as an online planner and as an offline simulator for ranking real-world robot policies, with rollout-based scores correlating at up to 0.99 with actual policy performance.
Limitations
Performance under highly multimodal future distributions (multiple equally plausible action outcomes) is not fully explored. The evaluation focuses on manipulation tasks; applicability to high-dynamic motion planning scenarios remains to be validated. Generalization across diverse simulation environments also needs further study.
Industry Implications
DriftWorld dramatically reduces the inference cost of world models in robotic systems, making real-time imagine-and-plan cycles practical. This has direct application value for autonomous robots requiring real-time decisions — warehouse logistics, service robotics, and manufacturing. As an offline policy evaluation tool, it enables robotics teams to screen policies before deployment, reducing expensive physical testing.