TY - JOUR
T1 - Railway network delay evolution
T2 - A heterogeneous graph neural network approach
AU - Li, Zhongcan
AU - Huang, Ping
AU - Wen, Chao
AU - Dong, Wei
AU - Ji, Yindong
AU - Rodrigues, Filipe
N1 - Publisher Copyright:
© 2024 Elsevier B.V.
PY - 2024
Y1 - 2024
N2 - Accurate delay evolution prediction plays a pivotal role in train rescheduling decision-making for the railway network. Existing studies on delay prediction predominantly centered around predicting delays for each train in the subsequent stations (i.e., following a train-oriented perspective). Furthermore, train operations in the railway network involve different types of entities (stations, trains, etc.), making the current graph/network models with homogenous nodes (i.e., the same kind of nodes) incapable of effectively capturing the interactions between the entities. This paper develops a network-oriented model to investigate the train delay evolution on railway networks, by predicting the delays of running trains in the network after a given time interval. The proposed model combines the GraphSAGE graph neural network (GNN) and the heterogeneous graph neural network (HetGNN) architecture, thus called SAGE-Het, enabling it to capture interactions between heterogeneous nodes (i.e., different types of nodes) based on different edges (e.g., edges between trains, trains, and stations). Additionally, SAGE-Het allows for flexible inputs in contrast to conventional machine learning techniques, whose inputs must meet the consistent dimension requirement (e.g., in the form of rectangular or grid-like arrays). The performance and robustness of the suggested SAGE-Het model are assessed in experiments on the data from two sub-networks of the China railway network. The experimental results demonstrate that SAGE-Het outperforms the existing delay prediction methods and some advanced HetGNNs used for prediction tasks in other domains; SAGE-Het demonstrates excellent scalability, capable of handling various types of nodes; the predictive performances of SAGE-Het under different prediction time horizons (10/20/30 min ahead) all exhibit better performance over other baselines; the accuracies are over 90 % under the permissible 3-minute errors for the three prediction time horizons. Specifically, the impact of train interactions on delay evolution is investigated based on the flexible input characteristic of the proposed model. The results illustrate that train interactions become subtle with the increase of train headways. This finding directly contributes to decision-making in situations where conflict resolution or train-canceling actions are needed.
AB - Accurate delay evolution prediction plays a pivotal role in train rescheduling decision-making for the railway network. Existing studies on delay prediction predominantly centered around predicting delays for each train in the subsequent stations (i.e., following a train-oriented perspective). Furthermore, train operations in the railway network involve different types of entities (stations, trains, etc.), making the current graph/network models with homogenous nodes (i.e., the same kind of nodes) incapable of effectively capturing the interactions between the entities. This paper develops a network-oriented model to investigate the train delay evolution on railway networks, by predicting the delays of running trains in the network after a given time interval. The proposed model combines the GraphSAGE graph neural network (GNN) and the heterogeneous graph neural network (HetGNN) architecture, thus called SAGE-Het, enabling it to capture interactions between heterogeneous nodes (i.e., different types of nodes) based on different edges (e.g., edges between trains, trains, and stations). Additionally, SAGE-Het allows for flexible inputs in contrast to conventional machine learning techniques, whose inputs must meet the consistent dimension requirement (e.g., in the form of rectangular or grid-like arrays). The performance and robustness of the suggested SAGE-Het model are assessed in experiments on the data from two sub-networks of the China railway network. The experimental results demonstrate that SAGE-Het outperforms the existing delay prediction methods and some advanced HetGNNs used for prediction tasks in other domains; SAGE-Het demonstrates excellent scalability, capable of handling various types of nodes; the predictive performances of SAGE-Het under different prediction time horizons (10/20/30 min ahead) all exhibit better performance over other baselines; the accuracies are over 90 % under the permissible 3-minute errors for the three prediction time horizons. Specifically, the impact of train interactions on delay evolution is investigated based on the flexible input characteristic of the proposed model. The results illustrate that train interactions become subtle with the increase of train headways. This finding directly contributes to decision-making in situations where conflict resolution or train-canceling actions are needed.
KW - Delay evolution
KW - Graph neural network
KW - Heterogeneous node
KW - High-speed railway network
KW - Train interaction
U2 - 10.1016/j.asoc.2024.111640
DO - 10.1016/j.asoc.2024.111640
M3 - Journal article
AN - SCOPUS:85192276710
SN - 1568-4946
VL - 159
JO - Applied Soft Computing
JF - Applied Soft Computing
M1 - 111640
ER -