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

被引:49
作者
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
相关论文
共 39 条
[1]  
[Anonymous], 1776, TRANSPORT RES REC
[2]  
[Anonymous], 2015, Neural Networks and Deep Learning
[3]  
Bahdanau D, 2016, Arxiv, DOI arXiv:1409.0473
[4]  
Bandanau D, 2016, INT CONF ACOUST SPEE, P4945, DOI 10.1109/ICASSP.2016.7472618
[5]   Feed-forward neural networks [J].
Bebis, George ;
Georgiopoulos, Michael .
IEEE Potentials, 1994, 13 (04) :27-31
[6]   Complementary and alternative therapies in childhood attention and hyperactivity problems [J].
Chan, E ;
Rappaport, LA ;
Kemper, KJ .
JOURNAL OF DEVELOPMENTAL AND BEHAVIORAL PEDIATRICS, 2003, 24 (01) :4-8
[7]   Learning Graph Convolutional Networks for Multi-Label Recognition and Applications [J].
Chen, Zhao-Min ;
Wei, Xiu-Shen ;
Wang, Peng ;
Guo, Yanwen .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (06) :6969-6983
[8]   Short-Term Traffic Forecasting by Mining the Non-Stationarity of Spatiotemporal Patterns [J].
Cheng, Shifen ;
Lu, Feng ;
Peng, Peng .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2021, 22 (10) :6365-6383
[9]  
Cheng WY, 2018, AAAI CONF ARTIF INTE, P2151
[10]   NONPARAMETRIC REGRESSION AND SHORT-TERM FREEWAY TRAFFIC FORECASTING [J].
DAVIS, GA ;
NIHAN, NL .
JOURNAL OF TRANSPORTATION ENGINEERING-ASCE, 1991, 117 (02) :178-188