arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.02589cs.LGstat.ML
收藏

$z$ - SignFedAvg:一种用于联邦学习的统一随机符号压缩

$z$-SignFedAvg: A Unified Stochastic Sign-based Compression for Federated Learning

The Chinese University of Hong Kong Shenzhen(香港中文大学(深圳))

Zhiwei Tang, Yanmeng Wang, Tsung-Hui Chang

摘要:

联邦学习(Federated Learning,FL)是一种很有前途的隐私保护分布式学习范式,但在训练大规模机器学习模型时存在通信代价高的问题。基于符号的方法,例如SignSGD \cite{bernstein2018signsgd},已经被提出作为用于降低通信成本的偏置梯度压缩技术。然而,基于符号的算法在异构数据下可能会出现分歧,这促使了高级技术的发展,例如错误反馈方法和基于符号的随机压缩,以解决这个问题。然而,这些方法仍然遭受较慢的收敛速度。此外,它们都不像FedAvg \cite{mcmahan2017communication}那样允许多个本地SGD更新。在本文中,我们提出了一种新的噪声扰动计划与一般对称噪声分布的符号为基础的压缩,它不仅允许一个灵活地控制梯度偏置和收敛性能之间的权衡,但也提供了一个统一的观点,现有的随机符号为基础的方法。更重要的是,统一的噪声扰动计划,使第一个基于符号的FedAvg算法($z$ -SignFedAvg)的发展,以加速收敛。理论上,我们表明,$z$ -SignFedAvg实现了更快的收敛速度比现有的基于符号的方法,均匀分布的噪声下,可以享受相同的收敛速度,其未压缩的对应。大量的实验表明,$z$ -SignFedAvg可以在真实数据集上实现有竞争力的经验性能,并优于现有的计划。

英文摘要:

Federated Learning (FL) is a promising privacy-preserving distributed learning paradigm but suffers from high communication cost when training large-scale machine learning models. Sign-based methods, such as SignSGD \cite{bernstein2018signsgd}, have been proposed as a biased gradient compression technique for reducing the communication cost. However, sign-based algorithms could diverge under heterogeneous data, which thus motivated the development of advanced techniques, such as the error-feedback method and stochastic sign-based compression, to fix this issue. Nevertheless, these methods still suffer from slower convergence rates. Besides, none of them allows multiple local SGD updates like FedAvg \cite{mcmahan2017communication}. In this paper, we propose a novel noisy perturbation scheme with a general symmetric noise distribution for sign-based compression, which not only allows one to flexibly control the tradeoff between gradient bias and convergence performance, but also provides a unified viewpoint to existing stochastic sign-based methods. More importantly, the unified noisy perturbation scheme enables the development of the very first sign-based FedAvg algorithm ($z$-SignFedAvg) to accelerate the convergence. Theoretically, we show that $z$-SignFedAvg achieves a faster convergence rate than existing sign-based methods and, under the uniformly distributed noise, can enjoy the same convergence rate as its uncompressed counterpart. Extensive experiments are conducted to demonstrate that the $z$-SignFedAvg can achieve competitive empirical performance on real datasets and outperforms existing schemes.

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

↑