Vista: A Generalizable Driving World Model with High Fidelity and Versatile Controllability
About
World models can foresee the outcomes of different actions, which is of paramount importance for autonomous driving. Nevertheless, existing driving world models still have limitations in generalization to unseen environments, prediction fidelity of critical details, and action controllability for flexible application. In this paper, we present Vista, a generalizable driving world model with high fidelity and versatile controllability. Based on a systematic diagnosis of existing methods, we introduce several key ingredients to address these limitations. To accurately predict real-world dynamics at high resolution, we propose two novel losses to promote the learning of moving instances and structural information. We also devise an effective latent replacement approach to inject historical frames as priors for coherent long-horizon rollouts. For action controllability, we incorporate a versatile set of controls from high-level intentions (command, goal point) to low-level maneuvers (trajectory, angle, and speed) through an efficient learning strategy. After large-scale training, the capabilities of Vista can seamlessly generalize to different scenarios. Extensive experiments on multiple datasets show that Vista outperforms the most advanced general-purpose video generator in over 70% of comparisons and surpasses the best-performing driving world model by 55% in FID and 27% in FVD. Moreover, for the first time, we utilize the capacity of Vista itself to establish a generalizable reward for real-world action evaluation without accessing the ground truth actions.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Video Generation | nuScenes (val) | FVD89.4 | 37 | |
| Video Prediction | nuScenes (val) | FID6.9 | 16 | |
| Future Semantic Segmentation (Mid-term) | Cityscapes (val) | mIoU (All Classes)53.2 | 12 | |
| Future Semantic Segmentation (Short-term) | Cityscapes (val) | mIoU (All Classes)0.657 | 12 | |
| Multi-view Driving Video Generation | NuScenes v1.0 (test) | FVD112.7 | 11 | |
| Action-conditioned future prediction | Waymo Open (val) | Trajectory Difference (Action-free)5.68 | 4 | |
| Depth Forecasting | Cityscapes short-term | Delta 1 Accuracy93.4 | 3 | |
| Depth Forecasting | Cityscapes mid-term | Delta 187.4 | 3 | |
| Semantic segmentation | Cityscapes short-term | Overall Score65.7 | 3 | |
| Semantic segmentation | Cityscapes mid-term | mIoU (All)53.2 | 3 |