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

Kexin Ke, Zhengyu Li, Huining Chen, Hao Wang, Xian Wei, Jian Yang, Xuan Tang

Abstract: 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.