Causal Composition Diffusion Model for Closed-loop Traffic Generation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Lin, Haohong, Huang, Xin, Phan-Minh, Tung, Hayden, David S., Zhang, Huan, Zhao, Ding, Srinivasa, Siddhartha, Wolff, Eric M., Chen, Hongge
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910969398034432
author Lin, Haohong
Huang, Xin
Phan-Minh, Tung
Hayden, David S.
Zhang, Huan
Zhao, Ding
Srinivasa, Siddhartha
Wolff, Eric M.
Chen, Hongge
author_facet Lin, Haohong
Huang, Xin
Phan-Minh, Tung
Hayden, David S.
Zhang, Huan
Zhao, Ding
Srinivasa, Siddhartha
Wolff, Eric M.
Chen, Hongge
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2412_17920
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Causal Composition Diffusion Model for Closed-loop Traffic Generation
Lin, Haohong
Huang, Xin
Phan-Minh, Tung
Hayden, David S.
Zhang, Huan
Zhao, Ding
Srinivasa, Siddhartha
Wolff, Eric M.
Chen, Hongge
Artificial Intelligence
Machine Learning
Robotics
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.
title Causal Composition Diffusion Model for Closed-loop Traffic Generation
topic Artificial Intelligence
Machine Learning
Robotics
url https://arxiv.org/abs/2412.17920