arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.13136cs.LGcs.CRcs.CV
收藏

基于标签掩码蒸馏的联邦学习

Federated Learning with Label-Masking Distillation

Institute of Information Engineering, Chinese Academy of Sciences(中国科学院信息工程研究所);School of Cyber Security, UCAS(中国科学院大学网络空间安全学院);School of Computer Science, Fudan University(复旦大学计算机科学技术学院)

Jianghu Lu, Shikun Li, Kexin Bao, Pengju Wang, Zhenxing Qian, Shiming Ge

AI总结:

针对联邦学习中的标签分布倾斜问题,提出标签掩码蒸馏方法 FedLMD,通过屏蔽多数标签预测使全局模型专注保留少数标签知识,并设计无需额外教师的轻量级变体 FedLMD-Tf,实现最优性能。

摘要:

联邦学习提供了一种隐私保护的方式,通过全局服务器的协调,在分布于多个本地客户端的数据上协同训练模型。本文重点关注联邦学习中的标签分布倾斜问题,由于客户端用户行为不同,不同客户端之间的标签分布存在显著差异。面对这种情况,大多数现有方法由于未能充分利用客户端的标签分布信息,会导致次优优化。受此启发,我们提出了一种称为 FedLMD 的标签掩码蒸馏方法,通过感知每个客户端的各种标签分布来促进联邦学习。在训练过程中,我们根据每个类别的样本数量将标签分为多数标签和少数标签。客户端模型从本地数据中学习多数标签的知识。蒸馏过程会屏蔽全局模型对多数标签的预测,使其能更专注于保留客户端的少数标签知识。一系列实验表明,所提方法能在各种情况下达到最先进的性能。此外,考虑到客户端资源有限,我们提出了变体 FedLMD-Tf,无需额外教师模型,在不增加计算成本的情况下优于以往的轻量级方法。代码可在 https://github.com/wnma3mz/FedLMD 获取。

英文摘要:

Federated learning provides a privacy-preserving manner to collaboratively train models on data distributed over multiple local clients via the coordination of a global server. In this paper, we focus on label distribution skew in federated learning, where due to the different user behavior of the client, label distributions between different clients are significantly different. When faced with such cases, most existing methods will lead to a suboptimal optimization due to the inadequate utilization of label distribution information in clients. Inspired by this, we propose a label-masking distillation approach termed FedLMD to facilitate federated learning via perceiving the various label distributions of each client. We classify the labels into majority and minority labels based on the number of examples per class during training. The client model learns the knowledge of majority labels from local data. The process of distillation masks out the predictions of majority labels from the global model, so that it can focus more on preserving the minority label knowledge of the client. A series of experiments show that the proposed approach can achieve state-of-the-art performance in various cases. Moreover, considering the limited resources of the clients, we propose a variant FedLMD-Tf that does not require an additional teacher, which outperforms previous lightweight approaches without increasing computational costs. Our code is available at https://github.com/wnma3mz/FedLMD.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑