通过损失向量嵌入聚类,实现无需初始化的个性化联邦学习。
CLoVE: Personalized Federated Learning through Clustering of Loss Vector Embeddings
- 用模型在客户端数据上的损失向量生成嵌入,聚类识别不同数据分布的客户端。
- 仅需几轮训练即可精准恢复集群,线性设置下收敛速度指数级提升。
- 适用于监督与无监督任务,无需理想初始模型,适合真实场景应用。
我们提出 CLoVE(损失向量嵌入聚类),一种用于集群联邦学习(CFL)的新算法。在 CFL 中,客户端因其数据分布自然分组,但客户归属未知。CLoVE 利用客户端在自身数据上模型损失生成的嵌入,基于同一簇内客户端损失值相似、不同簇间损失模式相异的洞察,迭代识别并分离不同簇的客户端,并通过联邦聚合优化各簇专属模型。相比现有 CFL 算法,其优势在于:(1)方法简单;(2)可适配监督与无监督场景;(3)无需近似最优模型初始化,更具鲁棒性,更适用于真实应用。我们建立了理论收敛界,表明在单轮中高概率准确恢复集群,并在线性设定下指数级快速收敛至最优模型。全面实验在多种数据集和非独立同分布(non-IID)设置下对比了多种 CFL 与通用个性化联邦学习(PFL)算法,结果表明,仅经数轮训练,CLoVE 即能实现高精度集群恢复,并在各类监督与无监督 PFL 任务中达到顶尖模型精度。
原文摘要 · Abstract (English)
We propose CLoVE (Clustering of Loss Vector Embeddings), a novel algorithm for Clustered Federated Learning (CFL). In CFL, clients are naturally grouped into clusters based on their data distribution. However, identifying these clusters is challenging, as client assignments are unknown. CLoVE utilizes client embeddings derived from model losses on client data, and leverages the insight that clients in the same cluster share similar loss values, while those in different clusters exhibit distinct loss patterns. Based on these embeddings, CLoVE is able to iteratively identify and separate clients from different clusters and optimize cluster-specific models through federated aggregation. Key advantages of CLoVE over existing CFL algorithms are (1) its simplicity, (2) its applicability to both supervised and unsupervised settings, and (3) the fact that it eliminates the need for near-optimal model initialization, which makes it more robust and better suited for real-world applications. We establish theoretical convergence bounds, showing that CLoVE can recover clusters accurately with high probability in a single round and converges exponentially fast to optimal models in a linear setting. Our comprehensive experiments comparing with a variety of both CFL and generic Personalized Federated Learning (PFL) algorithms on different types of datasets and an extensive array of non-IID settings demonstrate that CLoVE achieves highly accurate cluster recovery in just a few rounds of training, along with state-of-the-art model accuracy, across a variety of both supervised and unsupervised PFL tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。