Unleashing the Potential of Diffusion Models for End-to-End Autonomous Driving

📅 2026-02-26
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
This work addresses the limited empirical validation of diffusion models in end-to-end autonomous driving under large-scale real-world conditions. For the first time, it systematically evaluates and optimizes diffusion-based planners in actual urban road environments, establishing key design principles concerning trajectory representation, loss formulation, and data augmentation. To further enhance safety, the authors introduce a reinforcement learning-based post-training strategy. The resulting Hyper Diffusion Planner is evaluated over 200 kilometers of real-world driving across six diverse urban scenarios, demonstrating a tenfold improvement in planning performance over baseline methods. These results substantiate the feasibility and superiority of diffusion models for end-to-end autonomous driving in complex, real-world settings.

Technology Category

Application Category

📝 Abstract
Diffusion models have become a popular choice for decision-making tasks in robotics, and more recently, are also being considered for solving autonomous driving tasks. However, their applications and evaluations in autonomous driving remain limited to simulation-based or laboratory settings. The full strength of diffusion models for large-scale, complex real-world settings, such as End-to-End Autonomous Driving (E2E AD), remains underexplored. In this study, we conducted a systematic and large-scale investigation to unleash the potential of the diffusion models as planners for E2E AD, based on a tremendous amount of real-vehicle data and road testing. Through comprehensive and carefully controlled studies, we identify key insights into the diffusion loss space, trajectory representation, and data scaling that significantly impact E2E planning performance. Moreover, we also provide an effective reinforcement learning post-training strategy to further enhance the safety of the learned planner. The resulting diffusion-based learning framework, Hyper Diffusion Planner} (HDP), is deployed on a real-vehicle platform and evaluated across 6 urban driving scenarios and 200 km of real-world testing, achieving a notable 10x performance improvement over the base model. Our work demonstrates that diffusion models, when properly designed and trained, can serve as effective and scalable E2E AD planners for complex, real-world autonomous driving tasks.
Problem

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

Diffusion Models
End-to-End Autonomous Driving
Real-World Deployment
Autonomous Driving Planning
Scalability
Innovation

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

diffusion models
end-to-end autonomous driving
trajectory planning
real-world deployment
reinforcement learning post-training
🔎 Similar Papers
No similar papers found.
Yinan Zheng
Yinan Zheng
Tsinghua University
Reinforcement LearningDiffusion ModelsAutonomous DrivingRobotics
T
Tianyi Tan
Institute for AI Industry Research (AIR), Tsinghua University
B
Bin Huang
Xiaomi EV
E
Enguang Liu
Xiaomi EV
Ruiming Liang
Ruiming Liang
Institute of Automation, CAS
Physical AIGenerative Models
J
Jianlin Zhang
Xiaomi EV
Jianwei Cui
Jianwei Cui
University of Science & Technology of China
Speech Processing
G
Guang Chen
Xiaomi EV
Kun Ma
Kun Ma
University of Jinan
Model-driven EngineeringBig Data ManagementData Intensive Computing
H
Hangjun Ye
Xiaomi EV
L
Long Chen
Xiaomi EV
Y
Ya-Qin Zhang
Institute for AI Industry Research (AIR), Tsinghua University
Xianyuan Zhan
Xianyuan Zhan
Associate Professor, Institute for AI Industry Research (AIR), Tsinghua University
Data-driven decision-makingReal-world RL/ILEmbodied AIAutonomous Driving
J
Jingjing Liu
Institute for AI Industry Research (AIR), Tsinghua University