A CNN-Bi_LSTM parallel network approach for train travel time prediction

被引:39
作者
Guo, Jingwei [1 ,2 ]
Wang, Wei [2 ]
Tang, Yinying [3 ,4 ]
Zhang, Yongxiang [3 ,4 ]
Zhuge, Hengying [5 ]
机构
[1] Luoyang Polytech, Luoyang 471000, Peoples R China
[2] Henan Polytech Univ, Sch Energy Sci & Engn, Jiaozuo 454000, Henan, Peoples R China
[3] Southwest Jiaotong Univ, Sch Transportat & Logist, Chengdu 611756, Peoples R China
[4] Southwest Jiaotong Univ, Natl United Engn Lab Integrated & Intelligent Tran, Chengdu 610031, Peoples R China
[5] China Acad Railway Sci Co Ltd, Transportat & Econ Res Inst, Beijing 100081, Peoples R China
基金
中国国家自然科学基金;
关键词
Railway transportation; Train travel time prediction; Deep learning; China Railway Express; CNN-bi_LSTM; CONVOLUTIONAL NEURAL-NETWORK; SPEED PREDICTION; DELAY PREDICTION; MODEL;
D O I
10.1016/j.knosys.2022.109796
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Convolutional neural networks (CNNs) offer a broad technical framework to deal with spatial feature extraction and nonlinearity capture, whereas they cannot process sequence data and cannot capture the dependencies between the sequence information. Therefore, this paper proposes an improved deep learning model CNN-Bi_LSTM that combines the CNN, Bi_LSTM (i.e., bidirectional long short-term memory network), and fully connected neural network (FCNN) to process the complex dataset for the train travel time prediction. As a result, the presented deep learning framework can capture both the long-and short-term features of complex datasets and the characteristics of time series data. Besides, the multi-feature data fusion processing method is realized with the help of a parallel learning mechanism and the fully connected neural network. Based on a real-life case study of China Railway Express (Chengdu-Europe), the superiority of the CNN-Bi_LSTM model on the train travel time prediction is systemically evaluated and demonstrated, compared with the baseline models of Holt-Winters model, random forest (RF), support vector regression (SVR), LSTM, Bi_LSTM, LSTM with attention mechanism (LSTM_Attention), convolution-based LSTM (CLSTM), CNN_LSTM, hybrid deep learning model (CNN_GRU1), temporal convolutional network (TCN), and parallel deep learning model (CNN_GRU2). Moreover, the values of MSE, RMSE, MAPE, and MAE obtained from the CNN-Bi_LSTM model are equal to 4.647, 2.156, 2.643, and 1.769 respectively Consequently, it is concluded that our proposed CNN-Bi_LSTM model has good prediction results, and it is suitable for the train travel time prediction of China Railway Express.(c) 2022 Elsevier B.V. All rights reserved.
引用
收藏
页数:15
相关论文
共 62 条
[61]   LSTM network: a deep learning approach for short-term traffic forecast [J].
Zhao, Zheng ;
Chen, Weihai ;
Wu, Xingming ;
Chen, Peter C. Y. ;
Liu, Jingmeng .
IET INTELLIGENT TRANSPORT SYSTEMS, 2017, 11 (02) :68-75
[62]   Using Deep Learning to Forecast Maritime Vessel Flows [J].
Zhou, Xiangyu ;
Liu, Zhengjiang ;
Wang, Fengwu ;
Xie, Yajuan ;
Zhang, Xuexi .
SENSORS, 2020, 20 (06)