Adaptive Graph and Multi-Head Attention-Based Vehicle Trajectory Prediction

Pingyi Zhang,Wen Hu,Bangji Zhang,Zhaoxuan Ma,Penghao Li
DOI: https://doi.org/10.1109/iecon51785.2023.10311937
2023-01-01
Abstract:Trajectory prediction is a critical and challenging task in autonomous driving systems due to the various driving behavior of surrounding vehicles in complex environments. The existing trajectory prediction methods considering vehicles interaction are generally designed for specific scenarios. Besides, the complex network construction of learning-based methods results in burdened computation and worse practicability. To address the above problem, this work proposes an adaptive graph and multi-head attention-based LSTM model (AGA-LSTM) to realize accurate trajectory prediction. The AGA-LSTM uses the two-channel encoder-decoder architecture, and contains position encoder, interaction encoder, and future decoder. The position encoder is to code the history trajectory through a fully connected layer (FC). The interaction encoder integrates the interaction information in spatial and temporal, through a multi-head attention mechanism and a LSTM network, respectively. Additionally, the future decoder predicts the future trajectories at one time by using FC, interference, and original position matrix (OPM). Finally, the results show that the AGA-LSTM can capture vehicle interactions in various environments, and has superior computation efficiency and real-time performance.
What problem does this paper attempt to address?