GraphSAGE-Based Dynamic Spatial–Temporal Graph Convolutional Network for Traffic Prediction

Traffic networks exhibit complex spatial-temporal dependencies, and accurately capturing such dependencies is critical to improving prediction accuracy. Recently, many deep learning models have been proposed for spatial-temporal dependency modeling. While numerous deep learning models have been developed for spatial-temporal dependency modeling, most rely on different types of convolutions to extract spatial and temporal correlations separately. To address this limitation, the authors propose a novel deep learning framework for traffic prediction called GraphSAGE-based Dynamic Spatial-Temporal Graph Convolutional Network (DST-GraphSAGE), which can capture dynamic spatial and temporal dependencies simultaneously. Their model utilizes a spatial-temporal GraphSAGE module to extract localized spatial-temporal correlations from past observations of a node’s spatial neighbors. Meanwhile, the attention mechanism is incorporated to dynamically learn weights between traffic nodes based on graph features. Additionally, to capture long-term trends in traffic data, they employ dilated causal convolution as the temporal convolution layer. A series of numerical experiments are conducted on five real-world datasets, which demonstrates the effectiveness of their model for spatial-temporal dependency modeling.

Language

  • English

Media Info

Subject/Index Terms

Filing Info

  • Accession Number: 01908047
  • Record Type: Publication
  • Files: TRIS
  • Created Date: Feb 14 2024 9:15AM