China Safety Science Journal ›› 2022, Vol. 32 ›› Issue (6): 123-130.doi: 10.16265/j.cnki.issn1003-3033.2022.06.2634

• Safety engineering technology • Previous Articles     Next Articles

A train running time prediction model based on domestic and foreign railway operation data

TANG Tao1(), GAN Jing2   

  1. 1 China Railway Siyuan Survey and Design Group Co., Ltd., Wuhan Hubei 430063,China
    2 School of Modern Posts, Nanjing University of Posts and Telecommunications, Nanjing Jiangsu 210003,China
  • Received:2022-01-10 Revised:2022-04-20 Online:2022-06-28 Published:2022-12-28

Abstract:

In order to accurately predict train running time while considering requirement on accuracy and timeliness of its operation, a time prediction model was established. Firstly, data of railway timetable were extracted and analyzed to determine possible influencing factors of its running time, and Box-Cox transformation was used to normalize operation data considering their non-normality. Then, the model's input features and hyper-parameters were optimized based on decision tree and grid-search algorithm, respectively, and its performance was improved. Finally, a train running time prediction model was established by adopting HGBT based on optimized input features and hyper-parameters, and operation data of a Chinese and European railway were utilized to evaluate works at each stage. The results show that Box-Cox transformation can significantly improve data normality and goodness-of-fit of the prediction model, while grid-search algorithm can simultaneously improve the model's efficiency and accuracy. Compared with other commonly used machine learning algorithms for train running time prediction, the proposed HGBT model features higher accuracy and efficiency.

Key words: railway operation data, train running time, prediction model, Box-Cox transformation, decision tree, histogram-based gradient boosting tree (HGBT)