解释联邦学习在数据异构下无法完美拟合的原因
Why Federated Optimization Fails to Achieve Perfect Fitting? A Theoretical Perspective on Client-Side Optima
- 假设客户端数据导致不同局部最优,推导全局目标下界上升
- 训练末期全局模型在区域震荡,无法收敛到单一最优解
- 为非独立同分布场景下的性能下降提供理论依据
联邦优化是一种受限的分布式优化方法,可在不直接共享客户端数据的情况下训练全局模型。尽管现有算法理论上可保证收敛,实践中也常表现稳定,但在数据异构情况下性能下降的原因仍不明确。本文从理论角度揭示该现象根源:假设异构客户端数据导致不同的局部最优,由此推导出两个关键结论:1)客户端局部最优间的距离抬高了全局目标函数的下界,使得完全拟合所有客户端数据成为不可能;2)在训练末期,全局模型在某一区域内震荡而非收敛至单一最优,限制了其对数据的完全拟合能力。这些结果为非独立同分布(non-iid)设置下的性能退化提供了原理性解释,并通过多种任务和神经网络架构的实验进行了验证。论文框架已在 https://github.com/NPCLEI/fedtorch 开源。
原文摘要 · Abstract (English)
Federated optimization is a constrained form of distributed optimization that enables training a global model without directly sharing client data. Although existing algorithms can guarantee convergence in theory and often achieve stable training in practice, the reasons behind performance degradation under data heterogeneity remain unclear. To address this gap, the main contribution of this paper is to provide a theoretical perspective that explains why such degradation occurs. We introduce the assumption that heterogeneous client data lead to distinct local optima, and show that this assumption implies two key consequences: 1) the distance among clients' local optima raises the lower bound of the global objective, making perfect fitting of all client data impossible; and 2) in the final training stage, the global model oscillates within a region instead of converging to a single optimum, limiting its ability to fully fit the data. These results provide a principled explanation for performance degradation in non-iid settings, which we further validate through experiments across multiple tasks and neural network architectures. The framework used in this paper is open-sourced at: https://github.com/NPCLEI/fedtorch.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。