Research breakthrough possible @S-Logix pro@slogix.in

Office Address

  • 2nd Floor, #7a, High School Road, Secretariat Colony Ambattur, Chennai-600053 (Landmark: SRM School) Tamil Nadu, India
  • pro@slogix.in
  • +91- 81240 01111

Social List

Learning dynamics and heterogeneity of spatial-temporal graph data for traffic forecasting - 2021

Learning Dynamics And Heterogeneity Of Spatial-Temporal Graph Data For Traffic Forecasting

Research Area:  Machine Learning

Abstract:

Accurate traffic forecasting is critical in improving safety, stability, and efficiency of intelligent transportation systems. Despite years of studies, accurate traffic prediction still faces the following challenges, including modeling the dynamics of trafc data along both temporal and spatial dimensions, and capturing the periodicity and the spatial heterogeneity of trafc data, and the problem is more difficult for long-term forecast. In this paper, we propose an Attention based Spatial-Temporal Graph Neural Network (ASTGNN) for traffic forecasting. Specifically, in the temporal dimension, we design a novel self-attention mechanism that is capable of utilizing the local context, which is specialized for numerical sequence representation transformation. It enables our prediction model to capture the temporal dynamics of traffic data and to enjoy global receptive elds that is beneficial for long-term forecast. In the spatial dimension, we develop a dynamic graph convolution module, employing self-attention to capture the spatial correlations in a dynamic manner. Furthermore, we explicitly model the periodicity and capture the spatial heterogeneity through embedding modules. Experiments on five real-world traffic flow datasets demonstrate that ASTGNN outperforms the state-of-the-art baselines.

Keywords:  

Author(s) Name:  Shengnan Guo; Youfang Lin; Huaiyu Wan; Xiucheng Li; Gao Cong

Journal name:  IEEE Transactions on Knowledge and Data Engineering ( Early Access )

Conferrence name:  

Publisher name:  IEEE

DOI:  10.1109/TKDE.2021.3056502

Volume Information:  Page(s): 1 - 1