arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.02002cs.CV
收藏

UniDriveDreamer: 一种用于自动驾驶的单阶段多模态世界模型

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

GigaAI(极佳科技);CASIA;BYD

Guosheng Zhao, Yaozeng Wang, Xiaofeng Wang, Zheng Zhu, Tingdong Yu, Guan Huang, Yongchen Zai, Ji Jiao, Changliang Xue, Xiaole Wang, Zhen Yang, Futang Zhu, Xingang Wang

AI总结:

UniDriveDreamer是一种用于自动驾驶的单阶段多模态世界模型,通过统一的多模态生成方法提升视频和激光雷达数据合成的性能。

摘要:

世界模型在自动驾驶的数据合成中展现出了显著的潜力。然而,现有方法大多集中在单模态生成上,通常侧重于多摄像头视频或激光雷达序列合成。在本文中,我们提出UniDriveDreamer,一种用于自动驾驶的单阶段统一多模态世界模型,该模型可以直接生成多模态的未来观测,而无需依赖中间表示或级联模块。我们的框架引入了一个专门针对激光雷达的变分自编码器(VAE),用于编码输入的激光雷达序列,以及一个视频VAE用于多摄像头图像。为了确保跨模态兼容性和训练稳定性,我们提出了统一潜在锚定(ULA),它明确对齐两种模态的潜在分布。对齐后的特征通过扩散变换器进行融合和处理,该变换器联合建模它们的几何对应关系和时间演变。此外,结构化的场景布局信息会按模态投影为条件信号以指导合成。大量实验表明,UniDriveDreamer在视频和激光雷达生成方面均优于先前的最先进方法,同时在下游任务中也产生了可测量的改进。

英文摘要:

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

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑