联邦学习泛化误差分析的启示:您或许无需频繁通信!
机器学习
2024-06-11 v2 信息论
机器学习
math.IT
摘要
我们研究联邦学习(FL)设定下统计学习模型的泛化误差。具体而言,我们研究泛化误差随 个客户端与参数服务器(PS)之间通信轮数 的演化,即客户端本地模型在 PS 处聚合的频率对泛化误差的影响。在我们的设定中,客户端与 PS 通信越多,每轮用于本地训练的数据就越少,从而使得各客户端训练数据总量对于不同的 值保持一致。我们建立了显式考虑轮数 以及参与设备数 和个体数据集大小 影响的泛化误差的 PAC-Bayes 与率失真理论界。这些界适用于一大类损失函数与学习算法,似乎是 FL 设定下的首创。此外,我们将所提界应用于 FL 型支持向量机(FSVM),并推导出(更)显式的界。特别地,我们表明 FSVM 的泛化界随 增大,意味着与 PS 更频繁的通信削弱了泛化能力。这意味着总体风险随 的下降慢于经验风险。而且,我们的界表明 FSVM 的泛化误差比集中式学习快一个因子 下降。最后,我们给出使用神经网络(ResNet-56)获得的实验结果,其证据表明不仅我们对 FSVM 的观察可能更普遍地成立,而且总体风险甚至可能在某些 值之后开始上升。
引用
@article{arxiv.2306.05862,
title = {Lessons from Generalization Error Analysis of Federated Learning: You May Communicate Less Often!},
author = {Milad Sefidgaran and Romain Chor and Abdellatif Zaidi and Yijun Wan},
journal= {arXiv preprint arXiv:2306.05862},
year = {2024}
}
备注
Accepted at ICML 2024