GeoDrive: 3D Geometry-Informed Driving World Model with Precise Action Control

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chen, Anthony, Zheng, Wenzhao, Wang, Yida, Zhang, Xueyang, Zhan, Kun, Jia, Peng, Keutzer, Kurt, Zhang, Shanghang
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910973621698560
author Chen, Anthony
Zheng, Wenzhao
Wang, Yida
Zhang, Xueyang
Zhan, Kun
Jia, Peng
Keutzer, Kurt
Zhang, Shanghang
author_facet Chen, Anthony
Zheng, Wenzhao
Wang, Yida
Zhang, Xueyang
Zhan, Kun
Jia, Peng
Keutzer, Kurt
Zhang, Shanghang
contents Recent advancements in world models have revolutionized dynamic environment simulation, allowing systems to foresee future states and assess potential actions. In autonomous driving, these capabilities help vehicles anticipate the behavior of other road users, perform risk-aware planning, accelerate training in simulation, and adapt to novel scenarios, thereby enhancing safety and reliability. Current approaches exhibit deficiencies in maintaining robust 3D geometric consistency or accumulating artifacts during occlusion handling, both critical for reliable safety assessment in autonomous navigation tasks. To address this, we introduce GeoDrive, which explicitly integrates robust 3D geometry conditions into driving world models to enhance spatial understanding and action controllability. Specifically, we first extract a 3D representation from the input frame and then obtain its 2D rendering based on the user-specified ego-car trajectory. To enable dynamic modeling, we propose a dynamic editing module during training to enhance the renderings by editing the positions of the vehicles. Extensive experiments demonstrate that our method significantly outperforms existing models in both action accuracy and 3D spatial awareness, leading to more realistic, adaptable, and reliable scene modeling for safer autonomous driving. Additionally, our model can generalize to novel trajectories and offers interactive scene editing capabilities, such as object editing and object trajectory control.
format Preprint
id arxiv_https___arxiv_org_abs_2505_22421
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle GeoDrive: 3D Geometry-Informed Driving World Model with Precise Action Control
Chen, Anthony
Zheng, Wenzhao
Wang, Yida
Zhang, Xueyang
Zhan, Kun
Jia, Peng
Keutzer, Kurt
Zhang, Shanghang
Computer Vision and Pattern Recognition
Robotics
Recent advancements in world models have revolutionized dynamic environment simulation, allowing systems to foresee future states and assess potential actions. In autonomous driving, these capabilities help vehicles anticipate the behavior of other road users, perform risk-aware planning, accelerate training in simulation, and adapt to novel scenarios, thereby enhancing safety and reliability. Current approaches exhibit deficiencies in maintaining robust 3D geometric consistency or accumulating artifacts during occlusion handling, both critical for reliable safety assessment in autonomous navigation tasks. To address this, we introduce GeoDrive, which explicitly integrates robust 3D geometry conditions into driving world models to enhance spatial understanding and action controllability. Specifically, we first extract a 3D representation from the input frame and then obtain its 2D rendering based on the user-specified ego-car trajectory. To enable dynamic modeling, we propose a dynamic editing module during training to enhance the renderings by editing the positions of the vehicles. Extensive experiments demonstrate that our method significantly outperforms existing models in both action accuracy and 3D spatial awareness, leading to more realistic, adaptable, and reliable scene modeling for safer autonomous driving. Additionally, our model can generalize to novel trajectories and offers interactive scene editing capabilities, such as object editing and object trajectory control.
title GeoDrive: 3D Geometry-Informed Driving World Model with Precise Action Control
topic Computer Vision and Pattern Recognition
Robotics
url https://arxiv.org/abs/2505.22421