Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Communication-Efficient Federated Learning under Dynamic Device Arrival and Departure: Convergence Analysis and Algorithm Design

About

Most federated learning (FL) approaches assume a fixed device set. However, real-world scenarios often involve devices dynamically joining or leaving the system, driven by, e.g., user mobility patterns or handovers across cell boundaries. This dynamic setting introduces unique challenges: (1) the optimization objective evolves with the active device set, unlike traditional FL's static objective; and (2) the current global model may no longer serve as an effective initialization for subsequent rounds, potentially hindering adaptation, delaying convergence, and reducing resource efficiency. To address these challenges, we first provide a convergence analysis for FL under a dynamic device set, accounting for factors such as gradient noise, local training iterations, and data heterogeneity in this practical setting. Motivated by this analysis, we propose a model initialization algorithm that enables rapid adaptation whenever devices join or leave the network. Our key idea is to compute a weighted average of previous global models, guided by gradient similarity, to prioritize models trained on data distributions that closely align with the current device set, thereby accelerating recovery from distribution shifts in fewer training rounds. This plug-and-play algorithm is designed to integrate seamlessly with existing FL methods, offering broad applicability. Experiments demonstrate that our approach achieves convergence speedups typically an order of magnitude or more compared to baselines, which we show drastically reduces energy consumption to reach a target accuracy.

Zhan-Lun Chang, Dong-Jun Han, Seyyedali Hosseinalipour, Mung Chiang, Christopher G. Brinton• 2024

Related benchmarks

TaskDatasetResultRank
Image ClassificationTinyImageNet Second Session Transition
Accuracy93.67
66
Image ClassificationTinyImageNet Fourth Session Transition
Accuracy69.23
64
Text ClassificationAG News (Second Session Transition)
Accuracy76.27
48
Text ClassificationAG News (Fourth Session Transition)
Accuracy77.18
48
Image ClassificationMNIST 1.0 (Second Session Transition)
Proposed Score93.99
16
Image ClassificationMNIST 1.0 (Fourth Session Transition)
Proposed Score94.51
16
Image ClassificationFMNIST 1.0 (Second Session Transition)
Proposed Accuracy85.17
16
Image ClassificationFMNIST Fourth Session Transition 1.0
Proposed Score85.82
16
Image ClassificationSVHN Second Session Transition 1.0
Proposed Accuracy93.64
16
Image ClassificationCIFAR-100 Second Session Transition 1.0
Proposed Score53.52
16
Showing 10 of 18 rows

Other info

Follow for update