kth.sePublications KTH
Change search
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf
Semi-decentralized Training of Spatio-Temporal Graph Neural Networks for Traffic Prediction
University of Zagreb, Faculty of Electrical Engineering and Computing, Croatia.
RISE Research Institutes of Sweden, Sweden.
University of Zagreb, Faculty of Electrical Engineering and Computing, Croatia.
University of Zagreb, Faculty of Electrical Engineering and Computing, Croatia.
Show others and affiliations
2025 (English)In: Proceedings - 2025 IEEE International Conference on Edge Computing and Communications, EDGE 2025, Institute of Electrical and Electronics Engineers (IEEE) , 2025, p. 147-155Conference paper, Published paper (Refereed)
Abstract [en]

In smart mobility, large networks of geographically distributed sensors produce vast amounts of high-frequency spatio-temporal data that must be processed in real time to avoid major disruptions. Traditional centralized approaches are increasingly unsuitable to this task, as they struggle to scale with expanding sensor networks, and reliability issues in central components can easily affect the whole deployment. To address these challenges, we explore and adapt semi-decentralized training techniques for Spatio-Temporal Graph Neural Networks (ST-GNNs) in the smart mobility domain. We implement a simulation framework where sensors are grouped by proximity into multiple cloudlets, each handling a subgraph of the traffic graph, fetching node features from other cloudlets to train its own local ST-GNN model, and exchanging model updates with other cloudlets to ensure consistency, enhancing scalability and removing reliance on a centralized aggregator. We perform extensive comparative evaluation of four different ST-GNN training setups - centralized, traditional FL, server-free FL, and Gossip Learning - on large-scale traffic datasets, the METR-LA and PeMS-BAY datasets, for short-, mid-, and long-term vehicle speed predictions. Experimental results show that semi-decentralized setups are comparable to centralized approaches in performance metrics, while offering advantages in terms of scalability and fault tolerance. In addition, we highlight often overlooked issues in existing literature for distributed ST-GNNs, such as the variation in model performance across different geographical areas due to region-specific traffic patterns, and the significant communication overhead and computational costs. However, due to the planar nature of graphs, per-cloudlet costs remain consistent as the network grows, unlike the growing costs in a centralized approach.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE) , 2025. p. 147-155
Keywords [en]
semi-decentralized training, ST-GNN, traffic prediction
National Category
Computer Sciences
Identifiers
URN: urn:nbn:se:kth:diva-370835DOI: 10.1109/EDGE67623.2025.00025Scopus ID: 2-s2.0-105015719072OAI: oai:DiVA.org:kth-370835DiVA, id: diva2:2003125
Conference
2025 IEEE International Conference on Edge Computing and Communications, EDGE 2025, Helsinki, Finland, July 7-12, 2025
Note

Part of ISBN 9798331555597

QC 20251003

Available from: 2025-10-03 Created: 2025-10-03 Last updated: 2025-10-03Bibliographically approved

Open Access in DiVA

No full text in DiVA

Other links

Publisher's full textScopus

Authority records

Girdzijauskas, Šarūnas

Search in DiVA

By author/editor
Girdzijauskas, Šarūnas
By organisation
Software and Computer systems, SCS
Computer Sciences

Search outside of DiVA

GoogleGoogle Scholar

doi
urn-nbn

Altmetric score

doi
urn-nbn
Total: 26 hits
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf