English

STMGF: An Effective Spatial-Temporal Multi-Granularity Framework for Traffic Forecasting

Machine Learning 2024-04-10 v1 Artificial Intelligence

Abstract

Accurate Traffic Prediction is a challenging task in intelligent transportation due to the spatial-temporal aspects of road networks. The traffic of a road network can be affected by long-distance or long-term dependencies where existing methods fall short in modeling them. In this paper, we introduce a novel framework known as Spatial-Temporal Multi-Granularity Framework (STMGF) to enhance the capture of long-distance and long-term information of the road networks. STMGF makes full use of different granularity information of road networks and models the long-distance and long-term information by gathering information in a hierarchical interactive way. Further, it leverages the inherent periodicity in traffic sequences to refine prediction results by matching with recent traffic data. We conduct experiments on two real-world datasets, and the results demonstrate that STMGF outperforms all baseline models and achieves state-of-the-art performance.

Keywords

Cite

@article{arxiv.2404.05774,
  title  = {STMGF: An Effective Spatial-Temporal Multi-Granularity Framework for Traffic Forecasting},
  author = {Zhengyang Zhao and Haitao Yuan and Nan Jiang and Minxiao Chen and Ning Liu and Zengxiang Li},
  journal= {arXiv preprint arXiv:2404.05774},
  year   = {2024}
}
R2 v1 2026-06-28T15:47:56.870Z