English

DriveX: Omni Scene Modeling for Learning Generalizable World Knowledge in Autonomous Driving

Computer Vision and Pattern Recognition 2025-05-27 v1

Abstract

Data-driven learning has advanced autonomous driving, yet task-specific models struggle with out-of-distribution scenarios due to their narrow optimization objectives and reliance on costly annotated data. We present DriveX, a self-supervised world model that learns generalizable scene dynamics and holistic representations (geometric, semantic, and motion) from large-scale driving videos. DriveX introduces Omni Scene Modeling (OSM), a module that unifies multimodal supervision-3D point cloud forecasting, 2D semantic representation, and image generation-to capture comprehensive scene evolution. To simplify learning complex dynamics, we propose a decoupled latent world modeling strategy that separates world representation learning from future state decoding, augmented by dynamic-aware ray sampling to enhance motion modeling. For downstream adaptation, we design Future Spatial Attention (FSA), a unified paradigm that dynamically aggregates spatiotemporal features from DriveX's predictions to enhance task-specific inference. Extensive experiments demonstrate DriveX's effectiveness: it achieves significant improvements in 3D future point cloud prediction over prior work, while attaining state-of-the-art results on diverse tasks including occupancy prediction, flow estimation, and end-to-end driving. These results validate DriveX's capability as a general-purpose world model, paving the way for robust and unified autonomous driving frameworks.

Keywords

Cite

@article{arxiv.2505.19239,
  title  = {DriveX: Omni Scene Modeling for Learning Generalizable World Knowledge in Autonomous Driving},
  author = {Chen Shi and Shaoshuai Shi and Kehua Sheng and Bo Zhang and Li Jiang},
  journal= {arXiv preprint arXiv:2505.19239},
  year   = {2025}
}
R2 v1 2026-07-01T02:37:34.947Z