UM  > Faculty of Science and Technology
Residential Collegefalse
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 Name32nd IEEE International Conference on Tools with Artificial Intelligence (ICTAI)
Source PublicationProceedings - International Conference on Tools with Artificial Intelligence, ICTAI
Volume2020-November
Pages714-721
Conference Date2020/11/09-2020/11/11
Conference PlaceBaltimore, 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

KeywordGraph Convolutional Networks Spatial-temporal Model Traffic Forecasting
DOI10.1109/ICTAI50040.2020.00114
URLView the original
Indexed ByCPCI-S
Language英語English
WOS Research AreaComputer Science
WOS SubjectComputer Science, Artificial Intelligence
WOS IDWOS:000649734800104
Scopus ID2-s2.0-85098800848
Fulltext Access
Citation statistics
Document TypeConference paper
CollectionFaculty of Science and Technology
Corresponding AuthorYe,Kejiang
Affiliation1.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.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Tian,Kelang]'s Articles
[Guo,Jingjie]'s Articles
[Ye,Kejiang]'s Articles
Baidu academic
Similar articles in Baidu academic
[Tian,Kelang]'s Articles
[Guo,Jingjie]'s Articles
[Ye,Kejiang]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Tian,Kelang]'s Articles
[Guo,Jingjie]'s Articles
[Ye,Kejiang]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.