Attention-based spatial-temporal synchronous graph convolution networks for traffic flow forecastingAttention-based spatial-temporal synchronous graph convolution...X. Wei et al.

被引:0
|
作者
Xiaoduo Wei [1 ]
Dawen Xia [2 ]
Yunsong Li [3 ]
Yuce Ao [1 ]
Yan Chen [2 ]
Yang Hu [2 ]
Yantao Li [2 ]
Huaqing Li [2 ]
机构
[1] Kaili University,Engineering Research Center of Micro
[2] Guizhou Minzu University,nano and Intelligent Manufacturing, Ministry of Education
[3] Shizhen College of Guizhou University of Traditional Chinese Medicine,College of Data Science and Information Engineering
[4] Department of Automotive Engineering,College of Information Engineering
[5] Guizhou Traffic Technician and Transportation College,College of Computer Science
[6] Chongqing University,College of Electronic and Information Engineering
[7] Southwest University,undefined
关键词
Traffic forecasting; Graph neural networks; Spatial-temporal modeling; Fourier transform; Attention mechanism;
D O I
10.1007/s10489-025-06341-4
中图分类号
学科分类号
摘要
Accurate traffic flow forecasting is crucial for urban traffic control, planning, and detection. Most existing spatial-temporal modeling methods overlook the hidden dynamic correlations between road network nodes and the time series nonstationarity while synchronously capturing complex long- and short-term spatial-temporal dependencies. To this end, this paper proposes an Attention-based Spatial-Temporal Synchronous Graph Convolutional Network (AST-SGCN) to capture complex spatial-temporal correlations over long and short terms. Specifically, we design a self-attention mechanism that utilizes spatial-temporal synchronous computation to efficiently mine dynamic spatial-temporal correlations with changes in traffic and enhance computational efficiency. Then, we construct a residual adaptive adjacency matrix, which includes historical data and node vectors, to stimulate the information transfer of spatial-temporal graph nodes and mine the hidden spatial-temporal dependencies through the graph convolution layer. Next, we establish a Fourier transform layer (FTL) to handle the nonstationary data. Finally, we develop a spatial-temporal hybrid stacking module for capturing complex long-term spatial-temporal correlations, within which two layers of graph convolution and one layer of self-attention are deployed. Extensive experimental results on three real-world traffic flow datasets demonstrate that our AST-SGCN model outperforms the comparable models.
引用
收藏
相关论文
共 50 条
  • [1] Multi-Attention Based Spatial-Temporal Graph Convolution Networks for Traffic Flow Forecasting
    Hu, Jun
    Chen, Liyin
    2021 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2021,
  • [2] STHSGCN: Spatial-temporal heterogeneous and synchronous graph convolution network for traffic flow prediction
    Yu, Xian
    Bao, Yin-Xin
    Shi, Quan
    HELIYON, 2023, 9 (09)
  • [3] Traffic flow forecasting based on lightweight spatial-temporal graph convolution networks model
    He, Wenwu
    Pei, Boyu
    Mao, Guojun
    Chen, Weiya
    Journal of Railway Science and Engineering, 2022, 19 (09): : 2552 - 2562
  • [4] Attention Mechanism Based Spatial-Temporal Graph Convolution Network for Traffic Prediction
    Xiao, Wenjuan
    Wang, Xiaoming
    Journal of Computers (Taiwan), 2024, 35 (04) : 93 - 108
  • [5] Attention-based spatial-temporal graph transformer for traffic flow forecasting
    Qingyong Zhang
    Wanfeng Chang
    Changwu Li
    Conghui Yin
    Yixin Su
    Peng Xiao
    Neural Computing and Applications, 2023, 35 : 21827 - 21839
  • [6] Attention-based spatial-temporal graph transformer for traffic flow forecasting
    Zhang, Qingyong
    Chang, Wanfeng
    Li, Changwu
    Yin, Conghui
    Su, Yixin
    Xiao, Peng
    NEURAL COMPUTING & APPLICATIONS, 2023, 35 (29): : 21827 - 21839
  • [7] Multiple Information Spatial-Temporal Attention based Graph Convolution Network for traffic prediction
    Tao, Shiming
    Zhang, Huyin
    Yang, Fei
    Wu, Yonghao
    Li, Cong
    APPLIED SOFT COMPUTING, 2023, 136
  • [8] Predicting Traffic Flow Using Dynamic Spatial-Temporal Graph Convolution Networks
    Liu, Yunchang
    Wan, Fei
    Liang, Chengwu
    CMC-COMPUTERS MATERIALS & CONTINUA, 2024, 78 (03): : 4343 - 4361
  • [9] STAGCN: Spatial-Temporal Attention Graph Convolution Network for Traffic Forecasting
    Gu, Yafeng
    Deng, Li
    MATHEMATICS, 2022, 10 (09)
  • [10] Spatial-temporal graph neural network based on gated convolution and topological attention for traffic flow prediction
    Bai, Dewei
    Xia, Dawen
    Huang, Dan
    Hu, Yang
    Li, Yantao
    Li, Huaqing
    APPLIED INTELLIGENCE, 2023, 53 (24) : 30843 - 30864