As a core technology of Intelligent Transportation System (ITS), traffic flow prediction has a wide range of applications. Traffic flow data are spatial-temporal, which are not only correlated to spatial locations in road networks, but also vary with temporal time indices. Existing methods have solved the challenges in traffic flow prediction partly, focusing on modeling spatial-temporal dependencies effectively, while not all intrinsic properties of traffic flow data are utilized fully. Besides, there are very few attempts at incremental learning of spatial-temporal data mining, and few previous works can be easily transferred to the traffic flow prediction task. Motivated by the challenge of incremental learning methods for traffic flow prediction and the underutilization of intrinsic properties of road networks, we propose a Transport-Hub-aware Spatial-Temporal adaptive graph transFormer (H-STFormer) for traffic flow prediction. Specifically, we first design a novel spatial self-attention module to capture the dynamic spatial dependencies. Three graph masking matrices are integrated into spatial self-attentions to highlight both short- and long-term dependences. Additionally, we employ a temporal self-attention module to detect dynamic temporal patterns in the traffic flow data. Finally, we design an extra spatial-temporal knowledge distillation module for incremental learning of traffic flow prediction tasks. Through extensive experiments, we show the effectiveness of H-STFormer in normal and incremental traffic flow prediction tasks. The code is available at https://github.com/Fantasy-Shaw/H-STFormer.


    Access

    Download


    Export, share and cite



    Title :

    Transport-Hub-Aware Spatial-Temporal Adaptive Graph Transformer for Traffic Flow Prediction


    Contributors:
    Xu, Xiao (author) / Zhang, Lei (author) / Liu, Bailong (author) / Liang, Zhizhen (author) / Zhang, Xuefei (author)

    Publication date :

    2023


    Remarks:

    Previous submissions on arxiv are early versions of this manuscript, the latest of which has been rejected by T-ITS. Incremental learning in this work requires more rigorous problem definition and more extensive experiments


    Type of media :

    Preprint


    Type of material :

    Electronic Resource


    Language :

    English





    Spatial-Temporal Graph-Based Transformer Model for Traffic Flow Forecasting

    Wang, Qichao / He, Guojun / Lu, Peiyu et al. | IEEE | 2022


    Spatial-Temporal Position-Aware Graph Convolution Networks for Traffic Flow Forecasting

    Zhao, Yiji / Lin, Youfang / Wen, Haomin et al. | IEEE | 2023


    Counterfactual Graph Transformer for Traffic Flow Prediction

    Yang, Ying / Du, Kai / Dai, Xingyuan et al. | IEEE | 2023


    Hierarchical Traffic Flow Prediction Based on Spatial-Temporal Graph Convolutional Network

    Wang, Hanqiu / Zhang, Rongqing / Cheng, Xiang et al. | IEEE | 2022