用扩散模型预测停车场内车人轨迹,提升自动泊车安全性
ParkDiffusion: Heterogeneous Multi-Agent Multi-Modal Trajectory Prediction for Automated Parking using Diffusion Models
- 基于扩散模型建模轨迹不确定性与多模式特性
- 在DLP和inD数据集上显著优于现有方法
- 适合自动驾驶泊车系统研发与算法评估
自动泊车是高级驾驶辅助系统(ADAS)的关键功能,准确的轨迹预测对连接感知与规划模块至关重要。尽管意义重大,该领域研究仍相对有限,多数工作聚焦于单一模态的车辆轨迹预测。本文提出ParkDiffusion,一种新型方法,可预测自动泊车场景中车辆与行人的轨迹。该方法采用扩散模型捕捉未来轨迹的内在不确定性和多模态特性,并引入三项关键创新:首先,设计双路地图编码器,通过两阶段交叉注意力机制处理软语义线索与硬几何约束;其次,提出自适应代理类型嵌入模块,动态依据车辆与行人特征条件化预测过程;第三,为确保运动学可行性,模型输出控制信号,并在运动学框架内生成物理可行轨迹。我们在龙湖停车场(DLP)与无人机交叉口(inD)数据集上评估了ParkDiffusion。本工作建立了停车场景异构轨迹预测的新基准,性能显著优于现有方法。
原文摘要 · Abstract (English)
Automated parking is a critical feature of Advanced Driver Assistance Systems (ADAS), where accurate trajectory prediction is essential to bridge perception and planning modules. Despite its significance, research in this domain remains relatively limited, with most existing studies concentrating on single-modal trajectory prediction of vehicles. In this work, we propose ParkDiffusion, a novel approach that predicts the trajectories of both vehicles and pedestrians in automated parking scenarios. ParkDiffusion employs diffusion models to capture the inherent uncertainty and multi-modality of future trajectories, incorporating several key innovations. First, we propose a dual map encoder that processes soft semantic cues and hard geometric constraints using a two-step cross-attention mechanism. Second, we introduce an adaptive agent type embedding module, which dynamically conditions the prediction process on the distinct characteristics of vehicles and pedestrians. Third, to ensure kinematic feasibility, our model outputs control signals that are subsequently used within a kinematic framework to generate physically feasible trajectories. We evaluate ParkDiffusion on the Dragon Lake Parking (DLP) dataset and the Intersections Drone (inD) dataset. Our work establishes a new baseline for heterogeneous trajectory prediction in parking scenarios, outperforming existing methods by a considerable margin.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。