arXiv AI

RSTGCN: Railway-centric Spatio-Temporal Graph Convolutional Network for Train Delay Prediction

The paper introduces RSTGCN, a Railway-centric Spatio-Temporal Graph Convolutional Network that predicts average arrival delays for all incoming trains at a specific station during a given time period. It incorporates train frequency-aware spatial attention and other architectural innovations to improve predictive accuracy. Experiments on a newly released dataset covering 4,735 Indian Railway stations show RSTGCN outperforms state‑of‑the‑art baselines by 18% in MAE, 14% in MAPE, and 1–8% in RMSE.

arXiv Machine Learning
Jun 4

RIDE: An Open Dataset and Benchmark for Train Delay Prediction

arXiv:2606. 05070v1 Announce Type: new Abstract: Train delay prediction is an important problem for both passengers and railway operators, yet progress in the field remains difficult to assess due to the lack of standardized datasets, prediction targets, and evaluation protocols.

By Cl\'ement Elliker, Mathis Le Bail, Cl\'ement Mantoux, Jesse Read, Sonia Vanier
arXiv Machine Learning
Sep 11

Predicting Train Delays in Finland Using Machine Learning and Weather Data

The paper presents a machine‑learning approach to predict train delays in Finland using the Finland Integrated Train‑Weather (FI‑TW) dataset, which merges railway operational records with data from about 200 nationwide weather sensors. Three feature sets were tested with XGBoost at Oulu central station: full weather features, instant observations only, and derived weather categories. The category‑based features—hierarchical classes such as Blizzard, Heavy Snow, and Extreme Cold—yielded the best performance, achieving an R² of 0.78, RMSE of 8.5 min, and MAE of 3.7 min, an 11 % R² improvement and 10 % error reduction over the other configurations.

By Vinicius Pozzobon Borin, Jean Michel de Souza Sant'Ana, Nurul Huda Mahmood
arXiv AI
Sep 24

Learning Local Heterogeneity and Cross-Region Context for Large-Scale Traffic Forecasting

The paper introduces LoReST, a Local-Region Spatial Temporal network designed for large-scale traffic forecasting. LoReST captures local spatial heterogeneity by relation-aware aggregation within node neighborhoods and incorporates cross-region context through mean pooling, inter-region attention, and broadcasting back to nodes. Experiments on the LargeST benchmark demonstrate significant improvements, reducing MAE, RMSE, and MAPE by 4.78%, 3.60%, and 5.75% respectively.

By Qi Feng, Zidong Wang, Bo Li, Xiaoguang Gao, Jiayu Zhang, Chenfeng Wang, Kaifang Wan
arXiv Machine Learning
Sep 18

Explaining spatial information flow in short-term traffic forecasting models using a gated graph attention network

The paper introduces a gated graph attention network (GAT) to better explain spatial information flow in short‑term traffic forecasting models. By adding a gate that learns how much of a sensor’s updated state comes from its neighbors versus itself, the authors can progressively withdraw neighbor information and observe its impact on accuracy. Applied to the ST‑MetaNet architecture on data from 498 loop detectors in England, the gated GAT shows that neighbor influence is higher for busy sensors, improves accuracy slightly with mild regularisation, and reveals that the two GAT layers are largely redundant.

By Yue Li, Shujuan Chen, Ying Jin