MagicDrive: Street View Generation with Diverse 3D Geometry Control

TL;DR

MagicDrive employs multi-view 3D geometry control with diffusion models, achieving high-fidelity street view synthesis for perception tasks.

cs.CV 🔴 Advanced 2023-10-04 50 views
Ruiyuan Gao Kai Chen Enze Xie Lanqing Hong Zhenguo Li Dit-Yan Yeung Qiang Xu
street view 3D control diffusion models multi-view consistency autonomous driving

Key Findings

Methodology

MagicDrive builds upon pre-trained stable diffusion, integrating multi-level encoding of scene elements—road maps, 3D bounding boxes, and camera poses—via separate encoders. Cross-view attention modules ensure multi-camera consistency. The model combines scene text, geometric data, and camera parameters using Fourier embeddings, CLIP text encodings, and an enhanced UNet architecture. It employs classifier-free guidance, randomly dropping conditions during training to improve controllability. This design allows precise manipulation of scene attributes and geometry, supporting multi-conditional generation with high realism.

Key Results

  • On nuScenes, MagicDrive achieves an FID of 16.20, outperforming BEVGen and BEVControl, indicating superior image quality. It demonstrates state-of-the-art performance in BEV segmentation (vehicle mIoU 61.05) and 3D detection (AP 27.01). Synthetic data augmentation improves perception model accuracy, especially at higher resolutions, validating its practical utility.
  • Multi-level encoding enables detailed control over scene attributes like elevation, object placement, and weather. The cross-view attention module maintains spatial consistency across multiple camera views, essential for realistic street scene synthesis. Ablation studies confirm these components' critical roles. The generated scenes support downstream perception tasks, significantly boosting detection and segmentation performance.
  • The approach generalizes well across diverse scenarios, maintaining high fidelity and geometric accuracy. It surpasses existing methods in controllability and realism, offering a versatile tool for virtual data generation in autonomous driving, simulation, and training pipelines.

Significance

This work advances street scene synthesis by integrating multi-view 3D geometric control into diffusion models, overcoming limitations of BEV-only approaches. It provides a flexible framework for generating realistic, controllable, multi-perspective street views, crucial for training perception systems without extensive real-world data collection. The ability to manipulate scene attributes and geometry enhances the diversity and robustness of training datasets, accelerating progress in autonomous driving and scene understanding. The methodology also opens avenues for future research in multi-modal, multi-view generative modeling, with broad implications for virtual simulation, urban planning, and intelligent transportation systems.

Technical Contribution

The paper introduces a novel multi-level encoding scheme for scene text, road maps, and 3D bounding boxes, combined with a cross-view attention mechanism to ensure multi-camera consistency. It extends the diffusion-based generative framework with a flexible, multi-conditional control strategy, leveraging Fourier embeddings and CLIP encodings. The model’s architecture allows independent geometric control and scene attribute manipulation, surpassing prior BEV-centric methods. Its training strategy with classifier-free guidance and augmentation techniques enhances controllability and robustness, setting new standards for multi-view scene synthesis.

Novelty

This is the first work to incorporate multi-view 3D geometric control within a diffusion model for street scene synthesis. Unlike prior BEV-dependent methods, MagicDrive encodes scene geometry separately and employs cross-view attention to maintain multi-camera consistency. Its multi-condition control, combining text, geometry, and camera parameters, allows precise scene manipulation, representing a significant leap forward in controllable, high-fidelity virtual scene generation for autonomous driving.

Limitations

  • The model’s performance degrades in highly complex or adverse weather scenarios, limited by training data diversity and model capacity.
  • High-resolution generation incurs substantial computational costs, hindering real-time applications.
  • Fine-grained geometric details, such as complex object shapes, are not fully captured, requiring further geometric modeling and multi-modal integration.

Future Work

Future directions include enriching geometric representations with detailed shape modeling, integrating multi-modal sensor data, and optimizing model efficiency for real-time deployment. Extending the framework to handle more diverse and challenging scenarios, such as adverse weather or dense traffic, is also planned. Additionally, exploring unsupervised or semi-supervised training strategies could further improve generalization and scalability, broadening its application scope in virtual simulation, urban planning, and intelligent transportation.

AI Executive Summary

The rapid development of autonomous driving hinges on access to high-quality, diverse, and controllable street scene data. Traditional data collection is costly and limited in scope, prompting researchers to explore synthetic data generation. Early efforts relied on 2D layout-based methods, which struggled to capture the complex 3D structure of urban environments. Recent advances in diffusion models have opened new possibilities for high-fidelity image synthesis, but most approaches focus on single-view or limited geometric control.

This paper introduces MagicDrive, a novel framework that leverages multi-view 3D geometric control within a diffusion-based generative model. By independently encoding scene text, road maps, and 3D bounding boxes, and integrating them through cross-attention modules, MagicDrive ensures multi-camera consistency and precise geometric manipulation. The model employs Fourier embeddings and CLIP text encodings to fuse semantic and geometric information, enabling detailed scene attribute control, such as weather and time of day.

Experimental results on nuScenes demonstrate that MagicDrive surpasses existing methods like BEVGen and BEVControl in image quality (FID 16.20) and perception task performance. Its ability to generate realistic, multi-view street scenes with accurate geometry significantly enhances downstream tasks like BEV segmentation and 3D object detection. Synthetic data generated by MagicDrive improves perception model accuracy, validating its practical utility.

The framework’s key innovation lies in multi-level encoding and cross-view attention, which together address the longstanding challenge of multi-camera scene synthesis. This approach paves the way for more realistic virtual environments, reducing reliance on costly real-world data collection. Future work will focus on expanding geometric detail, optimizing computational efficiency, and broadening application scenarios, promising a transformative impact on autonomous driving and urban scene understanding.

Deep Dive

Abstract

Recent advancements in diffusion models have significantly enhanced the data synthesis with 2D control. Yet, precise 3D control in street view generation, crucial for 3D perception tasks, remains elusive. Specifically, utilizing Bird's-Eye View (BEV) as the primary condition often leads to challenges in geometry control (e.g., height), affecting the representation of object shapes, occlusion patterns, and road surface elevations, all of which are essential to perception data synthesis, especially for 3D object detection tasks. In this paper, we introduce MagicDrive, a novel street view generation framework, offering diverse 3D geometry controls including camera poses, road maps, and 3D bounding boxes, together with textual descriptions, achieved through tailored encoding strategies. Besides, our design incorporates a cross-view attention module, ensuring consistency across multiple camera views. With MagicDrive, we achieve high-fidelity street-view image & video synthesis that captures nuanced 3D geometry and various scene descriptions, enhancing tasks like BEV segmentation and 3D object detection.

cs.CV cs.AI