Trajectory Prediction Model of Electric Vehicle Autonomous Driving Based on Hybrid Attention Transformer Network

被引:0
作者
Wang, Bo [1 ]
Liu, Yao [2 ]
Wang, Rui [3 ]
Sun, Qiuye [1 ]
机构
[1] Shenyang Univ Technol, Sch Elect Engn, Shenyang, Peoples R China
[2] Northeastern Univ, Sch Comp Sci & Engn, Shenyang, Peoples R China
[3] Northeastern Univ, Sch Informat Sci & Engn, Shenyang, Peoples R China
关键词
autonomous driving; hybrid attention mechanisms; trajectory prediction; transformer network;
D O I
10.1049/itr2.70022
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Current electric vehicle trajectory prediction fails to fully consider the interaction between the target vehicle and other vehicles, resulting in poor prediction results. In order to solve this problem, this paper proposes a hybrid attention transformer network (HATN), which is designed for more accurate trajectory prediction. Firstly, based on the transformer network, this paper introduces a self-attention mechanism and a cross attention mechanism, and proposes a feature embedding and position encoding module as well as an interactive feature extraction module, so as to achieve accurate modelling of vehicle state information. With this approach, the interactive information between traffic participants can be fully extracted by effectively utilizing the map information. Secondly, a trajectory prediction decoder is proposed to expand the solution space of the model and enhance its ability to understand the real driving rules based on the driving intention recognization results of the surrounding vehicles, so that the prediction results can be more reasonable with stronger robustness. Thirdly, according to the experiments and analysis conducted based on the large-scale open datasets BDD100K and Waymo, the results show that the proposed model has a significant improvement in prediction accuracy compared with the comparison models, which verifies the effectiveness of the proposed model.
引用
收藏
页数:13
相关论文
共 48 条
[1]   Social LSTM: Human Trajectory Prediction in Crowded Spaces [J].
Alahi, Alexandre ;
Goel, Kratarth ;
Ramanathan, Vignesh ;
Robicquet, Alexandre ;
Li Fei-Fei ;
Savarese, Silvio .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :961-971
[2]   Deep Learning with Graph Convolutional Networks: An Overview and Latest Applications in Computational Intelligence [J].
Bhatti, Uzair Aslam ;
Tang, Hao ;
Wu, Guilu ;
Marjan, Shah ;
Hussain, Aamir .
INTERNATIONAL JOURNAL OF INTELLIGENT SYSTEMS, 2023, 2023
[3]   Pedestrian Graph plus : A Fast Pedestrian Crossing Prediction Model Based on Graph Convolutional Networks [J].
Cadena, Pablo Rodrigo Gantier ;
Qian, Yeqiang ;
Wang, Chunxiang ;
Yang, Ming .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2022, 23 (11) :21050-21061
[4]   Spectral Temporal Graph Neural Network for Trajectory Prediction [J].
Cao, Defu ;
Li, Jiachen ;
Ma, Hengbo ;
Tomizuka, Masayoshi .
2021 IEEE INTERNATIONAL CONFERENCE ON ROBOTICS AND AUTOMATION (ICRA 2021), 2021, :1839-1845
[5]   CAE-GAN: A hybrid model for vehicle trajectory prediction [J].
Chen, Long ;
Zhou, Qiyang ;
Cai, Yingfeng ;
Wang, Hai ;
Li, Yicheng .
IET INTELLIGENT TRANSPORT SYSTEMS, 2022, 16 (12) :1682-1696
[6]  
Chen SH, 2019, Arxiv, DOI arXiv:1906.11359
[7]  
Chen WH, 2021, PR MACH LEARN RES, V157, P454
[8]   Focal Sparse Convolutional Networks for 3D Object Detection [J].
Chen, Yukang ;
Li, Yanwei ;
Zhang, Xiangyu ;
Sun, Jian ;
Jia, Jiaya .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, :5418-5427
[9]   Deep Learning for Image and Point Cloud Fusion in Autonomous Driving: A Review [J].
Cui, Yaodong ;
Chen, Ren ;
Chu, Wenbo ;
Chen, Long ;
Tian, Daxin ;
Li, Ying ;
Cao, Dongpu .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2022, 23 (02) :722-739
[10]   A Comprehensive Review of Smart Charging Strategies for Electric Vehicles and Way Forward [J].
Dahiwale, Payal Vyankat ;
Rather, Zakir H. ;
Mitra, Indradip .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2024, 25 (09) :10462-10482