arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2310.03461cs.LGmath.OC

深入理解集中式与去中心化联邦学习之间的泛化与稳定性差距

Towards Understanding Generalization and Stability Gaps between Centralized and Decentralized Federated Learning

University of Sydney(悉尼大学);Sun Yat-sen University(中山大学);Nanyang Technological University(南洋理工大学)

Yan Sun, Li Shen, Dacheng Tao

AI总结:

本文探讨了集中式与去中心化联邦学习的泛化与稳定性差距,证明在平滑非凸目标下集中式泛化更优且部分参与最佳,并指出去中心化需满足拓扑条件以防性能崩溃,实验验证了理论分析。

摘要:

作为联邦学习(FL)中的两种主流框架,集中式和去中心化方法在实际场景中都展现了巨大的应用价值。然而,现有研究并未为分析联邦学习社区中哪种方法表现更好提供充分的证据和明确的指导。尽管去中心化方法已被证明能够以更少的通信量达到与集中式相当的收敛性,但在实证研究中,其测试性能总是不及预期。为了全面且公平地比较它们在联邦学习中的效率差距,本文探讨了它们的稳定性和泛化效率。具体而言,我们证明在一般平滑非凸目标上:1)集中式联邦学习(CFL)的泛化能力始终优于去中心化联邦学习(DFL);2)CFL通过采用部分参与而非全参与来实现最佳性能;3)随着训练规模的增加,DFL中的拓扑结构必须满足必要条件以避免性能崩溃。我们还在联邦学习的几种常见设置上进行了广泛实验,验证了我们的理论分析与实验现象一致,并在若干一般性和实用性场景中具有上下文有效性。

英文摘要:

As two mainstream frameworks in federated learning (FL), both centralized and decentralized approaches have shown great application value in practical scenarios. However, existing studies do not provide sufficient evidence and clear guidance for analysis of which performs better in the FL community. Although decentralized methods have been proven to approach the comparable convergence of centralized with less communication, their test performance always falls short of expectations in empirical studies. To comprehensively and fairly compare their efficiency gaps in FL, in this paper, we explore their stability and generalization efficiency. Specifically, we prove that on the general smooth non-convex objectives, 1) centralized FL (CFL) always generalizes better than decentralized FL (DFL); 2) CFL achieves the best performance via adopting partial participation instead of full participation; and, 3) there is a necessary requirement for the topology in DFL to avoid performance collapse as the training scale increases. We also conduct extensive experiments on several common setups in FL to validate that our theoretical analysis is consistent with experimental phenomena and contextually valid in several general and practical scenarios.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑