Multi-modal trajectory prediction (MTP) has become the research trend in the field of autonomous driving, as it provides multiple plausible trajectories. However, related works lack attention to the temporal dependence of historical features and inherent association between multiple trajectory modes, which may lead to large deviations. To address these critical limitations, we propose a multi-modal trajectory prediction network that integrates historical motion and spatio-temporal interaction (MTPN-IMI). In MTPN-IMI, a local spatio-temporal graph (LSTG) is constructed to model local agent-agent interaction. Furthermore, a Causal Convolution Module (CCM) and a Causal Self-Attention Module (CSAM) are introduced to focus on local and global temporal dependence in historical motion feature and local agent-agent interaction feature. Moreover, a Cross Attention Module (CAM) is utilized to capture the inherent association between multiple modes. Experiments show that our model outperforms related models on Argoverse1.1 validation set, achieving superior prediction accuracy.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Multi-modal Trajectory Prediction Network that Integrates Historical Motion and Spatio-Temporal Interaction

  • Chenlong Li,
  • Mingxing Li,
  • Jian Zhao

摘要

Multi-modal trajectory prediction (MTP) has become the research trend in the field of autonomous driving, as it provides multiple plausible trajectories. However, related works lack attention to the temporal dependence of historical features and inherent association between multiple trajectory modes, which may lead to large deviations. To address these critical limitations, we propose a multi-modal trajectory prediction network that integrates historical motion and spatio-temporal interaction (MTPN-IMI). In MTPN-IMI, a local spatio-temporal graph (LSTG) is constructed to model local agent-agent interaction. Furthermore, a Causal Convolution Module (CCM) and a Causal Self-Attention Module (CSAM) are introduced to focus on local and global temporal dependence in historical motion feature and local agent-agent interaction feature. Moreover, a Cross Attention Module (CAM) is utilized to capture the inherent association between multiple modes. Experiments show that our model outperforms related models on Argoverse1.1 validation set, achieving superior prediction accuracy.