Rethinking Driving World Model as Synthetic Data Generator for Perception Tasks
By: Kai Zeng , Zhanqian Wu , Kaixin Xiong and more
Potential Business Impact:
Makes self-driving cars better at seeing tricky situations.
Recent advancements in driving world models enable controllable generation of high-quality RGB videos or multimodal videos. Existing methods primarily focus on metrics related to generation quality and controllability. However, they often overlook the evaluation of downstream perception tasks, which are $\mathbf{really\ crucial}$ for the performance of autonomous driving. Existing methods usually leverage a training strategy that first pretrains on synthetic data and finetunes on real data, resulting in twice the epochs compared to the baseline (real data only). When we double the epochs in the baseline, the benefit of synthetic data becomes negligible. To thoroughly demonstrate the benefit of synthetic data, we introduce Dream4Drive, a novel synthetic data generation framework designed for enhancing the downstream perception tasks. Dream4Drive first decomposes the input video into several 3D-aware guidance maps and subsequently renders the 3D assets onto these guidance maps. Finally, the driving world model is fine-tuned to produce the edited, multi-view photorealistic videos, which can be used to train the downstream perception models. Dream4Drive enables unprecedented flexibility in generating multi-view corner cases at scale, significantly boosting corner case perception in autonomous driving. To facilitate future research, we also contribute a large-scale 3D asset dataset named DriveObj3D, covering the typical categories in driving scenarios and enabling diverse 3D-aware video editing. We conduct comprehensive experiments to show that Dream4Drive can effectively boost the performance of downstream perception models under various training epochs. Project: $\href{https://wm-research.github.io/Dream4Drive/}{this\ https\ URL}$
Similar Papers
Cosmos-Drive-Dreams: Scalable Synthetic Driving Data Generation with World Foundation Models
CV and Pattern Recognition
Creates fake driving videos to train self-driving cars.
RealDriveSim: A Realistic Multi-Modal Multi-Task Synthetic Dataset for Autonomous Driving
CV and Pattern Recognition
Creates realistic fake driving scenes for self-driving cars.
MoVieDrive: Multi-Modal Multi-View Urban Scene Video Generation
CV and Pattern Recognition
Makes self-driving cars see in 3D and color.