交通运输系统工程与信息 ›› 2023, Vol. 23 ›› Issue (2): 148-160.DOI: 10.16097/j.cnki.1009-6744.2023.02.016

• 系统工程理论与方法 • 上一篇    下一篇

基于双向长短期记忆网络的公交到站时间预测模型

张兵*1,周丹丹1,孙健2,倪训友1   

  1. 1. 华东交通大学,交通运输工程学院,南昌 330013;2. 长安大学,未来交通学院,西安 710064
  • 收稿日期:2022-12-11 修回日期:2023-01-26 接受日期:2023-02-07 出版日期:2023-04-25 发布日期:2023-04-19
  • 作者简介:张兵(1981- ),男,山东济宁人,副教授,博士
  • 基金资助:
    国家自然科学基金(71971138, 52162042, 72161012)

Bus Arrival Time Prediction Model Based on Bidirectional Long Short-term Memory Network

ZHANG Bing*1, ZHOU Dan-dan1, SUN Jian2, NI Xun-you1   

  1. 1. School of Transportation Engineering, East China Jiaotong University, Nanchang 330013, China; 2. College of Future Transportation, Chang'an University, Xi'an 710064, China
  • Received:2022-12-11 Revised:2023-01-26 Accepted:2023-02-07 Online:2023-04-25 Published:2023-04-19
  • Supported by:
    National Natural Science Foundation of China (71971138, 52162042, 72161012)

摘要: 为实现准确预测公交到站时间,提高城市公交出行分担率,本文提出一种基于双向长短期记忆神经网络(BiLSTM)并考虑超参数寻优的公交到站时间预测模型。通过引入非线性收敛因子、正弦余弦算子及自适应参数改进海鸥算法对双向 LSTM 模型实现超参数寻优,并增加Attention机制以提高双向LSTM处理信息能力,构建改进海鸥算法优化增加Attention机制的双向LSTM(ISOA-BiLSTM-Attention)预测模型。使用中国江西省南昌市220路公交GPS数据分方向和分时段预测车辆到站时间,验证模型预测精度。结果表明:整体上来说,Attention机制优化后的双向LSTM模型比单独采用双向LSTM模型预测精度更好;进的海鸥算法可对双向LSTMAttention 模型实现较好的优化效果,相较于现有模型及标准海鸥算法(SOA)优化双向LSTMAttention模型,ISOA-BiLSTM-Attention对于不同方向及不同时段公交到站时间预测的平均绝对百分比误差、均方根误差及平均绝对误差至少分别降低 5.96%、9.87%及 7.99%;同时,ISOABiLSTM-Attention具有最大的模型决定系数 R2 值体现了该预测模型泛化能力及稳定性较好,可针对公交到站时间进行较为准确地拟合。

关键词: 城市交通, 公交到站时间预测, 改进海鸥优化算法, 双向LSTM模型, Attention机制

Abstract: To improve the accuracy of bus arrival time prediction and increase the bus usage in the cities, this paper proposes a bus arrival time prediction model based on a bidirectional Long Short-term Memory (BiLSTM) neural network and the hyperparameter search. The improved seagull algorithm optimization adding Attention mechanism to bidirectional LSTM (ISOA-BiLSTM-Attention) prediction model was developed by introducing nonlinear convergence factor, sine cosine operator, and adaptive parameters to improve the seagull algorithm to achieve hyperparametric optimization of the bidirectional LSTM model. The Attention mechanism was added to improve the information processing ability of bidirectional LSTM. Then, the trajectory data of bus route 220 in Nanchang, Jiangxi Province of China, were used to predict the bus arrival time for different directions and time to validate the model prediction accuracy. The results show that, the proposed model has better performance than the traditional bidirectional LSTM model. The improved seagull algorithm can achieve a better optimization effect on the bidirectional LSTM-Attention model. Compared with the existing model and seagull algorithm (SOA) optimized bidirectional LSTM-Attention model, the mean absolute percentage error was reduced by 5.96%, the root mean square error was reduced by 9.87%, and the mean absolute error was reduced by 7.99% in the ISOA-BiLSTM- Attention for bus arrival time prediction. Moreover, the ISOA-BiLSTM-Attention has the largest model decision coefficient R2 value, which indicates the good generalization ability and stability of the proposed model, and can provide good fitness of accuracy for bus arrival time.

Key words: urban traffic, bus arrival time prediction, improved seagull optimization algorithm, bidirectional LSTM model, Attention mechanism

中图分类号: