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] Spatial-Temporal Attention Mechanism and Graph Convolutional Networks for Destination Prediction
    Li, Cong
    Zhang, Huyin
    Wang, Zengkai
    Wu, Yonghao
    Yang, Fei
    FRONTIERS IN NEUROROBOTICS, 2022, 16
  • [22] Dynamic multi-granularity spatial-temporal graph attention network for traffic forecasting
    Sang, Wei
    Zhang, Huiliang
    Kang, Xianchang
    Nie, Ping
    Meng, Xin
    Boulet, Benoit
    Sun, Pei
    INFORMATION SCIENCES, 2024, 662
  • [23] Attention-Based Spatial-Temporal Convolution Gated Recurrent Unit for Traffic Flow Forecasting
    Zhang, Qingyong
    Chang, Wanfeng
    Yin, Conghui
    Xiao, Peng
    Li, Kelei
    Tan, Meifang
    ENTROPY, 2023, 25 (06)
  • [24] Generalized spatial-temporal regression graph convolutional transformer for traffic forecasting
    Xiong, Lang
    Su, Liyun
    Zeng, Shiyi
    Li, Xiangjing
    Wang, Tong
    Zhao, Feng
    COMPLEX & INTELLIGENT SYSTEMS, 2024, 10 (06) : 7943 - 7964
  • [25] Attention based spatial-temporal graph convolutional networks for boiler NOx prediction
    Zhou, Yongqing
    Hao, Dawei
    Fan, Yuchen
    Wen, Xintong
    Wei, Chang
    Liu, Xin
    Zhang, Wenzhen
    Wang, Heyang
    Meitan Xuebao/Journal of the China Coal Society, 2024, 49 (10): : 4127 - 4137
  • [26] Spatial-temporal hypergraph convolutional network for traffic forecasting
    Zhao, Zhenzhen
    Shen, Guojiang
    Zhou, Junjie
    Jin, Junchen
    Kong, Xiangjie
    PEERJ COMPUTER SCIENCE, 2023, 9
  • [27] Multi-Dimensional Attention Based Spatial-Temporal Networks for Traffic Forecasting
    Xu, Guangxia
    Hu, Xinting
    WIRELESS COMMUNICATIONS & MOBILE COMPUTING, 2022, 2022
  • [28] Attention-based global and local spatial-temporal graph convolutional network for vehicle emission prediction
    Fei, Xihong
    Ling, Qiang
    NEUROCOMPUTING, 2023, 521 : 41 - 55
  • [29] STAD-GCN: Spatial-Temporal Attention-based Dynamic Graph Convolutional Network for retail market price prediction
    Kim, Sodam
    Park, Eunil
    EXPERT SYSTEMS WITH APPLICATIONS, 2024, 255
  • [30] Spatial-temporal graph convolutional networks for traffic flow prediction considering multiple traffic parameters
    Su, Ziyi
    Liu, Tong
    Hao, Xiatong
    Hu, Xiaojian
    JOURNAL OF SUPERCOMPUTING, 2023, 79 (16) : 18293 - 18312