DriveDreamer: Towards Real-world-driven World Models for Autonomous Driving

Our team is actively working towards releasing the code for this project. However, due to the involvement of company patents and ongoing product initiatives, we are currently undergoing internal reviews. We anticipate that the research code will be released after March, along with the introduction of DriveDreamer V2.

We appreciate your patience and understanding as we navigate the necessary processes.

Project Page | Paper

Abstract

World models, especially in autonomous driving, are trending and drawing extensive attention due to its capacity for comprehending driving environments. The established world model holds immense potential for the generation of high-quality driving videos, and driving policies for safe maneuvering. However, a critical limitation in relevant research lies in its predominant focus on gaming environments or simulated settings, thereby lacking the representation of real-world driving scenarios. Therefore, we introduce DriveDreamer, a pioneering world model entirely derived from real-world driving scenarios. Regarding that modeling the world in intricate driving scenes entails an overwhelming search space, we propose harnessing the powerful diffusion model to construct a comprehensive representation of the complex environment. Furthermore, we introduce a two-stage training pipeline. In the initial phase, DriveDreamer acquires a deep understanding of structured traffic constraints, while the subsequent stage equips it with the ability to anticipate future states. The proposed DriveDreamer is the first world model established from real-world driving scenarios. We instantiate DriveDreamer on the challenging nuScenes benchmark, and extensive experiments verify that DriveDreamer empowers precise, controllable video generation that faithfully captures the structural constraints of real-world traffic scenarios.
Additionally, DriveDreamer enables the generation of realistic and reasonable driving policies, opening avenues for interaction and practical applications.

News

• [2023/09/17] Repository Initialization.

Demo

Diverse Driving Video Generation.

diverse.mp4

Driving Video Generation with Traffic Condition and Different Text Prompts (Sunny, Rainy, Night).

text_driven.mp4

Future Driving Video Generation with Action Interaction.

action_interact_concat.mp4

Future Driving Action Generation.

action_pred_concat.mp4

DriveDreamer Framework

Bibtex

If this work is helpful for your research, please consider citing the following BibTeX entry.

@article{wang2023drivedreamer,
      title={DriveDreamer: Towards Real-world-driven World Models for Autonomous Driving}, 
      author={Xiaofeng Wang and Zheng Zhu and Guan Huang and Xinze Chen and Jiwen Lu},
      journal={arXiv preprint arXiv:2309.09777},
      year={2023}
}