论文标题
转基金会:长期序列的汽车跟踪轨迹通过变压器预测
TransFollower: Long-Sequence Car-Following Trajectory Prediction through Transformer
论文作者
论文摘要
跟随汽车的指代于控制过程,其中以下车辆(FV)试图通过对前方车辆的行为来调整其加速度来保持自身与领先车辆(LV)之间的安全距离。相应的跟随型号描述了一辆车在交通流中的另一个车辆的方式,构成了微观交通模拟和智能车辆开发的基石。汽车跟随模型的一个主要动机是复制人类驾驶员的纵向驾驶轨迹。为了模拟未来动作对历史驾驶情况的长期依赖性,我们基于基于注意力的变压器模型开发了一个长期的汽车跟踪轨迹预测模型。该模型遵循编码器架构的一般格式。编码器将历史速度和间隔数据作为输入,并使用多头自我注意力对历史驾驶环境形成混合表示。解码器将未来的LV速度配置文件作为输入,并以生成方式输出预测的未来FV速度配置文件(而不是自动回归方式,避免复合错误)。通过编码器和解码器之间的跨注意,解码器学会了在历史驾驶和未来LV速度之间建立联系,这是基于对未来FV速度的预测。我们通过从上海自然主义驾驶研究(SH-NDS)提取的112,597个现实世界中的实际事件来训练和测试模型。结果表明,该模型优于传统的智能驱动器模型(IDM),完全连接的神经网络模型以及长期的短期内存(LSTM)模型,就长期轨迹预测的准确性而言。我们还可以观察到自我注意事项和交叉注意热图,以解释该模型如何得出其预测。
Car-following refers to a control process in which the following vehicle (FV) tries to keep a safe distance between itself and the lead vehicle (LV) by adjusting its acceleration in response to the actions of the vehicle ahead. The corresponding car-following models, which describe how one vehicle follows another vehicle in the traffic flow, form the cornerstone for microscopic traffic simulation and intelligent vehicle development. One major motivation of car-following models is to replicate human drivers' longitudinal driving trajectories. To model the long-term dependency of future actions on historical driving situations, we developed a long-sequence car-following trajectory prediction model based on the attention-based Transformer model. The model follows a general format of encoder-decoder architecture. The encoder takes historical speed and spacing data as inputs and forms a mixed representation of historical driving context using multi-head self-attention. The decoder takes the future LV speed profile as input and outputs the predicted future FV speed profile in a generative way (instead of an auto-regressive way, avoiding compounding errors). Through cross-attention between encoder and decoder, the decoder learns to build a connection between historical driving and future LV speed, based on which a prediction of future FV speed can be obtained. We train and test our model with 112,597 real-world car-following events extracted from the Shanghai Naturalistic Driving Study (SH-NDS). Results show that the model outperforms the traditional intelligent driver model (IDM), a fully connected neural network model, and a long short-term memory (LSTM) based model in terms of long-sequence trajectory prediction accuracy. We also visualized the self-attention and cross-attention heatmaps to explain how the model derives its predictions.