Vista: A Generalizable Driving World Model with High Fidelity and Versatile Controllability

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Gao, Shenyuan, Yang, Jiazhi, Chen, Li, Chitta, Kashyap, Qiu, Yihang, Geiger, Andreas, Zhang, Jun, Li, Hongyang
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929561729499136
author Gao, Shenyuan
Yang, Jiazhi
Chen, Li
Chitta, Kashyap
Qiu, Yihang
Geiger, Andreas
Zhang, Jun
Li, Hongyang
author_facet Gao, Shenyuan
Yang, Jiazhi
Chen, Li
Chitta, Kashyap
Qiu, Yihang
Geiger, Andreas
Zhang, Jun
Li, Hongyang
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2405_17398
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Vista: A Generalizable Driving World Model with High Fidelity and Versatile Controllability
Gao, Shenyuan
Yang, Jiazhi
Chen, Li
Chitta, Kashyap
Qiu, Yihang
Geiger, Andreas
Zhang, Jun
Li, Hongyang
Computer Vision and Pattern Recognition
Artificial Intelligence
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.
title Vista: A Generalizable Driving World Model with High Fidelity and Versatile Controllability
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2405.17398