• ISSN 0258-2724
  • CN 51-1277/U
  • EI Compendex
  • Scopus 收录
  • 全国中文核心期刊
  • 中国科技论文统计源期刊
  • 中国科学引文数据库来源期刊

基于时空特征增强的交叉口多车轨迹预测方法

杨澜 王筱珂 房山 瞿广跃 袁梦 李小龙

杨澜, 王筱珂, 房山, 瞿广跃, 袁梦, 李小龙. 基于时空特征增强的交叉口多车轨迹预测方法[J]. 西南交通大学学报. doi: 10.3969/j.issn.0258-2724.20250298
引用本文: 杨澜, 王筱珂, 房山, 瞿广跃, 袁梦, 李小龙. 基于时空特征增强的交叉口多车轨迹预测方法[J]. 西南交通大学学报. doi: 10.3969/j.issn.0258-2724.20250298
YANG Lan, WANG Xiaoke, FANG Shan, QU Guangyue, YUAN Meng, LI Xiaolong. Method for Multi-Vehicle Trajectory Prediction Based on Spatio-Temporal Feature Enhancement at Intersections[J]. Journal of Southwest Jiaotong University. doi: 10.3969/j.issn.0258-2724.20250298
Citation: YANG Lan, WANG Xiaoke, FANG Shan, QU Guangyue, YUAN Meng, LI Xiaolong. Method for Multi-Vehicle Trajectory Prediction Based on Spatio-Temporal Feature Enhancement at Intersections[J]. Journal of Southwest Jiaotong University. doi: 10.3969/j.issn.0258-2724.20250298

基于时空特征增强的交叉口多车轨迹预测方法

doi: 10.3969/j.issn.0258-2724.20250298
基金项目: 国家自然科学基金项目(52472446);陕西省留学人员科技活动择优资助项目(2023001);中央高校基本科研业务费专项(300102245104)
详细信息
    作者简介:

    杨澜(1985—),女,正高级工程师,博士,研究方向为自动驾驶决策控制及其测试技术,E-mail:lanyang@chd.edu.cn

    通讯作者:

    房山(1994—),男,讲师,博士,研究方向为宏观交通流调控及自动驾驶决策控制,E-mail: fang6100146@gmail.com

  • 中图分类号: TP181

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%,验证了在复杂交叉口场景下的预测精度与鲁棒性优势.

     

  • 图 1  模型框架

    Figure 1.  Diagram of model framework

    图 2  车辆空间关联性分析

    Figure 2.  Spatial correlation analysis of vehicle

    图 3  空间图卷积操作

    Figure 3.  Spatial graph convolution operation

    图 4  tAPE-Transformer结构

    Figure 4.  tAPE-Transformer structure

    图 5  数据处理及对比

    Figure 5.  Data processing and comparison

    图 6  训练与验证收敛曲线

    Figure 6.  Training and validation convergence curves

    图 7  单车预测结果

    Figure 7.  Single vehicle prediction results

    图 8  多车预测轨迹分布

    Figure 8.  Predicted trajectory distribution of multiple vehicles

    表  1  不同交互机制误差对比

    Table  1.   Comparison of errors among different interaction mechanisms

    模型2 s预测时域3 s预测时域平均值
    eADEeFDEeADEeFDEeADEeFDE
    ONE-GCN2.66916.02523.59868.38893.13397.2071
    RD-GCN1.87653.62972.98376.22082.43014.9253
    WAGT0.77202.05201.73064.52581.25133.2889
    下载: 导出CSV

    表  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
    下载: 导出CSV

    表  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
    下载: 导出CSV

    表  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
    下载: 导出CSV

    表  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
    下载: 导出CSV
  • [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.20220321

    Yang 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.007

    Qu 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.018

    Xiang 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.001

    Chen 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.022

    Hu 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.20230216

    Yang 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.20220728

    Huang 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.029

    Cao 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
  • 加载中
图(8) / 表(5)
计量
  • 文章访问数:  32
  • HTML全文浏览量:  18
  • PDF下载量:  0
  • 被引次数: 0
出版历程
  • 收稿日期:  2025-06-01
  • 修回日期:  2026-01-26
  • 网络出版日期:  2026-06-11

目录

    /

    返回文章
    返回