Search papers, labs, and topics across Lattice.
This paper introduces RiskFlow, a novel framework for generating safety-critical traffic scenarios that enhances the evaluation of autonomous driving systems by addressing the limitations of existing diffusion-based methods. By formulating future trajectory generation as transport in the action space and employing a JVP-based objective, RiskFlow enables efficient and stable training, achieving a significant reduction in computational costs while improving the realism of generated scenarios. Experimental results on the nuScenes dataset demonstrate that RiskFlow not only outperforms baseline methods in realism and safety-critical generation but also substantially decreases inference time, making it a practical solution for real-world applications.
RiskFlow achieves a remarkable balance between realism and safety in traffic scenario generation, cutting inference time while enhancing the fidelity of multi-agent interactions.
Safety-critical traffic scenario generation is essential for evaluating autonomous driving systems under rare but high-risk interactions. Existing diffusion-based methods offer strong controllability in closed-loop generation, but their iterative denoising process is computationally expensive and may accumulate sampling and guidance errors over long rollouts, causing unrealistic motion artifacts such as jitter, abnormal acceleration, and off-road behavior. To address these issues, we propose RiskFlow, a closed-loop safety-critical multi-agent traffic generation framework that formulates future trajectory generation as transport in the action space. Instead of relying on iterative denoising, RiskFlow learns an average velocity field over a finite interval to transform Gaussian action sequences into future acceleration and yaw-rate commands with a single forward pass, using a JVP-based objective for efficient and stable training. At test time, RiskFlow applies output-space guidance to the generated actions, steering selected critical agents toward risky interactions while regularizing off-road behavior, and reconstructs physically feasible trajectories through vehicle dynamics. Experiments on nuScenes with tbsim closed-loop evaluation show that RiskFlow achieves a strong adversariality-realism trade-off across multi-agent and long-horizon settings. Compared with representative baselines, RiskFlow consistently improves realism while maintaining competitive safety-critical generation capability, and substantially reduces inference time for evaluation.
