基于时空图注意力网络的车辆多模态轨迹预测模型

陈文强,王东丹,朱文英,汪勇杰,王涛

Vehicle multimodal trajectory prediction model based on spatio-temporal graph attention network

Wenqiang CHEN,Dongdan WANG,Wenying ZHU,Yongjie WANG,Tao WANG

表 3 消融实验均方根误差对比

Tab.3 RMSE comparison of ablation experiment

数据集 tf/s RMSE(lat) RMSE(lon) STGAMT (w/o IFE) STGAMT STGAMT (w/o IFE) STGAMT HighD 1 0.03 0.03 0.07 0.07 2 0.12 0.11 0.14 0.14 3 0.25 0.23 0.24 0.22 4 0.37 0.36 0.53 0.48 5 0.49 0.47 1.09 1.04 NGSIM 1 0.08 0.08 0.20 0.20 2 0.20 0.20 0.77 0.75 3 0.30 0.29 1.49 1.46 4 0.39 0.37 2.43 2.37 5 0.49 0.40 3.66 3.54