arXiv:2501.05496cs.LGcs.AI2025-01AAAI被引 16

用语义锚点统一联邦学习表示,解决数据与模型异构下的表征不一致问题。

FedSA: A Unified Representation Learning via Semantic Anchors for Prototype-based Federated Learning

  • 引入语义锚点作为原型,分离本地表征学习与原型生成过程。
  • 在统计与模型异构下,准确率提升12.3%以上,显著优于现有方法。
  • 适合数据分布差异大、模型结构不一致的联邦学习场景使用。

基于原型的联邦学习通过共享轻量级原型,在模型无关的前提下实现客户端间知识迁移。然而,现有方法直接从本地模型提取原型,受数据分布偏斜和模型架构差异影响,导致表征学习不一致。本文指出统计与模型异构引发表征不一致、分类器发散与原型对齐偏差的恶性循环,损害客户端性能。为此提出联邦学习语义锚点框架(FedSA),将原型生成与本地表征学习解耦。通过引入简单有效的语义锚点作为原型,引导本地模型学习一致表征。结合锚点正则化与增强对比学习、锚点校准分类器,实现原型类内紧凑、类间可分,且保证决策边界一致。迭代更新语义锚点以融合一致判别性原型,协同构建统一鲁棒表征。在统计与模型异构设置下大量实验表明,FedSA在多个分类任务上显著超越现有原型联邦学习方法。

原文摘要 · Abstract (English)

Prototype-based federated learning has emerged as a promising approach that shares lightweight prototypes to transfer knowledge among clients with data heterogeneity in a model-agnostic manner. However, existing methods often collect prototypes directly from local models, which inevitably introduce inconsistencies into representation learning due to the biased data distributions and differing model architectures among clients. In this paper, we identify that both statistical and model heterogeneity create a vicious cycle of representation inconsistency, classifier divergence, and skewed prototype alignment, which negatively impacts the performance of clients. To break the vicious cycle, we propose a novel framework named Federated Learning via Semantic Anchors (FedSA) to decouple the generation of prototypes from local representation learning. We introduce a novel perspective that uses simple yet effective semantic anchors serving as prototypes to guide local models in learning consistent representations. By incorporating semantic anchors, we further propose anchor-based regularization with margin-enhanced contrastive learning and anchor-based classifier calibration to correct feature extractors and calibrate classifiers across clients, achieving intra-class compactness and inter-class separability of prototypes while ensuring consistent decision boundaries. We then update the semantic anchors with these consistent and discriminative prototypes, which iteratively encourage clients to collaboratively learn a unified data representation with robust generalization. Extensive experiments under both statistical and model heterogeneity settings show that FedSA significantly outperforms existing prototype-based FL methods on various classification tasks.

联邦学习原型学习语义锚点表征统一

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