Continuous Geodesic Self-Attention Models with Gated Fusion for Trajectory Prediction

Continuous Geodesic Self-Attention Models with Gated Fusion for Trajectory Prediction

연구 분야: Verification

학회: 2024 International Joint Conference on Neural Networks (IJCNN)

Driven by the rapid development of intelligent driving vehicles, predicting the trajectories of pedestrians on the road is crucial for decision-making during driving and even road safety. In this paper, we propose a novel method for trajectory prediction, namely, Continuous Geodesic Self-Attention Models with Gated Fusion (CGSAG). We use geodesic attention to measure the similarity between trajectory points, and utilize a gating mechanism to fuse the geodesic features extracted by multi-layer graph convolution. We then use Neural Ordinary Differential Equations (Neural ODE) to model the continuous-time dynamics of the trajectory. We show that CGSAG improves state-of-the-art performances on several human trajectory prediction datasets, including ETH/UCY, SDD, and Ind. At the same time, we conduct ablation studies to prove the effectiveness and efficiency of our proposed method.