STAFGCN: a spatial-temporal attention-based fusion graph convolution network for pedestrian trajectory prediction

Abstract

Pedestrian trajectory prediction provides crucial data support for the development of smart cities. Existing pedestrian trajectory prediction methods often overlook the different types of pedestrian interactions and the micro-level spatial-temporal relationships when handling the interaction information in spatial and temporal dimensions. The model employs a spatial-temporal attention-based fusion graph convolutional framework to predict future pedestrian trajectories. For the different types of local and global relationships between pedestrians, it first employs spatial-temporal attention mechanisms to capture dependencies in pedestrian sequence data, obtaining the social interactions of pedestrians in spatial contexts and the movement trends of pedestrians over time. Subsequently, a fusion graph convolutional module merges the temporal weight matrix and the spatial weight matrix into a spatial-temporal fusion feature map. Finally, a decoder section utilizes time-stacked convolutional neural networks to predict future trajectories. The final validation on the ETH and UCY datasets yielded experimental results with an average displacement error (ADE) of 0.34 and an final displacement error (FDE) of 0.55. The visualization results further demonstrated the rationality of the model.