Back to Search Start Over

MVSTT: A Multiview Spatial-Temporal Transformer Network for Traffic-Flow Forecasting

Authors :
Pu, Bin
Liu, Jiansong
Kang, Yan
Chen, Jianguo
Yu, Philip S.
Source :
IEEE Transactions on Cybernetics; 2024, Vol. 54 Issue: 3 p1582-1595, 14p
Publication Year :
2024

Abstract

Accurate traffic-flow prediction remains a critical challenge due to complicated spatial dependencies, temporal factors, and unpredictable events. Most existing approaches focus on single- or dual-view learning and thus face limitations in systematically learning complex spatial-temporal features. In this work, we propose a novel multiview spatial-temporal transformer (MVSTT) network that can effectively learn complex spatial-temporal domain correlations and potential patterns from multiple views. First, we examine a temporal view and design a short-range gated convolution component from a short-term subview, and a long-range gated convolution component from a long-term subview. These two components effectively aggregate knowledge of the temporal domain at multiple granularities and mine patterns of node evolution across time steps. Meanwhile, in the spatial view, we design a dual-graph spatial learning module that captures fixed and dynamic spatial dependencies of nodes, as well as the evolution patterns of edges, from the static and dynamic graph subviews, respectively. In addition, we further design a spatial-temporal transformer to mine different levels of spatial-temporal features through multiview knowledge fusion. Extensive experiments on four real-world traffic datasets show that our method consistently outperforms the state-of-the-art baseline. The code of MVSTT is available at <uri>https://github.com/JianSoL/MVSTT</uri>.

Details

Language :
English
ISSN :
21682267
Volume :
54
Issue :
3
Database :
Supplemental Index
Journal :
IEEE Transactions on Cybernetics
Publication Type :
Periodical
Accession number :
ejs65491298
Full Text :
https://doi.org/10.1109/TCYB.2022.3223918