Unified Spatial-Temporal Neighbor Attention Network for Dynamic Traffic Prediction

Wangchen Long, Zhu Xiao, Dong Wang, Hongbo Jiang, Jie Chen, You Li, Mamoun Alazab

Research output: Contribution to journalArticlepeer-review

54 Citations (Scopus)

Abstract

Traffic prediction plays an essential role in many real-world applications ranging from route planning to vehicular communications. The goal of making accurate prediction is challenging due to the dynamic and various spatial-temporal correlations in traffic network. Numerous methods have been proposed to capture the spatial-temporal correlations, however, there are two major challenges: i) The task of synchronously capturing dynamic and various spatial-temporal correlations has not been fully explored yet. ii) Lack of an adaptive mechanism to model different influences caused by different external factors. To overcome these challenges, we propose a unified spatial-temporal attention network entitled USTAN. First, it constructs the spatial neighbor graph and the temporal neighbor array to describe the spatial and temporal correlations respectively. By expanding the spatial neighbors in temporal perspective according to the temporal neighbor array, all the spatial-temporal neighbors can be gathered together. Second, a single unified attention component is performed on all the spatial-temporal neighbors for capturing the spatial-temporal correlations all at once. In addition, a gated fusion module is designed to fuse the external factors adaptively to reduce predicting error. Extensive experiments on real-world traffic prediction tasks demonstrate the superiority of our proposed framework.

Original languageEnglish
Pages (from-to)1515 - 1529
Number of pages15
JournalIEEE Transactions on Vehicular Technology
Volume72
Issue number2
Early online date2022
DOIs
Publication statusPublished - 1 Feb 2023

Fingerprint

Dive into the research topics of 'Unified Spatial-Temporal Neighbor Attention Network for Dynamic Traffic Prediction'. Together they form a unique fingerprint.

Cite this