LLM2D
域偏移下的增强型联邦原型学习方法
An Enhanced Federated Prototype Learning Method under Domain Shift
作者: Liang Kuang, Kuangpu Guo, Jian Liang, Jianguo Zhang
发布日期: 9/30/2024
arXiv ID: oai:arXiv.org:2409.18578v1

摘要

联邦学习 (FL) 允许在不共享私有数据的情况下进行协作式机器学习训练。大量研究表明,影响联邦学习模型性能的一个重要因素是不同客户端之间数据的异质性,尤其是在数据来自不同领域的情况下。一篇最近的论文介绍了方差感知双层原型聚类,并使用了一种新颖的 $\alpha$ -稀疏原型损失,它增加了类内相似度并减少了类间相似度。为了确保特征在特定集群内收敛,我们引入了一种改进的算法,即具有收敛集群的联邦原型学习,简称 FedPLCC。为了增加类间距离,我们根据每个原型所代表的集群的大小对每个原型进行加权。为了减少类内距离,考虑到距离较大的原型可能来自不同的领域,我们只选择一定比例的原型用于损失函数计算。在 Digit-5、Office-10 和 DomainNet 数据集上的评估表明,我们的方法优于现有方法。