Multi-head attention-based intelligent vehicle lane change decision and trajectory prediction model in highways

被引:8
作者
Cai, Junyu [1 ]
Jiang, Haobin [1 ]
Wang, Junyan [2 ]
Li, Aoxue [1 ]
机构
[1] Jiangsu Univ, Automot Engn Res Inst, Zhenjiang, Peoples R China
[2] Zhenjiang Coll, Sch Transportat, Zhenjiang, Peoples R China
关键词
deep neural networks; multi-head attention; naturalistic driving data; trajectory prediction; vehicle trajectory; MOTION;
D O I
10.1080/15472450.2024.2341392
中图分类号
U [交通运输];
学科分类号
08 ; 0823 ;
摘要
With the aim to improve the interaction between intelligent vehicles and human drivers, this article proposes the MCLG (multi-head attention + convolutional social pooling + long short-term memory + Gaussian mixture model) lane change decision and trajectory prediction model, which includes a lane-changing intention decision module. The model comprises a lane change decision module responsible for determining three lane change intentions: left lane change, right lane change, and car-following. Subsequently, a multi-head attention mechanism processes complex vehicle interaction information to enhance modeling accuracy and intelligence. In addition, uncertainty in trajectory prediction is considered by using multimodal trajectory prediction and Gaussian mixture model, and diversity and uncertainty are combined by combining trajectory prediction from several different modalities through probabilistic combinatorial sampling patterns. Test results indicate that the MCLG model, based on the multi-head attention module, outperforms existing methods in trajectory prediction. The decision module, which takes interactive information into account, exhibits higher predictability and accuracy. Furthermore, the MCLG model, considering the lane-changing decision module, significantly enhances trajectory prediction accuracy, providing robust decision-making support for autonomous driving systems.
引用
收藏
页数:18
相关论文
共 23 条
[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]   SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation [J].
Badrinarayanan, Vijay ;
Kendall, Alex ;
Cipolla, Roberto .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (12) :2481-2495
[3]   Recent advances in decision trees: an updated survey [J].
Costa, Vinicius G. ;
Pedreira, Carlos E. .
ARTIFICIAL INTELLIGENCE REVIEW, 2023, 56 (05) :4765-4800
[4]  
Coylar J., 2006, Technical Report No. FHWA-HRT-07-030
[5]  
Deo N, 2018, IEEE INT VEH SYM, P1179, DOI 10.1109/IVS.2018.8500493
[6]   How Would Surround Vehicles Move? A Unified Framework for Maneuver Classification and Motion Prediction [J].
Deo, Nachiket ;
Rangesh, Akshay ;
Trivedi, Mohan M. .
IEEE TRANSACTIONS ON INTELLIGENT VEHICLES, 2018, 3 (02) :129-140
[7]   Interpretable AI Agent Through Nonlinear Decision Trees for Lane Change Problem [J].
Ghosh, Abhiroop ;
Dhebar, Yashesh ;
Guha, Ritam ;
Deb, Kalyanmoy ;
Nageshrao, Subramanya ;
Zhu, Ling ;
Tseng, Eric ;
Filev, Dimitar .
2021 IEEE SYMPOSIUM SERIES ON COMPUTATIONAL INTELLIGENCE (IEEE SSCI 2021), 2021,
[8]   Social GAN: Socially Acceptable Trajectories with Generative Adversarial Networks [J].
Gupta, Agrim ;
Johnson, Justin ;
Li Fei-Fei ;
Savarese, Silvio ;
Alahi, Alexandre .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :2255-2264
[9]  
Hasan M., 2021, arXiv, DOI DOI 10.48550/ARXIV.2104.14079
[10]  
Hochreiter S, 1997, NEURAL COMPUT, V9, P1735, DOI [10.1162/neco.1997.9.8.1735, 10.1162/neco.1997.9.1.1, 10.1007/978-3-642-24797-2]