摘要
arXiv:2501.05496v2 宣告类型: 替换交叉
摘要:原型驱动的联邦学习已成为一种有前途的方法,能够在保持模型通用性的同时,通过共享轻量级原型在数据异质性的情况下将知识转移到客户端之间。然而,现有的方法通常直接从本地模型中收集原型,由于客户端之间存在有偏数据分布和模型架构差异,这不可避免地会在表征学习中引入不一致性。在本文中,我们发现统计异质性和模型异质性共同形成了表征不一致、分类器发散和偏斜原型对齐的恶性循环,这对客户端的性能产生了负面影响。为了打破这个恶性循环,我们提出了一种名为基于语义锚的联邦学习(FedSA)的新框架,以解耦原型生成与局部表征学习。我们引入了一种新颖的观点,即使用简单而有效的语义锚作为原型,指导本地模型学习一致的表征。通过结合语义锚,我们进一步提出基于语义锚的正则化,包括增强对比学习和基于语义锚的分类器校准,以纠正特征提取器并跨客户端校准分类器,从而在确保一致决策边界的前提下实现原型内的紧凑性和类别间可分性。然后,我们使用这些一致且具有区分性的原型来更新语义锚,这会逐步鼓励客户端协作学习具有稳健泛化的统一数据表示。在统计异质性和模型异质性设置下的广泛实验表明,FedSA 在各种分类任务中显著优于现有的基于原型的联邦学习方法。