摘要
arXiv:2503.22971v1 公告类型: cross
摘要:联邦学习(FL)已成为机器学习领域的一个有前途的方法,它能够在不共享原始数据的情况下,通过分散设备之间的协作进行模型训练。在联邦学习中,全球模型会在各个设备上本地数据集上传递训练,每个设备都会为模型改进做出贡献。然而,这些本地数据集的异质性,源于不同的用户行为、设备能力和数据分布,提出了显著的挑战。联邦学习中固有的异质性导致了各种问题,包括模型性能差异、收敛挑战以及潜在的隐私问题。随着全球模型在训练轮次中的进展,本地数据质量与数量的差异可能会阻碍联邦学习系统的整体效果。此外,确保不同用户群体的公平性和隐私变得极为重要。为了解决这一问题,本文提出了一种名为ClusterGuardFL的新联邦学习框架,该框架采用差异性得分、k-means聚类和调整信任得分来动态分配客户端更新的权重。全局模型与本地模型之间的差异性得分指导簇的形成,簇的大小影响权重分配。在每个簇内,为个别数据点计算调整信任得分,softmax层生成针对客户端的定制化权重。这些权重在聚合过程中使用,增强模型的稳健性和隐私性。实验结果表明,提出的方法在多样化的数据集上实现了更好的模型性能。