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