A decentralized federated learning-based spatial-temporal model for freight traffic speed forecasting

被引:4
|
作者
Shen, Xiuyu [1 ]
Chen, Jingxu [1 ]
Zhu, Siying [2 ]
Yan, Ran [3 ]
机构
[1] Southeast Univ, Sch Transportat, Nanjing 210096, Peoples R China
[2] Singapore Univ Social Sci, Sch Business, Singapore, Singapore
[3] Nanyang Technol Univ, Sch Civil & Environm Engn, Singapore, Singapore
基金
中国国家自然科学基金;
关键词
Freight traffic speed forecasting; Decentralized federated learning; Transformer network; Metropolitan area; Traffic management; EXTENDED KALMAN FILTER; FLOW PREDICTION;
D O I
10.1016/j.eswa.2023.122302
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Accurately understanding the spatial-temporal information of future freight traffic speed in the metropolitan area is of vital importance to formulate freight-related traffic management strategies. In this study, we develop a novel decentralized federated learning-based spatial-temporal model for freight traffic speed forecasting while implementing collaborative training among multiple participants instead of requiring an exclusive cloud center server for centralized data processing. First, a tailored spatial-temporal transformer network is proposed to substitute the existing graph convolutional network for local personalized learning of each participant. Second, a decentralized federated learning model is designed to fuse local personalization models for freight traffic speed forecasting. The associated convergence properties are theoretically illustrated. rgb]0,0,0Finally, the experiments based on a real-world freight dataset of member cities in the Nanjing Metropolitan Area demonstrate that the proposed approach can accurately forecast freight traffic speed and outperform existing traffic speed forecasting methods, with the average improvement of 8.3% for MAE, 8.2% for RMSE, and 8.6% for MAPE respectively. The visualization results reveal that the proposed approach is able to effectively capture the internal spatial-temporal dependencies among urban regions from various neighboring cities, providing the insights for collaboratively developing proactive freight-related traffic management strategies.
引用
下载
收藏
页数:17
相关论文
共 50 条
  • [1] Graph learning-based spatial-temporal graph convolutional neural networks for traffic forecasting
    Hu, Na
    Zhang, Dafang
    Xie, Kun
    Liang, Wei
    Hsieh, Meng-Yen
    CONNECTION SCIENCE, 2022, 34 (01) : 429 - 448
  • [2] Learning to effectively model spatial-temporal heterogeneity for traffic flow forecasting
    Minrui Xu
    Xiyang Li
    Fucheng Wang
    Jedi S. Shang
    Tai Chong
    Wanjun Cheng
    Jiajie Xu
    World Wide Web, 2023, 26 : 849 - 865
  • [3] Learning to effectively model spatial-temporal heterogeneity for traffic flow forecasting
    Xu, Minrui
    Li, Xiyang
    Wang, Fucheng
    Shang, Jedi S.
    Chong, Tai
    Cheng, Wanjun
    Xu, Jiajie
    WORLD WIDE WEB-INTERNET AND WEB INFORMATION SYSTEMS, 2023, 26 (03): : 849 - 865
  • [4] Learning Dynamic Spatial-Temporal Dependence in Traffic Forecasting
    Ren, Chaoyu
    Li, Yuezhu
    IEEE Access, 2024, 12 : 190039 - 190053
  • [5] Spatial-Temporal Graph Attention Model on Traffic Forecasting
    Zhang, Xinlan
    Zhang, Zhenguo
    Jin, Xiaofeng
    2020 13TH INTERNATIONAL CONGRESS ON IMAGE AND SIGNAL PROCESSING, BIOMEDICAL ENGINEERING AND INFORMATICS (CISP-BMEI 2020), 2020, : 999 - 1003
  • [6] DeepTransport: Learning Spatial-Temporal Dependency for Traffic Condition Forecasting
    Cheng, Xingyi
    Zhang, Ruiqing
    Zhou, Jie
    Xu, Wei
    arXiv, 2017,
  • [7] DeepTransport: Learning Spatial-Temporal Dependency for Traffic Condition Forecasting
    Cheng, Xingyi
    Zhang, Ruiqing
    Zhou, Jie
    Xu, Wei
    2018 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2018,
  • [8] Spatial-Temporal Graph-Based Transformer Model for Traffic Flow Forecasting
    Wang, Qichao
    He, Guojun
    Lu, Peiyu
    Chen, Qiyang
    Chen, Yanrong
    Huang, Wei
    2022 IEEE 25TH INTERNATIONAL CONFERENCE ON INTELLIGENT TRANSPORTATION SYSTEMS (ITSC), 2022, : 2806 - 2811
  • [9] Attention-based dynamic spatial-temporal graph convolutional networks for traffic speed forecasting
    Zhao, Jianli
    Liu, Zhongbo
    Sun, Qiuxia
    Li, Qing
    Jia, Xiuyan
    Zhang, Rumeng
    EXPERT SYSTEMS WITH APPLICATIONS, 2022, 204
  • [10] Learning Dynamics and Heterogeneity of Spatial-Temporal Graph Data for Traffic Forecasting
    Guo, Shengnan
    Lin, Youfang
    Wan, Huaiyu
    Li, Xiucheng
    Cong, Gao
    IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2022, 34 (11) : 5415 - 5428