摘要
arXiv:2503.04231v2 宣布类型: replace-cross
摘要:联邦学习(FL)是一种广泛采用的去中心化学习范式,允许从多个来源训练一个模型,而无需直接在参与客户端之间传输数据。自2015年诞生以来,联邦学习不断被划分为多个子领域,以解决与应用相关的问题,无论是数据异质性还是资源分配。其中一个子领域,聚类联邦学习(CFL),致力于将客户端人群聚类成不同的群体,以提供个性化模型。尽管在此领域发表了一些令人瞩目的工作,但该问题仍然很大程度上未被探索,因为它基本假设和设置与标准联邦学习略有不同。在本文中,我们提出了One-Shot聚类联邦学习(OCFL),这是一种不依赖于聚类的算法,可以自动检测最适合聚类的最早时刻。我们的算法基于客户端梯度之间的余弦相似度计算和一个温度度量,该度量在判断联邦模型开始收敛时被检测到。我们通过在三个基准数据集上对三十多种不同任务的多种一次聚类算法进行实验性评估,来评估我们的方法。我们的实验展示了在无需调整超参数的情况下,使用我们的方法自动执行CFL的优异性能。