MST-GNN: graph neural network with multi-granularity in space and time for traffic prediction

Xinru Zhao,Wenhao Yu,Yifan Zhang
DOI: https://doi.org/10.1007/s10707-024-00532-w
IF: 2.7729
2024-11-23
GeoInformatica
Abstract:Traffic flow prediction based on vehicle trajectories is a crucial aspect of Intelligent Transportation Systems (ITS). Deep learning approaches have recently been widely adopted to capture spatio-temporal correlations in traffic conditions, and have achieved superior performance compared to traditional methods. However, most existing studies focus on traffic prediction at a single spatial scale, usually corresponding to the road-segment level. According to the Hierarchy Theory, processes at different scales form a hierarchy of organization, and meaningful patterns may emerge at multiple levels of details. Presetting traffic data at an inappropriate scale can cause misunderstanding in features learning. In this paper, we propose a graph learning model called MST-GNN, which captures the comprehensive behaviors and dynamics of traffic patterns by fusing multi-scale features from both space and time perspectives. In ITS applications, users usually consider traffic conditions at the larger-scale regional level, and a prediction model must attend to multi-scale application requirements. Moreover, the structure of multiple granularities in time series can fully unleash the potential of different temporal scales in learning dynamic traffic pattern features. We inject the multi-scale spatio-temporal structure into a graph neural architecture with a tailored fusion module. Our model achieves state-of-the-art accuracy prediction results on two traffic benchmark datasets.
computer science, information systems,geography, physical
What problem does this paper attempt to address?