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

Xiuyu Shen, Jingxu Chen*, Siying Zhu, Ran Yan

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

11 Citations (Scopus)

Abstract

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.

Original languageEnglish
Article number122302
JournalExpert Systems with Applications
Volume238
DOIs
Publication statusPublished - Mar 15 2024
Externally publishedYes

Bibliographical note

Publisher Copyright:
© 2023 Elsevier Ltd

ASJC Scopus Subject Areas

  • General Engineering
  • Computer Science Applications
  • Artificial Intelligence

Keywords

  • Decentralized federated learning
  • Freight traffic speed forecasting
  • Metropolitan area
  • Traffic management
  • Transformer network

Fingerprint

Dive into the research topics of 'A decentralized federated learning-based spatial–temporal model for freight traffic speed forecasting'. Together they form a unique fingerprint.

Cite this