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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | TinyImageNet Second Session Transition | Accuracy93.67 | 66 | |
| Image Classification | TinyImageNet Fourth Session Transition | Accuracy69.23 | 64 | |
| Text Classification | AG News (Second Session Transition) | Accuracy76.27 | 48 | |
| Text Classification | AG News (Fourth Session Transition) | Accuracy77.18 | 48 | |
| Image Classification | MNIST 1.0 (Second Session Transition) | Proposed Score93.99 | 16 | |
| Image Classification | MNIST 1.0 (Fourth Session Transition) | Proposed Score94.51 | 16 | |
| Image Classification | FMNIST 1.0 (Second Session Transition) | Proposed Accuracy85.17 | 16 | |
| Image Classification | FMNIST Fourth Session Transition 1.0 | Proposed Score85.82 | 16 | |
| Image Classification | SVHN Second Session Transition 1.0 | Proposed Accuracy93.64 | 16 | |
| Image Classification | CIFAR-100 Second Session Transition 1.0 | Proposed Score53.52 | 16 |