用因果结构引导扩散模型,生成更真实可控的自动驾驶交通场景。
Causal Composition Diffusion Model for Closed-loop Traffic Generation
- 通过因果结构注入扩散过程,实现可控与真实的平衡。
- 在碰撞率、偏离道路率等指标上优于现有方法。
- 适合需要高安全性的自动驾驶仿真测试场景。
仿真对自动驾驶安全性评估至关重要,尤其在捕捉复杂交互行为方面。然而,在长尾场景下生成既真实又可控的交通场景仍是一大挑战。现有生成模型在用户定义的可控性与真实性之间存在冲突,且在安全关键场景中尤为明显。本文提出因果组合扩散模型(CCDiff),一种结构引导的扩散框架。我们首先将可控且真实的闭环仿真学习建模为约束优化问题。CCDiff通过自动识别并直接注入因果结构到扩散过程中,最大化可控性的同时保证真实性,提供结构化指导以提升两者表现。在基准数据集和闭环仿真器上的严格评估显示,CCDiff在生成真实且用户偏好的轨迹方面显著优于现有最先进方法。结果表明,该模型能有效提取并利用因果结构,在碰撞率、偏离道路率、最终位移误差(FDE)及舒适度等关键指标上均实现性能提升。
原文摘要 · Abstract (English)
Simulation is critical for safety evaluation in autonomous driving, particularly in capturing complex interactive behaviors. However, generating realistic and controllable traffic scenarios in long-tail situations remains a significant challenge. Existing generative models suffer from the conflicting objective between user-defined controllability and realism constraints, which is amplified in safety-critical contexts. In this work, we introduce the Causal Compositional Diffusion Model (CCDiff), a structure-guided diffusion framework to address these challenges. We first formulate the learning of controllable and realistic closed-loop simulation as a constrained optimization problem. Then, CCDiff maximizes controllability while adhering to realism by automatically identifying and injecting causal structures directly into the diffusion process, providing structured guidance to enhance both realism and controllability. Through rigorous evaluations on benchmark datasets and in a closed-loop simulator, CCDiff demonstrates substantial gains over state-of-the-art approaches in generating realistic and user-preferred trajectories. Our results show CCDiff's effectiveness in extracting and leveraging causal structures, showing improved closed-loop performance based on key metrics such as collision rate, off-road rate, FDE, and comfort.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。