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] Spatial-Temporal Bipartite Graph Attention Network for Traffic Forecasting
    Lakma, Dimuthu
    Perera, Kushani
    Borovica-Gajic, Renata
    Karunasekera, Shanika
    ADVANCES IN KNOWLEDGE DISCOVERY AND DATA MINING, PT II, PAKDD 2024, 2024, 14646 : 68 - 80
  • [42] Multi-Hierarchical Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting
    Li, Zilong
    Ren, Qianqian
    Chen, Long
    Sui, Xiaohong
    Li, Jinbao
    2022 26TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2022, : 4913 - 4919
  • [43] Spatial-Temporal Similarity Fusion Graph Adversarial Convolutional Networks for traffic flow forecasting
    Wang, Bin
    Long, Zhendan
    Sheng, Jinfang
    Zhong, Qiang
    JOURNAL OF THE FRANKLIN INSTITUTE-ENGINEERING AND APPLIED MATHEMATICS, 2024, 361 (17):
  • [44] Efficient Mobile Cellular Traffic Forecasting using Spatial-Temporal Graph Attention Networks
    Mortazavi, SeyedMohammad
    Sousa, Elvino
    2023 IEEE 34TH ANNUAL INTERNATIONAL SYMPOSIUM ON PERSONAL, INDOOR AND MOBILE RADIO COMMUNICATIONS, PIMRC, 2023,
  • [45] 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
  • [46] Adaboosting graph attention recurrent network: A deep learning framework for traffic speed forecasting in dynamic transportation networks with spatial-temporal dependencies
    Zhang, Yunuo
    Wang, Xiaoling
    Yu, Jia
    Zeng, Tuocheng
    Wang, Jiajun
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 127
  • [47] A Novel Attention-Based Dynamic Multi-Graph Spatial-Temporal Graph Neural Network Model for Traffic Prediction
    Diao, Chunyan
    Zhang, Dafang
    Liang, Wei
    Jiang, Man
    Li, Kuanching
    IEEE TRANSACTIONS ON EMERGING TOPICS IN COMPUTATIONAL INTELLIGENCE, 2024,
  • [48] ST_AGCNT: Traffic Speed Forecasting Based on Spatial-Temporal Adaptive Graph Convolutional Network with Transformer
    Cheng, Rongjun
    Liu, Mengxia
    Xu, Yuanzi
    SUSTAINABILITY, 2025, 17 (05)
  • [49] 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)
  • [50] AdpSTGCN: Adaptive spatial-temporal graph convolutional network for traffic forecasting
    Zhang, Xudong
    Chen, Xuewen
    Tang, Haina
    Wu, Yulei
    Shen, Hanji
    Li, Jun
    KNOWLEDGE-BASED SYSTEMS, 2024, 301