Long Short-Term Memory Neural Network Applied to Train Dynamic Model and Speed Prediction

Algorithms

Chunhai Gao¹  Tao Tang²  Zhen Li² 

[1] National Engineering Laboratory for Urban Rail Transit Communication and Operation Control, Beijing 100044, China;School of Electronic and Information Engineering, Beijing Jiaotong University, Beijing 100044, China;

关键词: train dynamic model;    train speed prediction;    long short-term memory neural network;   

DOI  :  10.3390/a12080173

来源: DOAJ

【 摘 要 】

The automatic train operation system is a significant component of the intelligent railway transportation. As a fundamental problem, the construction of the train dynamic model has been extensively researched using parametric approaches. The parametric based models may have poor performances due to unrealistic assumptions and changeable environments. In this paper, a long short-term memory network is carefully developed to build the train dynamic model in a nonparametric way. By optimizing the hyperparameters of the proposed model, more accurate outputs can be obtained with the same inputs of the parametric approaches. The proposed model was compared with two parametric methods using actual data. Experimental results suggest that the model performance is better than those of traditional models due to the strong learning ability. By exploring a detailed feature engineering process, the proposed long short-term memory network based algorithm was extended to predict train speed for multiple steps ahead.

【 授权许可】

Unknown