Method for Multi-Vehicle Trajectory Prediction Based on Spatio-Temporal Feature Enhancement at Intersections
-
摘要:
交叉口作为自动驾驶汽车的典型应用场景,其复杂的交通参与者交互模式和多模态的行为轨迹特征给自动驾驶的预测系统带来了严峻挑战. 针对现有无地图轨迹预测方法在车辆交互建模精度不足、运动方向特征刻画粗糙以及长时序预测稳定性有限等问题,提出一种基于时空特征增强的多车轨迹预测方法(Wave-enhanced Spatio-Temporal Transformer,WAGT). 首先,构建基于图神经网络的空间交互建模框架,通过加权邻接矩阵刻画车辆间的交互强度;其次,引入基于波叠加机制的双通道特征增强模块,从振幅与相位角度自适应建模车辆横纵向运动差异;最后,设计融合时间感知位置编码的Transformer编解码器,以缓解长序列预测中的位置混淆与误差累积问题. 实验基于中国信号交叉口数据集SIND对所提方法性能进行量化评估,并与多种无地图轨迹预测模型进行对比,实验结果表明:本文方法在不同预测时域内均取得最优性能,其中相较于代表性图交互模型(GRIP),在平均位移误差(ADE)、最终位移误差(FDE)指标上分别提升39.70%、22.90%,验证了在复杂交叉口场景下的预测精度与鲁棒性优势.
-
关键词:
- 信号交叉口 /
- 车辆轨迹预测 /
- 图神经网络 /
- 时空特征建模 /
- Transformer
Abstract:As typical application scenarios for autonomous vehicles, intersections pose severe challenges to the prediction systems of autonomous driving due to the complex interaction patterns of traffic participants and multi-modal behavioral trajectory features. In view of the problems of existing map-free trajectory prediction methods, such as insufficient accuracy in modeling vehicle interactions, coarse characterization of motion direction features, and limited stability in long-term predictions, a multi-vehicle trajectory prediction method based on spatio-temporal feature enhancement (wave-enhanced spatio-temporal transformer, WAGT) was proposed. Firstly, a spatial interaction modeling framework based on graph neural networks was constructed to characterize the interaction intensity among vehicles through weighted adjacency matrices. Secondly, a dual-channel feature enhancement module based on a wave superposition mechanism was introduced to adaptively model the differences in lateral and longitudinal vehicle motions from the perspectives of amplitude and phase. Finally, a Transformer encoder-decoder integrating time-aware positional encoding was designed to alleviate the problems of position confusion and error accumulation in long-sequence prediction. Quantitative evaluation of the method performance was conducted based on the SIND dataset of Chinese signalized intersections, and comparisons with various map-free trajectory prediction models were performed. Experimental results indicate that the method achieves optimal performance in different prediction horizons. Compared with the representative graph interaction model GRIP, the average displacement error (ADE) and final displacement error (FDE) of the method are improved by 39.70% and 22.90%, respectively, which verifies the advantages of the model in prediction accuracy and robustness under complex intersection scenarios.
-
表 1 不同交互机制误差对比
Table 1. Comparison of errors among different interaction mechanisms
模型 2 s预测时域 3 s预测时域 平均值 eADE eFDE eADE eFDE eADE eFDE ONE-GCN 2.6691 6.0252 3.5986 8.3889 3.1339 7.2071 RD-GCN 1.8765 3.6297 2.9837 6.2208 2.4301 4.9253 WAGT 0.7720 2.0520 1.7306 4.5258 1.2513 3.2889 表 2 不同模块的eADE对比
Table 2. Comparison of ADE among different modules
模型 2 s 3 s 4 s 5 s 平均值 WGG 1.5007 2.9708 4.1396 4.6432 3.3136 t-Transformer 2.5697 2.4008 3.3699 5.0906 3.3578 WAGT 0.7720 1.7306 2.8621 3.6445 2.2523 表 3 不同模块的eFDE对比
Table 3. Comparison of FDE among different modules
模型 2 s 3 s 4 s 5 s 平均值 WGG 3.1382 6.0934 7.4461 11.1174 6.9488 t-Transformer 3.3052 6.1331 8.8050 11.9318 7.5438 WAGT 2.0520 4.5258 6.8940 10.6812 6.0383 表 4 不同模型的eADE对比
Table 4. Comparison of ADE among different models
模型 2 s 3 s 4 s 5 s GRIP 2.6691 2.9837 4.1569 5.1317 BAT 1.6315 2.6925 3.6614 4.5969 S-LSTM 1.4663 2.4314 3.4550 4.4907 CS-LSTM 1.4991 2.8855 3.5614 4.6637 IATN 1.6898 2.6678 3.8878 5.8642 WSiP 1.5493 1.8520 3.4626 5.7391 MHA 1.5448 1.8922 3.5736 5.8666 WAGT 0.7720 1.7306 2.8621 3.6445 表 5 不同模型的eFDE对比
Table 5. Comparison of FDE among different models
模型 2 s 3 s 4 s 5 s GRIP 6.0252 6.2208 8.0669 11.0153 BAT 3.8975 6.0171 9.8641 11.5508 S-LSTM 3.3226 5.5810 8.0072 10.3080 CS-LSTM 3.3994 5.6685 8.2617 10.8250 IATN 7.7133 6.2857 11.3770 13.7181 WSiP 3.1452 5.7253 8.2147 12.8365 MHA 2.9882 5.2624 7.0516 13.1267 WAGT 2.0520 4.5258 6.8940 10.6812 -
[1] Yang L, Liu S Y, Feng S, et al. Generation of critical pedestrian scenarios for autonomous vehicle testing[J]. Accident Analysis & Prevention, 2025, 214: 107962. doi: 10.1016/j.aap.2025.107962 [2] 杨涛, 马玉琴, 刘梦, 等. 智能网联环境下信号交叉口车辆轨迹重构模型[J]. 西南交通大学学报, 2024, 59(5): 1148-1157. doi: 10.3969/j.issn.0258-2724.20220321Yang Tao, Ma Yuqin, Liu Meng, et al. Vehicle trajectory reconstruction model of signalized intersection in connected automated environments[J]. Journal of Southwest Jiaotong University, 2024, 59(5): 1148-1157. doi: 10.3969/j.issn.0258-2724.20220321 [3] 瞿广跃, 杨澜, 袁梦, 等. 面向自动驾驶汽车的信号交叉口行人多模态轨迹预测方法[J]. 汽车安全与节能学报, 2024, 15(5): 689-701. doi: 10.3969/j.issn.1674-8484.2024.05.007Qu Guangyue, Yang Lan, Yuan Meng, et al. A multimodal trajectory prediction method of pedestrians at signalized intersections for autonomous vehicles[J]. Journal of Automotive Safety and Engergy, 2024, 15(5): 689-701. doi: 10.3969/j.issn.1674-8484.2024.05.007 [4] 韩勇, 李燕婷, 潘迪, 等. 交通碰撞场景下电动两轮车轨迹预测的轻量级建模方法[J]. 中国公路学报, 2024, 37(8): 302-310.Han Yong, Li Yanting, Pan Di, et al. Lightweight modeling of electric two-wheeler trajectory predictions in traffic collisions[J]. China Journal of Highway and Transport, 2024, 37(8): 302-310. [5] Rajamani R. Vehicle dynamics and control[M]. Boston: Springer US, 2012. [6] 徐婷, 邓恺龙, 刘永涛, 等. 基于航测数据的不同风格换道轨迹规划[J]. 西南交通大学学报, 2024, 59(3): 720-728.Xu Ting, Deng Kailong, Liu Yongtao, et al. Different styles of lane changing trajectory planning based on aerial survey data[J]. Journal of Southwest Jiaotong University, 2024, 59(3): 720-728. [7] Treiber M, Hennecke A, Helbing D. Congested traffic states in empirical observations and microscopic simulations[J]. Physical Review E, 2000, 62(2): 1805-1824. doi: 10.1103/PhysRevE.62.1805 [8] Rasmussen C E. Gaussian processes in machine learning[C]//Advanced Lectures on Machine Learning. Berlin: Springer, 2004: 63-71. [9] Lefèvre S, Vasquez D, Laugier C. A survey on motion prediction and risk assessment for intelligent vehicles[J]. ROBOMECH Journal, 2014, 1(1): 1-14. doi: 10.1186/s40648-014-0001-z [10] Zhang W D, Chai Q J, Zhang Q Q, et al. Obstacle-transformer: a trajectory prediction network based on surrounding trajectories[J]. IET Cyber-Systems and Robotics, 2023, 5(1): 1-8. [11] Wang W D, Qie T Q, Yang C, et al. An intelligent lane-changing behavior prediction and decision-making strategy for an autonomous vehicle[J]. IEEE Transactions on Industrial Electronics, 2022, 69(3): 2927-2937. doi: 10.1109/TIE.2021.3066943 [12] Gupta A, Johnson J, Li F F, et al. Social GAN: socially acceptable trajectories with generative adversarial networks[C]//2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Salt Lake City: IEEE, 2018: 2255-2264. [13] 向晓倩, 陈璟. 基于双重注意力时空图卷积网络的行人轨迹预测[J]. 浙江大学学报(工学版), 2024, 58(12): 2586-2595. doi: 10.3785/j.issn.1008-973X.2024.12.018Xiang Xiaoqian, Chen Jing. Pedestrian trajectory prediction based on dual-attention spatial-temporal graph convolutional network[J]. Journal of Zhejiang University (Engineering Science), 2024, 58(12): 2586-2595. doi: 10.3785/j.issn.1008-973X.2024.12.018 [14] 陈文强, 王东丹, 朱文英, 等. 基于时空图注意力网络的车辆多模态轨迹预测模型[J]. 浙江大学学报(工学版), 2025, 59(3): 443-450. doi: 10.3785/j.issn.1008-973X.2025.03.001Chen Wenqiang, Wang Dongdan, Zhu Wenying, et al. Vehicle multimodal trajectory prediction model based on spatio-temporal graph attention network[J]. Journal of Zhejiang University (Engineering Science), 2025, 59(3): 443-450. doi: 10.3785/j.issn.1008-973X.2025.03.001 [15] Sheng Z H, Huang Z L, Chen S K. Kinematics-aware multigraph attention network with residual learning for heterogeneous trajectory prediction[J]. Journal of Intelligent and Connected Vehicles, 2024, 7(2): 138-150. doi: 10.26599/JICV.2023.9210036 [16] 胡杰, 吴作伟, 张志凌, 等. 基于结构化道路的车辆多模态轨迹预测方法[J]. 中国公路学报, 2025, 38(2): 286-295. doi: 10.19721/j.cnki.1001-7372.2025.02.022Hu Jie, Wu Zuowei, Zhang Zhiling, et al. Multi-modal vehicle trajectory prediction method based on structured road[J]. China Journal of Highway and Transport, 2025, 38(2): 286-295. doi: 10.19721/j.cnki.1001-7372.2025.02.022 [17] Wang J C, Guo J Y, Feng M Y, et al. Trajectory grid diffusion for multimodal trajectory prediction in autonomous vehicles[J]. IEEE Transactions on Intelligent Vehicles, 2024, 2024: 1-15. doi: 10.1109/tiv.2024.3495037 [18] Lian J, Li S X, Yang D F, et al. Encoding the intrinsic interaction information for vehicle trajectory prediction[J]. IEEE Transactions on Intelligent Vehicles, 2024, 9(1): 2600-2611. doi: 10.1109/TIV.2023.3288976 [19] Wang Y X, Tang C, Sun L F, et al. Optimizing diffusion models forJoint trajectory prediction andControllable generation[C]//Computer Vision – ECCV 2024. Cham: Springer, 2025: 324-341. [20] 杨达, 冯婷薇, 钟家月, 等. 车路协同下交叉口前的无人车群体车道选择[J]. 西南交通大学学报, 2025, 60(5): 1250-1258, 1314. doi: 10.3969/j.issn.0258-2724.20230216Yang Da, Feng Tingwei, Zhong Jiayue, et al. Lane selection of automated vehicle groups approaching intersections based on vehicle-infrastructure cooperation[J]. Journal of Southwest Jiaotong University, 2025, 60(5): 1250-1258, 1314. doi: 10.3969/j.issn.0258-2724.20230216 [21] Wei J Q, Dolan J M, Litkouhi B. Autonomous vehicle social behavior for highway entrance ramp management[C]//2013 IEEE Intelligent Vehicles Symposium (IV). Gold Coast: IEEE, 2013: 201-207. [22] 黄玲, 崔躜, 游峰, 等. 适用于多车交互场景的车辆轨迹预测模型[J]. 吉林大学学报(工学版), 2024, 54(5): 1188-1195. doi: 10.13229/j.cnki.jdxbgxb.20220728Huang Ling, Cui Zuan, You Feng, et al. Vehicle trajectory prediction model for multi-vehicle interaction scenario[J]. Journal of Jilin University (Engineering and Technology Edition), 2024, 54(5): 1188-1195. doi: 10.13229/j.cnki.jdxbgxb.20220728 [23] Lee S, Lee J, Yu Y, et al. MART: multiscAle relational transformer networks for multi-agent trajectory prediction[C]//Computer Vision – ECCV 2024. Cham: Springer, 2025: 89-107. [24] Zgonnikov A, Abbink D, Markkula G. Should I stay or should I go Evidence accumulation drives decision making in human drivers[EB/OL]. [2025-09-10]. https://doi.org/10.31234/osf.io/p8dxn. [25] 曹越, 上官伟, VISSER Arnoud, 等. 基于模式匹配注意力机制的车辆轨迹预测[J]. 中国公路学报, 2025, 38(10): 386-401. doi: 10.19721/j.cnki.1001-7372.2025.10.029Cao Yue, Shangguan Wei, Arnoud V, et al. Pattern matching attention mechanism-based vehicle trajectory prediction[J]. China Journal of Highway and Transport, 2025, 38(10): 386-401. doi: 10.19721/j.cnki.1001-7372.2025.10.029 [26] Wang R Z, Wang S Z, Yan H, et al. WSiP: wave superposition inspired pooling for dynamic interactions-aware trajectory prediction[J]. Proceedings of the AAAI Conference on Artificial Intelligence, 2023, 37(4): 4685-4692. doi: 10.1609/aaai.v37i4.25592 [27] Foumani N M, Tan C W, Webb G I, et al. Improving position encoding of transformers for multivariate time series classification[J]. Data Mining and Knowledge Discovery, 2024, 38(1): 22-48. doi: 10.1007/s10618-023-00948-2 [28] Xu Y C, Shao W B, Li J, et al. SIND: a drone dataset at signalized intersection in China[C]//2022 IEEE 25th International Conference on Intelligent Transportation Systems (ITSC). Macau: IEEE, 2022: 2471-2478. [29] Li H B, Ren Y L, Li K X, et al. Trajectory prediction with attention-based spatial–temporal graph convolutional networks for autonomous driving[J]. Applied Sciences, 2023, 13(23): 1-12. [30] Chen J, Zhou S R, Wang W, et al. Vehicle dynamics and interaction for trajectory prediction and traffic control[J]. ACM Transactions on Autonomous and Adaptive Systems, 2025, 20(2): 1-19. [31] Alahi A, Goel K, Ramanathan V, et al. Social LSTM: human trajectory prediction in crowded spaces[C]//2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). Las Vegas: IEEE, 2016: 961-971. [32] Deo N, Trivedi M M. Convolutional social pooling for vehicle trajectory prediction[C]//2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW). Salt Lake City: IEEE, 2018: 1549-15498. [33] Liao H C, Li Z N, Shen H M, et al. BAT: behavior-aware human-like trajectory prediction for autonomous driving[J]. Proceedings of the AAAI Conference on Artificial Intelligence, 2024, 38(9): 10332-10340. doi: 10.1609/aaai.v38i9.28900 [34] Jiang W B, Ren W H, Tian J D, et al. Interaction-aware transformer network for human-object interaction detection[J]. IEEE Transactions on Cybernetics, 2025, 55(9): 4361-4373. doi: 10.1109/TCYB.2025.3587037 [35] Li X, Ying X W, Chuah M C. GRIP: graph-based interaction-aware trajectory prediction[C]//2019 IEEE Intelligent Transportation Systems Conference (ITSC). Auckland: IEEE, 2019: 3960-3966. [36] Messaoud K, Yahiaoui I, Verroust-blondet A, et al. Attention based vehicle trajectory prediction[J]. IEEE Transactions on Intelligent Vehicles, 2021, 6(1): 175-185. doi: 10.1109/TIV.2020.2991952 -
下载: