用联合扩散模型生成交通场景的起点与终点,提升仿真多样性与可解释性。
Top-down Traffic Scenario Generation via Joint Initial-Goal Diffusion and Trajectory Infilling

- 通过联合建模起点与终点,实现自上而下的交通场景生成。
- 在Argoverse 2上使速度分布距离降低55.3%,偏离道路率下降2.8%。
- 适合自动驾驶仿真、轨迹预测模型的初始化与场景约束设计。
稳健的交通仿真器对自动驾驶系统开发至关重要,可减少昂贵且耗时的真实世界数据采集和道路实测需求。然而,现有仿真器需依赖给定的初始状态生成轨迹,限制了规模与多样性。尽管已有大量研究关注数据驱动的智能体初始化,但生成的初始状态缺乏可解释性。在已知初始状态的前提下,轨迹生成仍具挑战,因模型需学习目的地的多样性及随时间抵达路径的演化。本文提出TrafficDiffuser,一种自上而下的交通场景生成框架,通过联合建模初始状态与目标状态对,生成高阶交通场景。该方法使初始状态更具可解释性,并将轨迹生成简化为填充问题。我们展示了生成的高阶场景如何用于约束不同轨迹模式,以及与现有轨迹生成模型集成。在Argoverse 2运动预测数据集上进行了广泛实验,验证生成结果是否捕捉真实世界分布。除生成目标状态外,TrafficDiffuser在智能体初始化上优于现有最佳方法,使速度分布距离降低55.3%,偏离道路率下降2.8%。
原文摘要 · Abstract (English)
Robust traffic simulators are crucial for developing and testing autonomous vehicles to reduce the costly, labor-intensive real-world data collection process and the need for physical presence on the road. However, existing simulators require agents' initial states to generate trajectories, which limits scalability and diversity due to restrictions on the given initial states. While data-driven agent initialization has been widely studied, the generated initial states are not interpretable in terms of why the agents are initialized at those specific locations. Given known initial states, trajectory generation is also a challenging problem, as the model must learn the variability of the destination and how agents should reach it over time. In this paper, we propose TrafficDiffuser, a top-down traffic scenario generation framework that generates high-level traffic scenarios, defined by initial and goal state pairs, by jointly modeling them. The high-level scenario generation makes initial states better interpretable and reduces trajectory generation into as simple as an infilling problem. We demonstrate how the generated high-level traffic scenarios can be used, including constraining based on different trajectory modes and integrating them with existing trajectory generation models. We conduct extensive experiments on the Argoverse 2 motion prediction dataset to evaluate how well the generated outputs capture real-world distributions. In addition to generating goal states, TrafficDiffuser outperforms the next-best approach for agent initialization, reducing speed distribution distance by 55.3% and the off-road rate by 2.8%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。