RIFT-STGNN: Robust Interleaved Frequency–Trend Spatio-Temporal Graph Neural Network for Traffic Flow Forecasting
TL;DR
RIFT-STGNN, a Robust Interleaved Frequency–Trend Spatio-Temporal Graph Neural Network for multi-step traffic flow forecasting, follows a coordinated information flow and shows competitive numerical performance relative to selected literature-reported baselines under the 12-step setting.
Abstract
Short-term traffic flow forecasting becomes especially difficult when incomplete observations, within-window frequency variation, and state-dependent sensor relations occur together. Missing readings can affect both node features and the spatial dependencies inferred from them, yet these issues are commonly modeled separately. We therefore propose RIFT-STGNN, a Robust Interleaved Frequency–Trend Spatio-Temporal Graph Neural Network for multi-step traffic flow forecasting. RIFT-STGNN follows a coordinated information flow: observation status is retained during temporal–frequency encoding, the frequency representation supports both node features and graph construction, and four graph sources are fused before each Graph-GRU update. Trend-aware temporal attention then produces direct multi-step forecasts. Experiments on PEMS03, PEMS04, PEMS07, and PEMS08 show competitive numerical performance relative to selected literature-reported baselines under the 12-step setting. On PEMS04 and PEMS08, the three-run mean MAE values are 17.73 and 13.14, respectively. These values are numerically 4.63% and 9.00% lower than the corresponding literature-reported sAMDGCN values. Component ablations, state-dependent graph analysis, and controlled missing-rate experiments support the roles of dynamic graph learning, within-window frequency encoding, and mask-aware input handling under the evaluated settings.