arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.14280math.OCcs.LG
收藏

加速随机额外梯度:混合 Hessian 与梯度相似性以减少分布式和联邦学习中的通信

Accelerated Stochastic ExtraGradient: Mixing Hessian and Gradient Similarity to Reduce Communication in Distributed and Federated Learning

MIPT(莫斯科物理技术学院);ISP RAS(俄罗斯科学院信息学问题研究所);Innopolis University(因诺波利斯大学)

Dmitry Bylinkin, Kirill Degtyarev, Aleksandr Beznosikov

AI总结:

针对分布式与联邦学习的通信和隐私问题,提出结合 Hessian 相似性、同质梯度及客户端采样的加速随机额外梯度方法,并通过加噪分析收敛,实验验证理论。

摘要:

现代学习的现实和趋势要求模型具备越来越强的泛化能力,这导致模型规模和训练样本数量同时增加。在单设备模式下解决此类任务已经十分困难。这正是分布式和联邦学习方法日益普及的原因。分布式计算涉及设备之间的通信,需要解决效率和隐私这两个关键问题。应对通信成本最著名的方法之一是利用本地数据的相似性。Hessian 相似性和同质性梯度在文献中都已有研究,但此前是分开进行的。本文将这两种假设结合起来,分析一种新方法;该方法融入了利用数据相似性和客户端采样的思想。此外,为解决隐私问题,我们应用了添加噪声的技术,并分析其对所提方法收敛性的影响。在真实数据集上进行的训练证实了该理论。

英文摘要:

Modern realities and trends in learning require more and more generalization ability of models, which leads to an increase in both models and training sample size. It is already difficult to solve such tasks in a single device mode. This is the reason why distributed and federated learning approaches are becoming more popular every day. Distributed computing involves communication between devices, which requires solving two key problems: efficiency and privacy. One of the most well-known approaches to combat communication costs is to exploit the similarity of local data. Both Hessian similarity and homogeneous gradients have been studied in the literature, but separately. In this paper, we combine both of these assumptions in analyzing a new method that incorporates the ideas of using data similarity and clients sampling. Moreover, to address privacy concerns, we apply the technique of additional noise and analyze its impact on the convergence of the proposed method. The theory is confirmed by training on real datasets.

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

↑