arXiv:2605.11165cs.LG2026-05

用伪标签实现服务器端个性化联邦学习,提升异构环境下的模型性能。

COSMOS: Model-Agnostic Personalized Federated Learning with Clustered Server Models and Pseudo-Label-Only Communication

论文配图:COSMOS: Model-Agnostic Personalized Federated Learning with Clustered Server Models and Pseudo-Label-Only Communication
图 1 · 摘自论文原文
  • 服务器基于客户端预测相似性聚类,训练专属模型并回传。
  • 理论证明可实现指数级个性化风险下降,优于传统方法。
  • 仅需伪标签通信,适合大规模异构设备部署。

在异构环境中,联邦学习面临客户端模型架构和数据分布差异的双重挑战。现有方法虽尝试通过客户端聚类和知识蒸馏缓解问题,但同时应对架构与统计异质性仍困难。本文提出COSMOS,一种模型无关框架,仅通过伪标签通信即可实现服务器端个性化。客户端本地训练并在公开数据上预测;服务器根据预测相似性聚类客户端,为每组独立训练集群模型,并将结果蒸馏回客户端。我们首次提供理论分析,证明从学习到的集群模型蒸馏可带来指数级个性化风险收缩,超越传统模型无关联邦学习的收敛至稳定保证。在多个基准测试中,COSMOS持续优于所有模型无关联邦学习基线,且与最先进的个性化联邦学习方法相当。结果表明,仅使用伪标签的个性化服务器学习是一种在高度异构环境下具有前景的可扩展范式。

原文摘要 · Abstract (English)

Federated learning (FL) in heterogeneous environments remains challenging because client models often differ in both architecture and data distribution. While recent approaches attempt to address this challenge through client clustering and knowledge distillation, simultaneously handling architectural and statistical heterogeneity remains difficult. We introduce COSMOS, a model-agnostic framework that enables server-side personalization using only pseudo-label communication. Clients train local models and predict on the public data; the server clusters clients by prediction similarity, trains a cluster-specific model for each group using its own compute, and distills the resulting models back to clients. We provide the first theoretical analysis showing that distillation from the learned cluster models can yield exponential personalization risk contraction, going beyond the convergence-to-stationarity guarantees typically provided in model-agnostic FL. Experiments across benchmarks demonstrate that COSMOS consistently outperforms all model-agnostic FL baselines while remaining competitive with state-of-the-art personalized FL methods. More broadly, our results highlight personalized server-side learning with pseudo-labels as a promising paradigm for scalable and model-agnostic federated learning in highly heterogeneous environments.

联邦学习个性化伪标签异构

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