Unleashing the Potential of Diffusion Models for End-to-End Autonomous Driving
AI 摘要
该论文探索了扩散模型在端到端自动驾驶中的应用,并进行了大规模实车验证。
主要贡献
- 系统性研究了扩散模型在端到端自动驾驶中的潜力
- 提出了Hyper Diffusion Planner (HDP) 框架
- 通过实车测试验证了HDP的有效性和可扩展性
方法论
基于大量真实车辆数据,使用扩散模型作为规划器,并结合强化学习进行后训练,提升安全性。
原文摘要
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.