Abstract:Trajectory prediction aims to predict the movement intention of traffic participants in the future based on the historical observation trajectories. For traffic scenarios, pedestrians, vehicles and other traffic participants have social interaction of surrounding traffic participants in both time and spatial dimensions. Most previous studies only use pooling methods to simulate the interaction process between participants and cannot fully capture the spatio-temporal dependence, possibly accumulating errors with the increase in prediction time. To overcome these problems, we propose the Spatial-Temporal Interaction Attention-based Trajectory Prediction Network (STIA-TPNet), which can effectively model the spatial-temporal interaction information. Based on trajectory feature extraction, the novel Spatial-Temporal Interaction Attention Module (STIA Module) is proposed to extract the interaction relationships between traffic participants, including temporal interaction attention, spatial interaction attention, and spatio-temporal attention fusion. By adaptive allocation of attention weights, temporal interaction attention is a temporal attention mechanism used to capture the movement pattern of each traffic participant in the scene, which can learn the importance of historical trajectories at different moments to future behaviors. Since the participants number in recent traffic scenes dynamically changes, the spatial interaction attention is designed to abstract the traffic participants in the scene into graph nodes, and abstract the social interaction between participants into graph edges. Coupling the temporal and spatial interaction attentions can adaptively model the temporal-spatial information and achieve accurate trajectory prediction. By performing experiments on the INTERACTION dataset and the UTP (Unmanned Aerial Vehicle-based Trajectory Prediction) dataset, the experimental results show that the proposed method significantly improves the accuracy of trajectory prediction and outperforms the representative methods in comparison.

A Spatial-Temporal Attention Model for Human Trajectory Prediction

HSTA: A Hierarchical Spatio-Temporal Attention Model for Trajectory Prediction

A Spatial-Temporal Attentive Network with Spatial Continuity for Trajectory Prediction

Pedestrian trajectory prediction based on spatio-temporal attention mechanism

Global Temporal Attention Optimization for Human Trajectory Prediction

A Spatio-Temporal Transformer Network for Human Motion Prediction in Human-Robot Collaboration

Traffic Agents Trajectory Prediction Based on Spatial-Temporal Interaction Attention

Intelligent Vehicles Trajectory Prediction with Spatial and Temporal Attention Mechanism

Spatial-Temporal Attention Networks for Vehicle Trajectory Prediction.

Spatial-Temporal Attentive LSTM for Vehicle-Trajectory Prediction

Situation-Aware Pedestrian Trajectory Prediction with Spatio-Temporal Attention Model

GA-STT: Human Trajectory Prediction with Group Aware Spatial-Temporal Transformer.

A Spatial-Temporal Self-Attention Network (STSAN) for Location Prediction

Improved Attention Mechanism for Human-like Intelligent Vehicle Trajectory Prediction

Human Trajectory Prediction using Spatially aware Deep Attention Models

S-T CRF: Spatial-Temporal Conditional Random Field for Human Trajectory Prediction

TCSA-Net: A Temporal-Context-Based Self-Attention Network for Next Location Prediction

Spatial-Temporal Graph Attention Network for Pedestrian Trajectory Prediction

An Efficient Spatial-Temporal Model Based on Gated Linear Units for Trajectory Prediction

SRA-LSTM: Social Relationship Attention LSTM for Human Trajectory Prediction

STF: Spatial Temporal Fusion for Trajectory Prediction