梯度压缩会损害联邦学习泛化能力,新方法用合成数据修复
Gradient Compression May Hurt Generalization: A Remedy by Synthetic Data Guided Sharpness Aware Minimization
- 用全局模型轨迹生成合成数据,精准估计扰动方向
- 在非独立同分布数据下,提升模型泛化性能1.5%以上
- 适合数据异构严重的联邦学习场景,尤其适用压缩通信
联邦学习中梯度压缩虽能显著提升通信效率,但本文发现其会引发更尖锐的损失曲面,尤其在非独立同分布(non-IID)数据下,削弱模型泛化能力。现有尖锐性感知最小化(SAM)方法因数据异构难以准确估计全局扰动。传统方法依赖前一轮模型更新作为近似,但在模型更新压缩时效果下降。本文提出FedSynSAM,利用全局模型轨迹构建合成数据,实现对全局扰动的精确估计。理论证明了算法收敛性,并通过大量实验验证其有效性,在多个非IID数据集上均取得显著性能提升。
原文摘要 · Abstract (English)
It is commonly believed that gradient compression in federated learning (FL) enjoys significant improvement in communication efficiency with negligible performance degradation. In this paper, we find that gradient compression induces sharper loss landscapes in federated learning, particularly under non-IID data distributions, which suggests hindered generalization capability. The recently emerging Sharpness Aware Minimization (SAM) effectively searches for a flat minima by incorporating a gradient ascent step (i.e., perturbing the model with gradients) before the celebrated stochastic gradient descent. Nonetheless, the direct application of SAM in FL suffers from inaccurate estimation of the global perturbation due to data heterogeneity. Existing approaches propose to utilize the model update from the previous communication round as a rough estimate. However, its effectiveness is hindered when model update compression is incorporated. In this paper, we propose FedSynSAM, which leverages the global model trajectory to construct synthetic data and facilitates an accurate estimation of the global perturbation. The convergence of the proposed algorithm is established, and extensive experiments are conducted to validate its effectiveness.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。