UniDriveDreamer: A Single-Stage Multimodal World Model for Autonomous Driving

Guosheng Zhao, Yaozeng Wang, Xiaofeng Wang, Zheng Zhu, Tingdong Yu +8 more
2/2/2026
cs.CV

No confident code implementation found for this paper.

Abstract

World models have demonstrated significant promise for data synthesis in autonomous driving. However, existing methods predominantly concentrate on single-modality generation, typically focusing on either multi-camera video or LiDAR sequence synthesis. In this paper, we propose UniDriveDreamer, a single-stage unified multimodal world model for autonomous driving, which directly generates multimodal future observations without relying on intermediate representations or cascaded modules. Our framework introduces a LiDAR-specific variational autoencoder (VAE) designed to encode input LiDAR sequences, alongside a video VAE for multi-camera images. To ensure cross-modal compatibility and training stability, we propose Unified Latent Anchoring (ULA), which explicitly aligns the latent distributions of the two modalities. The aligned features are fused and processed by a diffusion transformer that jointly models their geometric correspondence and temporal evolution. Additionally, structured scene layout information is projected per modality as a conditioning signal to guide the synthesis. Extensive experiments demonstrate that UniDriveDreamer outperforms previous state-of-the-art methods in both video and LiDAR generation, while also yielding measurable improvements in downstream

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{zhao2026unidrivedreamer,
  title  = {UniDriveDreamer: A Single-Stage Multimodal World Model for Autonomous Driving},
  author = {Guosheng Zhao and Yaozeng Wang and Xiaofeng Wang and Zheng Zhu and Tingdong Yu and Guan Huang and Yongchen Zai and Ji Jiao and Changliang Xue and Xiaole Wang and Zhen Yang and Futang Zhu and Xingang Wang},
  year   = {2026},
  eprint = {2602.02002},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2602.02002v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2602.02002.svg)](https://codeofpaper.com/paper/2602.02002)

Discussion