提出新型联邦学习算法,通信量随参数量对数增长,大幅降低通信开销。
Sketched Adaptive Federated Deep Learning: A Sharp Convergence Analysis
- 采用逐元素压缩与自适应优化结合,实现高效通信。
- 理论证明通信成本仅对数依赖参数量,非线性增长。
- 适用于异构数据场景,且在图像与语言任务中表现优异。
将梯度压缩(如CountSketch、量化)与自适应优化器(如Adam、AMSGrad)结合是联邦学习中的理想目标,有望减少通信轮次和每轮通信量。尽管此类方法已有初步实证成功,现有收敛分析显示通信成本与参数量呈线性依赖,对现代深度模型而言过高。本文提出特定的压缩自适应联邦学习(SAFL)算法,并首次在不同联邦学习设置下提供理论收敛分析,证明通信成本仅对数依赖于参数量。不同于以往分析,我们发现预处理矩阵和一阶矩中的逐元素压缩噪声可通过利用深度学习损失函数中最近流行的各向异性曲率(如损失海森矩阵特征值快速衰减)隐式缓解。在独立同分布客户端设置下,SAFL达到渐近$O(1/\sqrt{T})$收敛速度,初期收敛更快;在非独立同分布设置下,非自适应方法缺乏收敛保证,但本文提出的带剪裁的SAFL(SACFL)仍可保证收敛,即使存在重尾噪声。理论结论得到视觉与语言任务上的实证支持,涵盖微调与从头训练两种情形。意外的是,该分析还表明所提SAFL方法在通信效率上可媲美基于误差反馈的最先进算法。
原文摘要 · Abstract (English)
Combining gradient compression methods (e.g., CountSketch, quantization) and adaptive optimizers (e.g., Adam, AMSGrad) is a desirable goal in federated learning (FL), with potential benefits on both fewer communication rounds and less per-round communication. In spite of the preliminary empirical success of sketched adaptive methods, existing convergence analyses show the communication cost to have a linear dependence on the ambient dimension, i.e., number of parameters, which is prohibitively high for modern deep learning models. In this work, we introduce specific sketched adaptive federated learning (SAFL) algorithms and, as our main contribution, provide theoretical convergence analyses in different FL settings with guarantees on communication cost depending only logarithmically (instead of linearly) on the ambient dimension. Unlike existing analyses, we show that the entry-wise sketching noise existent in the preconditioners and the first moments of SAFL can be implicitly addressed by leveraging the recently-popularized anisotropic curvatures in deep learning losses, e.g., fast decaying loss Hessian eigen-values. In the i.i.d. client setting of FL, we show that SAFL achieves asymptotic $O(1/\sqrt{T})$ convergence, and converges faster in the initial epochs. In the non-i.i.d. client setting, where non-adaptive methods lack convergence guarantees, we show that SACFL (SAFL with clipping) algorithms can provably converge in spite of the additional heavy-tailed noise. Our theoretical claims are supported by empirical studies on vision and language tasks, and in both fine-tuning and training-from-scratch regimes. Surprisingly, as a by-product of our analysis, the proposed SAFL methods are competitive with the state-of-the-art communication-efficient federated learning algorithms based on error feedback.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。