This study proposes a short-term traffic flow prediction model that combines community detection-based federated learning with a graph convolutional network (GCN) to alleviate the time-consuming training, higher communication costs, and data privacy risks of global GCNs as the amount of data increases. The federated community GCN (FCGCN) can achieve timely, accurate, and safe traffic state predictions in the era of big traffic data, which is critical for the efficient operation of intelligent transportation systems. The FCGCN prediction process has four steps: dividing the local subnetwork with community detection, local training based on the global parameters, uploading the local model parameters, and constructing a global model prediction based on the aggregated parameters. Numerical results on the PeMS04 and PeMS08 datasets show that the FCGCN outperforms four benchmark models, namely, the long short-term memory (LSTM), convolutional neural network (CNN), ChebNet, and graph attention network (GAT) models. The FCGCN prediction is closer to the real value, with nearly the same performance as the global model at a lower time cost, thus achieving accurate and secure short-term traffic flow predictions with three parameters: flow, speed, and occupancy.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    Short-Term Traffic Flow Prediction Based on Graph Convolutional Networks and Federated Learning


    Contributors:
    Xia, Mengran (author) / Jin, Dawei (author) / Chen, Jingyu (author)


    Publication date :

    2023-01-01


    Size :

    7253637 byte




    Type of media :

    Article (Journal)


    Type of material :

    Electronic Resource


    Language :

    English



    Short-Term Traffic Flow Prediction Based on Graph Convolutional Network Embedded LSTM

    Huang, Yanguo / Zhang, Shuo / Wen, Junlin et al. | TIBKAT | 2020


    Short-Term Traffic Flow Prediction Based on Graph Convolutional Network Embedded LSTM

    Huang, Yanguo / Zhang, Shuo / Wen, Junlin et al. | ASCE | 2020


    Short-term traffic flow prediction method based on integrated graph convolutional neural network

    LIU LUYANG / LYU SHUAIQI / BAO XU | European Patent Office | 2024

    Free access

    Short-term traffic flow prediction method and system based on PSO-STGCN graph convolutional network

    WANG YAN / ZHU FENG / AHN YOUNG-TAE et al. | European Patent Office | 2024

    Free access

    Digital twin traffic flow prediction system based on graph federated learning

    XING LING / GAO JIANPING / LI BING et al. | European Patent Office | 2024

    Free access