Contrastive learning is an increasingly important research direction and has attracted considerable attention in the field of computer vision. It can greatly improve the representativeness of image features through data augmentation, unsupervised learning, and pre‐trained models. However, in the field of traffic flow forecasting, most graph‐based models focus on the construct of spatial–temporal relationships between road segments and ignore the use of temporal data augmentation and pre‐trained models, which can improve the representation ability of the forecasting model. Therefore, in this work, contrastive learning are used to expand the distribution of sequence samples and improve the quality and generalization of forecasting models. Based on this, a novel forecasting model called contrastive learning based on multi graph convolution network (CLMGCN) is proposed, which is combined with four components: multi graph convolution network, which learns the spatial–temporal feature of the input traffic data; temporal data augmentation, which obtains the augmentation data of the input traffic data; contrastive learning, which achieves the pre‐training phase and improve the quality of output feature of multi graph convolution network; output block, which utilizes the enhanced output feature of multi graph convolution network for predicting the future traffic data. Finally, by the experimental results of four public traffic flow datasets, it can be shown that CLMGCN achieves higher traffic forecasting accuracy with lower model complexity.


    Zugriff

    Download


    Exportieren, teilen und zitieren



    Titel :

    Contrastive learning for traffic flow forecasting based on multi graph convolution network


    Beteiligte:
    Guo, Kan (Autor:in) / Tian, Daxin (Autor:in) / Hu, Yongli (Autor:in) / Sun, Yanfeng (Autor:in) / Qian, Zhen (Sean) (Autor:in) / Zhou, Jianshan (Autor:in) / Gao, Junbin (Autor:in) / Yin, Baocai (Autor:in)

    Erschienen in:

    Erscheinungsdatum :

    2024-02-01


    Format / Umfang :

    12 pages




    Medientyp :

    Aufsatz (Zeitschrift)


    Format :

    Elektronische Ressource


    Sprache :

    Englisch




    Contrastive learning for traffic flow forecasting based on multi graph convolution network

    Kan Guo / Daxin Tian / Yongli Hu et al. | DOAJ | 2024

    Freier Zugriff

    TRAFFIC FLOW FORECASTING METHOD BASED ON MULTI-MODE DYNAMIC RESIDUAL GRAPH CONVOLUTION NETWORK

    HUANG XIAOHUI / YE YUMING / LING JIAHAO et al. | Europäisches Patentamt | 2023

    Freier Zugriff

    Spatio-Temporal Graph Attention Convolution Network for Traffic Flow Forecasting

    Liu, Kun / Zhu, Yifan / Wang, Xiao et al. | Transportation Research Record | 2024


    Graph transformer based dynamic multiple graph convolution networks for traffic flow forecasting

    Hu, Yongli / Peng, Ting / Guo, Kan et al. | Wiley | 2023

    Freier Zugriff

    Graph transformer based dynamic multiple graph convolution networks for traffic flow forecasting

    Yongli Hu / Ting Peng / Kan Guo et al. | DOAJ | 2023

    Freier Zugriff