Residential College | false |
Status | 已發表Published |
ST-MGAT: Spatial-Temporal Multi-Head Graph Attention Networks for Traffic Forecasting | |
Tian,Kelang1,2; Guo,Jingjie2; Ye,Kejiang2; Xu,Cheng Zhong3 | |
2020-11 | |
Conference Name | 32nd IEEE International Conference on Tools with Artificial Intelligence (ICTAI) |
Source Publication | Proceedings - International Conference on Tools with Artificial Intelligence, ICTAI |
Volume | 2020-November |
Pages | 714-721 |
Conference Date | 2020/11/09-2020/11/11 |
Conference Place | Baltimore, MD, USA |
Abstract | Graph Neural Networks (GNNs) have attracted increasing attention due to the significant representation learning capacity for graphs. The traffic forecasting is a typical graph representation learning task, but it is challenging to model the complex spatial and temporal relationships in traffics. Traditional spectral approaches get filters based on the eigendecomposition, which depends on the Laplacian matrix of the graph. However, these approaches have expensive matrix operation on graph convolutions neural networks and are insufficient to tackle the spatial dependency. In this paper, we propose a novel graph neural network-Spatial-Temporal Multi-head Graph ATtention network (ST-MGAT), to deal with the traffic forecasting problem. We build convolutions on the graph directly. We consider the features of neighborhood nodes and the weights of the edges to generate new node representation. More specifically, there are two main modules: i) Temporal convolution blocks to capture the dynamic time correlations; ii) Graph attention networks to capture the dynamic spatial relations between nodes. Experimental results show that our model achieves up to 13% improvement over the state-of-The-Art approaches in short-Term, medium-Term, and long-Term highway traffic forecasting. 1Code is available at https://github.com/Kelang-Tian/ST-Mgat |
Keyword | Graph Convolutional Networks Spatial-temporal Model Traffic Forecasting |
DOI | 10.1109/ICTAI50040.2020.00114 |
URL | View the original |
Indexed By | CPCI-S |
Language | 英語English |
WOS Research Area | Computer Science |
WOS Subject | Computer Science, Artificial Intelligence |
WOS ID | WOS:000649734800104 |
Scopus ID | 2-s2.0-85098800848 |
Fulltext Access | |
Citation statistics | |
Document Type | Conference paper |
Collection | Faculty of Science and Technology |
Corresponding Author | Ye,Kejiang |
Affiliation | 1.University of Science and Technology of China,China 2.Shenzhen Institutes of Advanced Technology,Chinese Academy of Sciences,China 3.State Key Lab of IoTSC,University of Macau,Faculty of Science and Technology,Macao |
Recommended Citation GB/T 7714 | Tian,Kelang,Guo,Jingjie,Ye,Kejiang,et al. ST-MGAT: Spatial-Temporal Multi-Head Graph Attention Networks for Traffic Forecasting[C], 2020, 714-721. |
APA | Tian,Kelang., Guo,Jingjie., Ye,Kejiang., & Xu,Cheng Zhong (2020). ST-MGAT: Spatial-Temporal Multi-Head Graph Attention Networks for Traffic Forecasting. Proceedings - International Conference on Tools with Artificial Intelligence, ICTAI, 2020-November, 714-721. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment