arXiv · 2004.03842
Multi-Head Attention based Probabilistic Vehicle Trajectory Prediction
Abstract
This paper presents online-capable deep learning model for probabilistic vehicle trajectory prediction. We propose a simple encoder-decoder architecture based on multi-head attention. The proposed model generates the distribution of the predicted trajectories for multiple vehicles in parallel. Our approach to model the interactions can learn to attend to a few influential vehicles in an unsupervised manner, which can improve the interpretability of the network. The experiments using naturalistic trajectories at highway show the clear improvement in terms of positional error on both longitudinal and lateral direction.
Explore related subjects
Keep this discovery
Hayoung Kim, Dongchan Kim, Gihoon Kim, Jeongmin Cho, Kunsoo Huh. 2020-04-08. Multi-Head Attention based Probabilistic Vehicle Trajectory Prediction. https://arxiv.org/abs/2004.03842
Cite the original work for its findings. Save a collection to share your selection of sources.