Paper Reading AI Learner

HERMES: A Unified Self-Driving World Model for Simultaneous 3D Scene Understanding and Generation

2025-01-24 18:59:51
Xin Zhou, Dingkang Liang, Sifan Tu, Xiwu Chen, Yikang Ding, Dingyuan Zhang, Feiyang Tan, Hengshuang Zhao, Xiang Bai

Abstract

Driving World Models (DWMs) have become essential for autonomous driving by enabling future scene prediction. However, existing DWMs are limited to scene generation and fail to incorporate scene understanding, which involves interpreting and reasoning about the driving environment. In this paper, we present a unified Driving World Model named HERMES. We seamlessly integrate 3D scene understanding and future scene evolution (generation) through a unified framework in driving scenarios. Specifically, HERMES leverages a Bird's-Eye View (BEV) representation to consolidate multi-view spatial information while preserving geometric relationships and interactions. We also introduce world queries, which incorporate world knowledge into BEV features via causal attention in the Large Language Model (LLM), enabling contextual enrichment for understanding and generation tasks. We conduct comprehensive studies on nuScenes and OmniDrive-nuScenes datasets to validate the effectiveness of our method. HERMES achieves state-of-the-art performance, reducing generation error by 32.4% and improving understanding metrics such as CIDEr by 8.0%. The model and code will be publicly released at this https URL.

Abstract (translated)

驾驶世界模型(DWMs)已成为自动驾驶不可或缺的一部分,通过实现未来场景预测来支持自动驾驶。然而,现有的DWM仅限于生成新场景,并未能整合对驾驶环境的理解,即无法解读和推理驾驶环境中的情况。在本文中,我们提出了一种统一的驾驶世界模型HERMES。我们在驾驶场景下通过一个统一框架无缝地集成了三维场景理解和未来场景演变(生成)。具体而言,HERMES利用鸟瞰图(BEV)表示来整合多视角空间信息的同时保持几何关系和互动性。此外,我们还引入了世界查询机制,该机制通过在大型语言模型中应用因果注意力将世界知识融入到BEV特征中,从而为理解和生成任务提供上下文增强。我们在nuScenes和OmniDrive-nuScenes数据集上进行了全面的研究以验证我们的方法的有效性。HERMES取得了最先进的性能,在减少生成误差方面比现有技术降低了32.4%,在理解指标如CIDEr的评估中提高了8.0%。该模型及其代码将在[此链接](https://this https URL)公开发布。

URL

https://arxiv.org/abs/2501.14729

PDF

https://arxiv.org/pdf/2501.14729.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot