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 条
  • [21] STGAT: Spatial-Temporal Graph Attention Networks for Traffic Flow Forecasting
    Kong, Xiangyuan
    Xing, Weiwei
    Wei, Xiang
    Bao, Peng
    Zhang, Jian
    Lu, Wei
    IEEE ACCESS, 2020, 8 : 134363 - 134372
  • [22] Adaptive spatial-temporal graph attention networks for traffic flow forecasting
    Xiangyuan Kong
    Jian Zhang
    Xiang Wei
    Weiwei Xing
    Wei Lu
    Applied Intelligence, 2022, 52 : 4300 - 4316
  • [23] Dynamic Spatial-Temporal Graph Attention Graph Convolutional Network for Short-Term Traffic Flow Forecasting
    Tang, Cong
    Sun, Jingru
    Sun, Yichuang
    2020 IEEE INTERNATIONAL SYMPOSIUM ON CIRCUITS AND SYSTEMS (ISCAS), 2020,
  • [24] Network Traffic Prediction with Attention-based Spatial-Temporal Graph Network
    Peng, Yufei
    Guo, Yingya
    Hao, Run
    Lin, Junda
    2023 IEEE 24TH INTERNATIONAL CONFERENCE ON HIGH PERFORMANCE SWITCHING AND ROUTING, HPSR, 2023,
  • [25] Spatial-Temporal Graph Attention Model on Traffic Forecasting
    Zhang, Xinlan
    Zhang, Zhenguo
    Jin, Xiaofeng
    2020 13TH INTERNATIONAL CONGRESS ON IMAGE AND SIGNAL PROCESSING, BIOMEDICAL ENGINEERING AND INFORMATICS (CISP-BMEI 2020), 2020, : 999 - 1003
  • [26] Graph attention temporal convolutional network for traffic speed forecasting on road networks
    Zhang, Ke
    He, Fang
    Zhang, Zhengchao
    Lin, Xi
    Li, Meng
    TRANSPORTMETRICA B-TRANSPORT DYNAMICS, 2021, 9 (01) : 153 - 171
  • [27] Network traffic prediction with Attention-based Spatial-Temporal Graph Network
    Peng, Yufei
    Guo, Yingya
    Hao, Run
    Xu, Chengzhe
    COMPUTER NETWORKS, 2024, 243
  • [28] Spatial-Temporal Dynamic Graph Convolutional Network With Interactive Learning for Traffic Forecasting
    Liu, Aoyu
    Zhang, Yaying
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2024, 25 (07) : 7645 - 7660
  • [29] Adaptive Spatial-Temporal Fusion Graph Convolutional Networks for Traffic Flow Forecasting
    Li, Senwen
    Ge, Liang
    Lin, Yongquan
    Zeng, Bo
    2022 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2022,
  • [30] DSTAGCN: Dynamic Spatial-Temporal Adjacent Graph Convolutional Network for Traffic Forecasting
    Zheng, Qi
    Zhang, Yaying
    IEEE TRANSACTIONS ON BIG DATA, 2023, 9 (01) : 241 - 253