ADM: Accelerated Diffusion Model via Estimated Priors for Robust Motion Prediction under Uncertainties

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Li, Jiahui, Shen, Tianle, Gu, Zekai, Sun, Jiawei, Yuan, Chengran, Han, Yuhang, Sun, Shuo, Ang Jr, Marcelo H.
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866913339039285248
author Li, Jiahui
Shen, Tianle
Gu, Zekai
Sun, Jiawei
Yuan, Chengran
Han, Yuhang
Sun, Shuo
Ang Jr, Marcelo H.
author_facet Li, Jiahui
Shen, Tianle
Gu, Zekai
Sun, Jiawei
Yuan, Chengran
Han, Yuhang
Sun, Shuo
Ang Jr, Marcelo H.
contents Motion prediction is a challenging problem in autonomous driving as it demands the system to comprehend stochastic dynamics and the multi-modal nature of real-world agent interactions. Diffusion models have recently risen to prominence, and have proven particularly effective in pedestrian motion prediction tasks. However, the significant time consumption and sensitivity to noise have limited the real-time predictive capability of diffusion models. In response to these impediments, we propose a novel diffusion-based, acceleratable framework that adeptly predicts future trajectories of agents with enhanced resistance to noise. The core idea of our model is to learn a coarse-grained prior distribution of trajectory, which can skip a large number of denoise steps. This advancement not only boosts sampling efficiency but also maintains the fidelity of prediction accuracy. Our method meets the rigorous real-time operational standards essential for autonomous vehicles, enabling prompt trajectory generation that is vital for secure and efficient navigation. Through extensive experiments, our method speeds up the inference time to 136ms compared to standard diffusion model, and achieves significant improvement in multi-agent motion prediction on the Argoverse 1 motion forecasting dataset.
format Preprint
id arxiv_https___arxiv_org_abs_2405_00797
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle ADM: Accelerated Diffusion Model via Estimated Priors for Robust Motion Prediction under Uncertainties
Li, Jiahui
Shen, Tianle
Gu, Zekai
Sun, Jiawei
Yuan, Chengran
Han, Yuhang
Sun, Shuo
Ang Jr, Marcelo H.
Robotics
Computer Vision and Pattern Recognition
Motion prediction is a challenging problem in autonomous driving as it demands the system to comprehend stochastic dynamics and the multi-modal nature of real-world agent interactions. Diffusion models have recently risen to prominence, and have proven particularly effective in pedestrian motion prediction tasks. However, the significant time consumption and sensitivity to noise have limited the real-time predictive capability of diffusion models. In response to these impediments, we propose a novel diffusion-based, acceleratable framework that adeptly predicts future trajectories of agents with enhanced resistance to noise. The core idea of our model is to learn a coarse-grained prior distribution of trajectory, which can skip a large number of denoise steps. This advancement not only boosts sampling efficiency but also maintains the fidelity of prediction accuracy. Our method meets the rigorous real-time operational standards essential for autonomous vehicles, enabling prompt trajectory generation that is vital for secure and efficient navigation. Through extensive experiments, our method speeds up the inference time to 136ms compared to standard diffusion model, and achieves significant improvement in multi-agent motion prediction on the Argoverse 1 motion forecasting dataset.
title ADM: Accelerated Diffusion Model via Estimated Priors for Robust Motion Prediction under Uncertainties
topic Robotics
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2405.00797