arXiv:2504.05153cs.LG2025-04被引 3

提出高效自适应的稀疏联邦训练方法,显著降低通信开销。

SparsyFed: Sparse Adaptive Federated Training

  • 通过自适应稀疏掩码生成,动态匹配异构数据分布。
  • 实现95%稀疏度下精度几乎无损失,仅需一个超参数。
  • 相比以往方法,每轮权重恢复量减少200倍,适合移动端应用。

稀疏训练常用于跨设备联邦学习(FL)环境,设备在私有数据上协同训练模型,通过交换伪梯度实现通信。尽管稀疏训练能降低通信开销与计算负担,但实际应用受限:(1) 数据异构性使客户端难以就稀疏模型达成共识,导致训练时间延长;(2) 稀疏掩码获取方法缺乏对高度异构数据分布的适应性;(3) 需额外调节超参数,而联邦学习中调参尤为困难。本文提出SparsyFed,一种实用的联邦稀疏训练方法,有效解决上述问题。现有工作通常只解决其中一两个挑战,且引入新权衡,如掩码共识与稀疏模式适应性的矛盾。本研究证明,SparsyFed可同时实现:(1) 生成95%稀疏模型,精度损失可忽略,且仅需单一超参数;(2) 每轮权重恢复量比先前方法小200倍;(3) 掩码能自适应高度异构数据分布,在此类条件下优于所有基线方法。

原文摘要 · Abstract (English)

Sparse training is often adopted in cross-device federated learning (FL) environments where constrained devices collaboratively train a machine learning model on private data by exchanging pseudo-gradients across heterogeneous networks. Although sparse training methods can reduce communication overhead and computational burden in FL, they are often not used in practice for the following key reasons: (1) data heterogeneity makes it harder for clients to reach consensus on sparse models compared to dense ones, requiring longer training; (2) methods for obtaining sparse masks lack adaptivity to accommodate very heterogeneous data distributions, crucial in cross-device FL; and (3) additional hyperparameters are required, which are notably challenging to tune in FL. This paper presents SparsyFed, a practical federated sparse training method that critically addresses the problems above. Previous works have only solved one or two of these challenges at the expense of introducing new trade-offs, such as clients' consensus on masks versus sparsity pattern adaptivity. We show that SparsyFed simultaneously (1) can produce 95% sparse models, with negligible degradation in accuracy, while only needing a single hyperparameter, (2) achieves a per-round weight regrowth 200 times smaller than previous methods, and (3) allows the sparse masks to adapt to highly heterogeneous data distributions and outperform all baselines under such conditions.

联邦学习稀疏训练自适应通信优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。