ST-GAT: A Spatio-Temporal Graph Attention Network for Accurate Traffic Speed Prediction
- Authors
- Song, Junho; Son, Jiwon; Seo, Dong-hyuk; Han, Kyungsik; Kim, Namhyuk; Kim, Sang-Wook
- Issue Date
- Oct-2022
- Publisher
- ASSOC COMPUTING MACHINERY
- Keywords
- Traffic speed prediction; Attention network; Spatio-temporal data
- Citation
- PROCEEDINGS OF THE 31ST ACM INTERNATIONAL CONFERENCE ON INFORMATION AND KNOWLEDGE MANAGEMENT, CIKM 2022, pp.4500 - 4504
- Indexed
- SCOPUS
- Journal Title
- PROCEEDINGS OF THE 31ST ACM INTERNATIONAL CONFERENCE ON INFORMATION AND KNOWLEDGE MANAGEMENT, CIKM 2022
- Start Page
- 4500
- End Page
- 4504
- URI
- https://scholarworks.bwise.kr/hanyang/handle/2021.sw.hanyang/192912
- DOI
- 10.1145/3511808.3557705
- Abstract
- Spatio-temporal models, which combine GNNs (Graph Neural Networks) and RNNs (Recurrent Neural Networks), have shown state-of-the-art accuracy in traffic speed prediction. However, we find that they consider the spatial and temporal dependencies between speeds separately in the two (i.e., space and time) dimensions, thereby unable to exploit the joint-dependencies of speeds in space and time. In this paper, with the evidence via preliminary analysis, we point out the importance of considering individual dependencies between two speeds from all possible points in space and time for accurate traffic speed prediction. Then, we propose an Individual Spatio-Temporal graph (IST-graph) that represents the Individual Spatio-Temporal dependencies (IST-dependencies) very effectively and a Spatio-Temporal Graph ATtention network (ST-GAT), a novel model to predict the future traffic speeds based on the IST-graph and the attention mechanism. The results from our extensive evaluation with five real-world datasets demonstrate (1) the effectiveness of the IST-graph in modeling traffic speed data, (2) the superiority of ST-GAT over 5 state-of-the-art models (i.e., 2-33% gains) in prediction accuracy, and (3) the robustness of our ST-GAT even in abnormal traffic situations.
- Files in This Item
-
Go to Link
- Appears in
Collections - 서울 공과대학 > 서울 컴퓨터소프트웨어학부 > 1. Journal Articles
Items in ScholarWorks are protected by copyright, with all rights reserved, unless otherwise indicated.