arXivDaily arXiv每日学术速递 周一至周五更新
看起来今天可以划水了
arXiv 2202.08402cs.LG
收藏

联邦随机梯度下降产生自诱导动量

Federated Stochastic Gradient Descent Begets Self-Induced Momentum

Zhejiang University/University of Illinois at Urbana-Champaign Institute(浙江大学-伊利诺伊大学厄巴纳香槟校区联合学院);Hong Kong Metropolitan University(香港都会大学);Singapore University of Technology and Design(新加坡科技设计大学);Princeton University(普林斯顿大学)

Howard H. Yang, Zuozhu Liu, Yaru Fu, Tony Q. S. Quek, H. Vincent Poor

摘要:

联邦学习(FL)是一种新兴的机器学习方法,可以应用于移动边缘系统,其中服务器和客户端主机利用客户端的数据和计算资源协作地训练统计模型,而不直接暴露其隐私敏感数据。我们表明,运行随机梯度下降(SGD)在这样的设置可以被视为添加一个动量项的全球聚集过程。基于这一发现,我们进一步分析了一个联邦学习系统的收敛速度,占参数陈旧和通信资源的影响。这些结果促进了对联邦SGD算法的理解,并且也在陈旧性分析和联邦计算系统之间建立了联系,这对系统设计者来说是有用的。

英文摘要:

Federated learning (FL) is an emerging machine learning method that can be applied in mobile edge systems, in which a server and a host of clients collaboratively train a statistical model utilizing the data and computation resources of the clients without directly exposing their privacy-sensitive data. We show that running stochastic gradient descent (SGD) in such a setting can be viewed as adding a momentum-like term to the global aggregation process. Based on this finding, we further analyze the convergence rate of a federated learning system by accounting for the effects of parameter staleness and communication resources. These results advance the understanding of the Federated SGD algorithm, and also forges a link between staleness analysis and federated computing systems, which can be useful for systems designers.

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

↑