Apr, 2024

使用对比共享表征进行个性化的联邦学习,解决非独立同分布数据中的标签差异性

TL;DR本文提出了一种名为Federated Contrastive Representation Learning (FedCRL)的个性化联邦学习算法,旨在处理分布式机器学习场景中由标签分布偏斜和数据稀缺性造成的异质性。FedCRL通过在共享表示上引入对比表示学习(CRL)来促进客户端的知识获取,从而提升本地模型的效果并避免受到标签分布偏斜的伤害。此外,FedCRL采用本地聚合机制来处理数据稀缺性,并引入损失加权机制来指导全局模型与每个客户端的协调,以帮助具有稀缺数据的客户端。通过在具有不同程度标签异质性的数据集上进行模拟,结果表明FedCRL在减轻标签异质性方面的有效性,实现了相对于现有方法的准确性改进。