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

被引:0
|
作者
Zhang B. [1 ]
Zhou D.-D. [1 ]
Sun J. [2 ]
Ni X.-Y. [1 ]
机构
[1] School of Transportation Engineering, East China Jiaotong University, Nanchang
[2] College of Future Transportation, Chang'an University, Xi'an
基金
中国国家自然科学基金;
关键词
Attention mechanism; bidirectional LSTM model; bus arrival time prediction; improved seagull optimization algorithm; urban traffic;
D O I
10.16097/j.cnki.1009-6744.2023.02.016
中图分类号
学科分类号
摘要
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. © 2023 Science Press. All rights reserved.
引用
收藏
页码:148 / 160
页数:12
相关论文
共 15 条
  • [1] REN Y, LV Y B, MA J H, Et al., Bus arrival time prediction based on particle filter, Journal of Transportation Systems Engineering and Information Technology, 16, 6, pp. 142-146, (2016)
  • [2] SHARMILA R B, VELAGA N R, HOUDHARY P C., Bus arrival time prediction and measure of uncertainties using survival models, Iet Intelligent Transport Systems, 14, 8, pp. 900-907, (2020)
  • [3] LAI Y X, YANG X, CAO Q, Et al., A bus running length prediction method based on Gradient Boosting, Big Data Research, 5, 5, pp. 58-78, (2019)
  • [4] PANG J, HUANG J, DU Y, Et al., Learning to predict bus arrival time from heterogeneous measurements via recurrent neural network, IEEE Transactions on Intelligent Transportation Systems, 9, 20, pp. 3283-3293, (2019)
  • [5] CHEN S K, SUN D., A multistate-based travel me schedule model for fixed transit route, Transportation Letters-the International Journal of Transportation Research, 11, 1, pp. 33-42, (2019)
  • [6] JI Y J, LU J W, CHEN X S, Et al., Prediction model of bus arrival Time based on particle swarm optimization and wavelet neural network, Journal of Transportation Systems Engineering and Information Technology, 16, 3, pp. 60-66, (2016)
  • [7] KUMAR B A, VANAJAKSHI L, SUBRAMANIAN S C., Bus travel time prediction using a time-space discretization approach, Transportation Research Part C: Emerging Technologies, 79, pp. 308-332, (2017)
  • [8] AN Y H, MA J H, LIU H Y, Et al., Prediction of bus arrival time based on deep learning and particle filter, Journal of Beijing Jiaotong University, 46, 1, pp. 88-97, (2022)
  • [9] ZHANG X M, CHEN X M, ZHANG X., Prediction of bus arrival time based on nonparametric regression and particle filtering model, Journal of Highway and Transportation Research and Development, 37, 4, pp. 118-124, (2020)
  • [10] XIE Z Y, HE Y R, CHEN C C, Et al., Multistep prediction of bus arrival time with the recurrent neural network, Mathematical Problems in Engineering, 2021, pp. 1-14, (2021)