arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2311.14922cs.CV

GDTS:用于多模态行人轨迹预测的基于树采样的目标引导扩散模型

GDTS: Goal-Guided Diffusion Model with Tree Sampling for Multi-Modal Pedestrian Trajectory Prediction

The Hong Kong University of Science and Technology(香港科技大学);The Hong Kong University of Science and Technology (Guangzhou)(香港科技大学(广州))

Ge Sun, Sheng Wang, Lei Zhu, Ming Liu, Jun Ma

AI总结:

提出GDTS模型,通过目标估计引导扩散网络并结合两阶段树采样算法,解决了行人轨迹多模态预测中推理时间长和输入信息利用不足的问题,实现了实时且高精度的预测。

摘要:

准确预测行人轨迹对于提高自动驾驶的安全性至关重要。然而,由于人类运动固有的随机性,这项任务通常并不简单,这自然要求预测器生成多模态预测。以往的研究利用各种生成方法(如GAN和VAE)进行行人轨迹预测。尽管如此,这些方法可能会遭受模式崩溃和结果质量相对较低的问题。去噪扩散概率模型(DDPM)近期因其简单的训练过程和强大的重建能力被应用于轨迹预测。然而,当前的基于扩散的方法并未充分利用输入信息,且通常需要大量去噪迭代,导致推理时间长或需要额外的初始化网络。为了解决这些挑战并促进扩散模型在多模态轨迹预测中的应用,我们提出了GDTS,一种用于多模态轨迹预测的基于树采样的目标引导扩散模型。考虑到人类运动“目标驱动”的特征,GDTS利用目标估计来引导扩散网络的生成。提出了一种两阶段树采样算法,利用共同特征来减少推理时间并提高多模态预测的准确性。实验结果表明,我们提出的框架在公开数据集上实现了与实时推理速度相当的最新性能。

英文摘要:

Accurate prediction of pedestrian trajectories is crucial for improving the safety of autonomous driving. However, this task is generally nontrivial due to the inherent stochasticity of human motion, which naturally requires the predictor to generate multi-modal prediction. Previous works leverage various generative methods, such as GAN and VAE, for pedestrian trajectory prediction. Nevertheless, these methods may suffer from mode collapse and relatively low-quality results. The denoising diffusion probabilistic model (DDPM) has recently been applied to trajectory prediction due to its simple training process and powerful reconstruction ability. However, current diffusion-based methods do not fully utilize input information and usually require many denoising iterations that lead to a long inference time or an additional network for initialization. To address these challenges and facilitate the use of diffusion models in multi-modal trajectory prediction, we propose GDTS, a novel Goal-Guided Diffusion Model with Tree Sampling for multi-modal trajectory prediction. Considering the "goal-driven" characteristics of human motion, GDTS leverages goal estimation to guide the generation of the diffusion network. A two-stage tree sampling algorithm is presented, which leverages common features to reduce the inference time and improve accuracy for multi-modal prediction. Experimental results demonstrate that our proposed framework achieves comparable state-of-the-art performance with real-time inference speed in public datasets.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑