notesum.ai

Published at December 4

Semi-decentralized Training of Spatio-Temporal Graph Neural Networks for Traffic Prediction

cs.LG

cs.AI

cs.DC

Released Date: December 4, 2024

Authors: Ivan Kralj¹, Lodovico Giaretta², Gordan Ježić, Ivana Podnar Žarko¹, Šarūnas Girdzijauskas

Aff.: ¹University of Zagreb; ²RISE Research Institutes of Sweden

Arxiv: http://arxiv.org/pdf/2412.03188v1

Refer to caption

Dataset	Setups	15 min			30 min			60 min
Dataset	Setups	MAE	RMSE	WMAPE	MAE	RMSE	WMAPE	MAE	RMSE	WMAPE
METR-LA	Centralized	3.78	9.05	7.45	5.14	11.61	10.12	7.35	14.59	14.47
	Traditional FL	3.97	9.01	7.82	5.40	11.41	10.63	7.82	14.48	15.40
	Server-free FL	3.90	8.98	7.78	5.35	11.37	10.67	7.79	14.41	15.55
	Gossip Learning	3.88	9.04	7.74	5.43	11.35	10.85	7.56	14.42	15.10
PeMS-BAY	Centralized	1.48	3.09	2.38	1.97	4.23	3.17	2.59	5.41	4.17
	Traditional FL	1.50	3.12	2.42	2.03	4.29	3.27	2.67	5.51	4.29
	Server-free FL	1.50	3.12	2.42	2.01	4.25	3.23	2.65	5.44	4.28
	Gossip Learning	1.51	3.12	2.43	2.03	4.28	3.26	2.63	5.39	4.24