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

基于局部步长的分散梯度跟踪

Decentralized Gradient Tracking with Local Steps

University of Toronto(多伦多大学);Westlake University(西湖大学);EPFL(洛桑联邦理工学院);CISPA Helmholtz Center for Information Security(亥姆霍兹信息安全中心)

Yue Liu, Tao Lin, Anastasia Koloskova, Sebastian U. Stich

摘要:

梯度跟踪(GT)是一种算法,设计用于解决网络上的分散式最优化问题(例如训练机器学习模型)。GT的一个关键特性是一个跟踪机制,可以克服节点之间的数据异构性。我们提出了一种新的分散跟踪机制,$K$ -GT,使通信有效的本地更新GT,同时继承的数据独立性属性的GT。我们证明了收敛速度$K$ -GT光滑非凸函数,并证明它减少了通信开销渐近的线性因子$K$,其中$K$表示的本地步骤的数量。我们说明了这种异质性校正的鲁棒性和有效性凸和非凸基准问题和非凸神经网络训练任务与MNIST数据集。

英文摘要:

Gradient tracking (GT) is an algorithm designed for solving decentralized optimization problems over a network (such as training a machine learning model). A key feature of GT is a tracking mechanism that allows to overcome data heterogeneity between nodes. We develop a novel decentralized tracking mechanism, $K$-GT, that enables communication-efficient local updates in GT while inheriting the data-independence property of GT. We prove a convergence rate for $K$-GT on smooth non-convex functions and prove that it reduces the communication overhead asymptotically by a linear factor $K$, where $K$ denotes the number of local steps. We illustrate the robustness and effectiveness of this heterogeneity correction on convex and non-convex benchmark problems and on a non-convex neural network training task with the MNIST dataset.

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

↑