解决联邦学习中客户端不稳定性问题,提升系统鲁棒性与效率
Efficient Federated Learning against Heterogeneous and Non-stationary Client Unavailability
- 设计轻量级补偿机制,仅额外消耗常数级内存与计算
- 在非平稳环境下实现本地更新均匀扩散,收敛至非凸目标的驻点
- 适合移动设备或环境不确定场景下的实际部署
解决联邦学习中客户端间歇性不可用问题对实际部署至关重要。现有工作或忽略客户端不可用动态的非平稳性,或需高内存/计算开销。本文研究异构且非平稳的客户端不可用情形,常见于环境不确定或客户端移动的场景。以最广泛使用的FedAvg为例,说明异构性与非平稳性对不可用的影响显著。提出FedAPM算法,包含创新结构:(i) 仅以标准FedAvg的$O(1)$额外内存与计算,补偿因不可用导致的计算缺失;(ii) 通过隐式消息传递,在不依赖非平稳动态的前提下,实现本地更新在系统中的均衡扩散。证明了FedAPM可收敛至非凸目标的驻点,并具备理想的线性加速特性。在真实数据集上针对多样化客户端不可用动态的数值实验验证了分析结果。
原文摘要 · Abstract (English)
Addressing intermittent client availability is critical for the real-world deployment of federated learning algorithms. Most prior work either overlooks the potential non-stationarity in the dynamics of client unavailability or requires substantial memory/computation overhead. We study federated learning in the presence of heterogeneous and non-stationary client availability, which may occur when the deployment environments are uncertain, or the clients are mobile. The impacts of heterogeneity and non-stationarity on client unavailability can be significant, as we illustrate using FedAvg, the most widely adopted federated learning algorithm. We propose FedAPM, which includes novel algorithmic structures that (i) compensate for missed computations due to unavailability with only $O(1)$ additional memory and computation with respect to standard FedAvg, and (ii) evenly diffuse local updates within the federated learning system through implicit gossiping, despite being agnostic to non-stationary dynamics. We show that FedAPM converges to a stationary point of even non-convex objectives while achieving the desired linear speedup property. We corroborate our analysis with numerical experiments over diversified client unavailability dynamics on real-world data sets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。