Attention-based dynamic spatial-temporal graph convolutional networks for traffic speed forecasting

被引:36
|
作者
Zhao, Jianli [1 ]
Liu, Zhongbo [1 ]
Sun, Qiuxia [2 ]
Li, Qing [2 ]
Jia, Xiuyan [2 ]
Zhang, Rumeng [1 ]
机构
[1] Shandong Univ Sci & Technol, Coll Comp Sci & Engn, Qingdao, Peoples R China
[2] Shandong Univ Sci & Technol, Coll Math & Syst Sci, Qingdao, Peoples R China
关键词
Traffic speed forecast; GCN; Dynamic spatial-temporal correlations; Attention mechanism; TIME; FLOW; REGRESSION; MODEL;
D O I
10.1016/j.eswa.2022.117511
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In recent years, spatial-temporal graph modeling based on graph convolutional neural networks (GCN) has become an effective method for mining spatial-temporal dependencies in traffic forecasting research. However, existing studies lack the capability of dynamic spatial-temporal modeling of traffic speeds. Furthermore, longterm forecasting is difficult because of the diversity of traffic conditions. In addition, traditional studies capture only the features of fixed graph structures, which do not reflect real spatial dependence. To address these challenges, this study proposes a novel attention-based dynamic spatial-temporal graph convolutional network (ADSTGCN) model. ADSTGCN mainly consists of multiple dynamic spatial-temporal blocks, each of which contains three modules: 1) a dynamic adjustment module to model the dynamic spatial-temporal correlations of traffic speed, 2) a gated dilated convolution module to mine long-term dependencies, and 3) a spatial convolution module to capture hidden spatial dependencies. Experiments on three public traffic datasets demonstrated the good performance of the model.
引用
收藏
页数:10
相关论文
共 50 条
  • [41] STGAFormer: Spatial-temporal Gated Attention Transformer based Graph Neural Network for traffic flow forecasting
    Geng, Zili
    Xu, Jie
    Wu, Rongsen
    Zhao, Changming
    Wang, Jin
    Li, Yunji
    Zhang, Chenlin
    INFORMATION FUSION, 2024, 105
  • [42] Spatial-temporal graph convolutional networks for traffic flow prediction considering multiple traffic parameters
    Ziyi Su
    Tong Liu
    Xiatong Hao
    Xiaojian Hu
    The Journal of Supercomputing, 2023, 79 : 18293 - 18312
  • [43] Learning Dynamics and Heterogeneity of Spatial-Temporal Graph Data for Traffic Forecasting
    Guo, Shengnan
    Lin, Youfang
    Wan, Huaiyu
    Li, Xiucheng
    Cong, Gao
    IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2022, 34 (11) : 5415 - 5428
  • [44] A graph-attention based spatial-temporal learning framework for tourism demand forecasting
    Zhou, Binggui
    Dong, Yunxuan
    Yang, Guanghua
    Hou, Fen
    Hu, Zheng
    Xu, Suxiu
    Ma, Shaodan
    KNOWLEDGE-BASED SYSTEMS, 2023, 263
  • [45] Attention-based spatial–temporal multi-graph convolutional networks for casualty prediction of terrorist attacks
    Zhiwen Hou
    Yuchen Zhou
    Xiaowei Wu
    Fanliang Bu
    Complex & Intelligent Systems, 2023, 9 : 6307 - 6328
  • [46] Spatial-temporal upsampling graph convolutional network for daily long-term traffic speed prediction
    Zhang, Song
    Liu, Yanbing
    Xiao, Yunpeng
    He, Rui
    JOURNAL OF KING SAUD UNIVERSITY-COMPUTER AND INFORMATION SCIENCES, 2022, 34 (10) : 8996 - 9010
  • [47] Attention-Based Gated Recurrent Graph Convolutional Network for Short-Term Traffic Flow Forecasting
    Lou, Ping
    Wu, Zihao
    Hu, Jiwei
    Liu, Quan
    Wei, Qin
    JOURNAL OF MATHEMATICS, 2023, 2023
  • [48] Short-term urban traffic forecasting in smart cities: a dynamic diffusion spatial-temporal graph convolutional network
    Yin, Xiang
    Yu, Junyang
    Duan, Xiaoyu
    Chen, Lei
    Liang, Xiaoli
    COMPLEX & INTELLIGENT SYSTEMS, 2025, 11 (02)
  • [49] Link traffic speed forecasting using convolutional attention-based gated recurrent unit
    Khodabandelou, Ghazaleh
    Kheriji, Walid
    Selem, Fouad Hadj
    APPLIED INTELLIGENCE, 2021, 51 (04) : 2331 - 2352
  • [50] Adaptive Multi-receptive Field Spatial-Temporal Graph Convolutional Network for Traffic Forecasting
    Wang, Xing
    Zhao, Juan
    Zhu, Lin
    Zhou, Xu
    Li, Zhao
    Feng, Junlan
    Deng, Chao
    Zhang, Yong
    2021 IEEE GLOBAL COMMUNICATIONS CONFERENCE (GLOBECOM), 2021,