GraphWorld: Long-Horizon Planning with World Models for End-to-End Autonomous Driving

📅 2026-06-15
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
Existing end-to-end autonomous driving approaches suffer from limited generalization and safety in complex interactive scenarios due to inadequate modeling of long-term temporal dependencies. This work proposes GraphWorld, a framework that constructs an ego-centric, adaptive neighborhood interaction graph and introduces a cross-node cross-attention mechanism to dynamically capture interactions among surrounding agents. By treating the implicit world state as a conditional signal for planning, GraphWorld enables end-to-end joint optimization of perception, prediction, and planning, facilitating long-horizon, safety-aware trajectory generation. Evaluated on Bench2Drive, NAVSIMv1/2, and nuScenes benchmarks, GraphWorld significantly reduces collision rates and substantially improves long-horizon planning performance in complex driving scenarios.
📝 Abstract
End-to-end autonomous driving has made significant progress by unifying perception, prediction, and planning within a single learning framework, achieving strong performance in short-horizon decision making. However, most existing E2E-AD methods remain confined to short-horizon planning and lack the ability to model long-term temporal dependencies, which severely limits their generalization and security in complex and highly interactive driving scenarios. In this work, we propose GraphWorld, an E2E-AD framework that explicitly enhances long-horizon planning through latent world modeling. We introduce an Ego-Centric Interaction Graph, which adaptively models critical neighboring agents based on spatial proximity, and propagates relational context to planning queries via cross-node cross-attention. We present a World-State-Conditioned Planning that learns ego-centric latent world representations by modeling interactions between an ego vehicle and surrounding agents. This latent world state captures key interaction dynamics and safety-relevant semantics, and serves as a conditioning signal to guide long-horizon, safety-aware trajectory planning. Extensive experiments on Bench2Drive, NAVSIMv1/2, and nuScenes demonstrate that GraphWorld significantly reduces collision rates and improves long-horizon planning performance, validating its effectiveness in complex driving environments.
Problem

Research questions and friction points this paper is trying to address.

long-horizon planning
end-to-end autonomous driving
temporal dependencies
world models
safety-aware planning
Innovation

Methods, ideas, or system contributions that make the work stand out.

world models
long-horizon planning
ego-centric interaction graph
cross-attention
end-to-end autonomous driving
🔎 Similar Papers
No similar papers found.
Ziying Song
Ziying Song
Beijing Jiaotong University
Object DetectionComputer VisionDeep Learning
C
Caiyan Jia
Beijing Key Laboratory of Traffic Data Mining and Embodied Intelligence, School of Computer Science and Technology, Beijing Jiaotong University
L
Lin Liu
Beijing Key Laboratory of Traffic Data Mining and Embodied Intelligence, School of Computer Science and Technology, Beijing Jiaotong University
L
Lei Yang
School of Mechanical and Aerospace Engineering, Nanyang Technological University, Singapore
S
Shengkai Zhang
Beijing Key Laboratory of Traffic Data Mining and Embodied Intelligence, School of Computer Science and Technology, Beijing Jiaotong University
Feiyang Jia
Feiyang Jia
Beijing Jiaotong University
F
Fengda Zhao
School of Artificial Intelligence (School of Software), Yanshan University, Qinhuangdao, China
P
Peiliang Wu
School of Artificial Intelligence (School of Software), Yanshan University, Qinhuangdao, China
Shaoqing Xu
Shaoqing Xu
University of Macau, BUAA, Xiaomi EV
3D Computer Vision3D GenerationVision and Language ModelEnd2EndWorld Model
C
Chen Lv
School of Mechanical and Aerospace Engineering, Nanyang Technological University, Singapore
Yadan Luo
Yadan Luo
ARC DECRA and Senior Lecturer, University of Queensland
Generalization3D VisionAutonomous Driving