arXiv:2505.18981cs.LG2025-05被引 1

通过结构化知识协作,提升非独立同分布数据下的联邦学习性能

FedSKC: Federated Learning with Non-IID Data via Structural Knowledge Collaboration

  • 从客户端内部结构中提取类别知识,缓解数据异构性
  • 在多个真实数据集上实现比现有方法更高的准确率
  • 适合处理边缘设备数据差异大的场景

随着边缘计算的发展,联邦学习(FL)作为一种保护隐私的协同学习范式展现出广阔前景。然而,其主要挑战之一是数据异构性问题,即不同客户端间存在标签偏好偏差,影响模型收敛与性能。以往方法多从局部或全局角度应对,忽略了客户端内部的类别级结构信息。本文首次分析数据异构对模型偏离的影响,并将其分解为本地、全局和采样漂移三类子问题。为此提出基于结构知识协作的联邦学习框架FedSKC:通过局部对比学习防止权重发散;全局差异聚合解决服务器与客户端间的参数偏差;全局周期审查修正服务器随机选设备带来的采样漂移。理论分析表明,在非凸目标下该方法具有收敛性,实验验证其在多个数据集上显著优于基线方法。

原文摘要 · Abstract (English)

With the advancement of edge computing, federated learning (FL) displays a bright promise as a privacy-preserving collaborative learning paradigm. However, one major challenge for FL is the data heterogeneity issue, which refers to the biased labeling preferences among multiple clients, negatively impacting convergence and model performance. Most previous FL methods attempt to tackle the data heterogeneity issue locally or globally, neglecting underlying class-wise structure information contained in each client. In this paper, we first study how data heterogeneity affects the divergence of the model and decompose it into local, global, and sampling drift sub-problems. To explore the potential of using intra-client class-wise structural knowledge in handling these drifts, we thus propose Federated Learning with Structural Knowledge Collaboration (FedSKC). The key idea of FedSKC is to extract and transfer domain preferences from inter-client data distributions, offering diverse class-relevant knowledge and a fair convergent signal. FedSKC comprises three components: i) local contrastive learning, to prevent weight divergence resulting from local training; ii) global discrepancy aggregation, which addresses the parameter deviation between the server and clients; iii) global period review, correcting for the sampling drift introduced by the server randomly selecting devices. We have theoretically analyzed FedSKC under non-convex objectives and empirically validated its superiority through extensive experimental results.

联邦学习非IID知识协作边缘计算

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