Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

UniST-Pred: A Robust Unified Framework for Spatio-Temporal Traffic Forecasting in Transportation Networks Under Disruptions

About

Spatio-temporal traffic forecasting is a core component of intelligent transportation systems, supporting various downstream tasks such as signal control and network-level traffic management. In real-world deployments, forecasting models must operate under structural and observational uncertainties, conditions that are rarely considered in model design. Recent approaches achieve strong short-term predictive performance by tightly coupling spatial and temporal modeling, often at the cost of increased complexity and limited modularity. In contrast, efficient time-series models capture long-range temporal dependencies without relying on explicit network structure. We propose UniST-Pred, a unified spatio-temporal forecasting framework that first decouples temporal modeling from spatial representation learning, then integrates both through adaptive representation-level fusion. To assess robustness of the proposed approach, we construct a dataset based on an agent-based, microscopic traffic simulator (MATSim) and evaluate UniST-Pred under severe network disconnection scenarios. Additionally, we benchmark UniST-Pred on standard traffic prediction datasets, demonstrating its competitive performance against existing well-established models despite a lightweight design. The results illustrate that UniST-Pred maintains strong predictive performance across both real-world and simulated datasets, while also yielding interpretable spatio-temporal representations under infrastructure disruptions. The source code and the generated dataset are available at https://anonymous.4open.science/r/UniST-Pred-EF27

Yue Wang, Areg Karapetyan, Djellel Difallah, Samer Madanat• 2026

Related benchmarks

TaskDatasetResultRank
Traffic Flow ForecastingSimSF-Bay
RMSE3.6
11
Traffic Outflow ForecastingNYCTaxi
RMSE13.39
11
Traffic speed forecastingPEMS-BAY
RMSE4.2
11
Traffic ForecastingSimSF-Bay
Params3.21e+6
2
Traffic ForecastingPEMS-BAY
Parameters1.66e+7
2
Traffic ForecastingNYCTaxi
Parameters1.68e+5
2
Showing 6 of 6 rows

Other info

Follow for update