首页|期刊导航|重庆理工大学学报|融合多头注意力的时空交互感知车辆轨迹预测模型

融合多头注意力的时空交互感知车辆轨迹预测模型OA

Spatiotemporal interaction perception vehicle trajectory prediction model with multi-head attention

中文摘要英文摘要

高精度车辆轨迹预测对提升自动驾驶安全与交通效率至关重要.提出融合多头注意力机制与双向长短期记忆递归神经网络(spatiotemporal attention and bidirectionalLSTM,STA-BiLSTM)轨迹预测模型,通过分层提取单车轨迹特征与协同表征多车间时空交互依赖,有效增强复杂动态场景下的预测能力.该模型首先采用卷积神经网络-长短期记忆网络(CNN-LSTM)编码单车历史轨迹特征;再通过多头注意力机制实现车辆间空间交互的自适应量化;最后利用双向长短期记忆网络(BiLSTM)捕捉交互关系的时序演化规律.试验结果表明,所提模型在 5s 时的均方根误差(RMSE)对比 double transformer(DT)下降9.8%,且具有较强的可解释性.通过单车和环境车辆时空依赖关系的耦合建模,为复杂交通环境下的轨迹预测提供可行解决方案.

High-precision vehicle trajectory prediction is crucial to improving both the safety and operational efficiency of autonomous driving systems.This paper proposes a spatiotemporal attention and bidirectional long short-term memory(STA-BiLSTM)trajectory prediction model that integrates multi-head attention mechanisms with bidirectional LSTM networks.By hierarchically extracting single-vehicle trajectory features and jointly modeling spatiotemporal interdependencies among multiple vehicles,the proposed model effectively enhances prediction accuracy in complex dynamic scenarios.First,the model employs a CNN-LSTM architecture to encode historical trajectory features of individual vehicles.Then,a multi-head attention mechanism is introduced to adaptively model spatial interactions among vehicles.Finally,a BiLSTM network is employed to capture the temporal evolution of these inter-vehicle dependencies.Experimental results demonstrate that the proposed model reduces the root mean square error(RMSE)at a 5s prediction horizon by9.8%compared with the Double Transformer baseline,while maintaining strong interpretability.By explicitly modeling the spatiotemporal dependencies between ego and surrounding vehicles,this study may provide an effective solution for trajectory prediction in complex traffic environments.

陈峥;靳云淇;郑嘉;魏福星;沈世全;郭凤香

昆明理工大学 交通工程学院,昆明 650500昆明理工大学 交通工程学院,昆明 650500昆明理工大学 交通工程学院,昆明 650500昆明理工大学 交通工程学院,昆明 650500昆明理工大学 交通工程学院,昆明 650500昆明理工大学 交通工程学院,昆明 650500

交通工程

车辆工程轨迹预测注意力机制时空建模双向长短时记忆网络深度学习

automotive engineeringtrajectory predictionattention mechanismspatiotemporal modelingbidirectional longdeep learning

《重庆理工大学学报》 2026 (7)

1-10,10

云南省重大科技专项(202503AA080005)云南省车路协同控制与运行安全创新团队(202505AS350024)云南省"兴滇英才支持计划"云岭学者专项(KKRC202402005)

10.3969/j.issn.1674-8425(z).2026.04.001

评论