arXiv Machine Learning

A Multi-Resolution Multi-Domain Pre-Training Framework for Universal Traffic Forecasting

arXiv AI
Sep 18

FedeRICo: Federated Region-Influenced Coupling for Traffic Flow Prediction

FedeRICo is a federated traffic forecasting framework that addresses heterogeneity across client sensor subgraphs by combining gradient-level collaboration with boundary-aware residual communication. It uses a dual-branch architecture: a globally guided branch for transferable forecasting structure and a private residual branch that preserves client-specific corrections and incorporates boundary residual signals. Experiments on four real-world traffic benchmarks show that FedeRICo outperforms state‑of‑the‑art federated spatial‑temporal baselines while keeping training runtime competitive.

By Fermin Orozco, Man Luo, Johan Wahlstr\"om
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
Aug 31

Learning to Transfer Across Modes: Towards Unified Urban Mobility Forecasting

The paper introduces TransMod, a unified framework for forecasting urban mobility demand across multiple transportation modes. It creates a shared zone-level spatial representation to align systems with different spatial granularities, reducing structural mismatch and distributional shift. TransMod then learns transferable spatio-temporal patterns from data-rich source modes and adapts them to data-scarce target modes, improving forecasting performance when target data is limited.

By Yixuan Zhao, Man Luo
arXiv AI
Sep 15

STHMoE: Hypergraph-Enhanced Heterogeneous Dependency Coordination for LLM-Based Urban Traffic Data Forecasting

STHMoE is a Spatio‑Temporal Hypergraph‑Enhanced Mixture of Experts framework designed for urban traffic forecasting. It separates traffic dynamics into frequency‑, time‑, spatial‑, and higher‑order representations, each handled by a prompt‑guided expert built on a partially frozen large language model. The higher‑order expert uses an adaptive hypergraph module to learn evolving spatial structures, while an entropy‑aware router balances expert usage and fuses outputs, achieving competitive results on ten real‑world traffic benchmarks.

By Jiawen Chen, Qi Shao, Yongjian Chang, Mingtong Zhou, Duxin Chen, Wenwu Yu
arXiv AI
4d ago

GAS-PFN: Graph-Aware Spatiotemporal Prior-Data Fitted Networks for Zero-Shot Traffic Forecasting

The paper introduces GAS-PFN, a Graph‑Aware Spatiotemporal Prior‑Data Fitted Network that incorporates road topology into synthetic task priors via random‑walk diffusion and uses a two‑mask Transformer architecture to separately model sensor‑specific temporal dynamics and spatiotemporal interactions. This design allows the model to adapt to new traffic networks without parameter updates, achieving a 34.0% average MSE reduction over TimePFN and a 27.7% reduction over recent industry zero‑shot forecasters across six benchmarks at a 90‑minute horizon.

By Joseph Natter, Long Vu, Yifan Zhang, Bobin Deng, Kun Suo, Rui Wu