Spatial-Temporal Graph Attention Network for Pedestrian Trajectory Prediction

Yanran Liu,Hongyan Guo,Qingyu Meng,Jialin Li
DOI: https://doi.org/10.1109/cvci56766.2022.9964939
2022-01-01
Abstract:Ensuring pedestrians who are vulnerable road users safety is the prerequisite for the widespread application of autonomous vehicles. Accurate prediction of pedestrian trajectory can effectively reduce the occurrence of traffic accidents. Therefore, we propose an encoder-decoder model framework using a Spatiai-Temporal Graph Attention Network to predict future trajectory. The movement of pedestrians are not only related to their own subjective will, but also affected by their neighbors. We introduce temporal attention mechanism to get the importance of historical trajectories at different moments and use spatial attention mechanism based on Graph Attention Network(GAT) to give neighbors different attention to obtain spatial correlation. The experiments are constructed on the ETH and UCY datasets. The results confirm the efficacy of the proposed method and indicate that our model is superior to advanced models in most instances.
What problem does this paper attempt to address?