基于时空融合的多头注意力车辆轨迹预测

宋秀兰,董兆航,单杭冠,陆炜杰

Vehicle trajectory prediction based on temporal-spatial multi-head attention mechanism

Xiu-lan SONG,Zhao-hang DONG,Hang-guan SHAN,Wei-jie LU

表 3 高速公路场景不同模型的预测性能对比

Tab.3 Comparison of prediction performance of different models in highway scenario

模型 RMSE te=1 s te=2 s te=3 s te=4 s te=5 s V-LSTM 0.70 1.79 3.22 4.96 7.04 CS-LSTM 0.61 1.27 2.09 3.10 4.37 S-GAN 0.57 1.32 2.22 3.26 4.40 MATF 0.66 1.34 2.08 2.97 4.13 STA 0.56 1.32 2.03 3.08 4.24 SIT 0.58 1.23 1.99 2.96 4.05 PIP 0.55 1.18 1.94 2.88 4.04 TSMHA 0.56 1.19 1.87 2.84 3.93