通过自蒸馏选择性约束本地更新,提升非独立同分布下的联邦学习性能
Learning Critically: Selective Self Distillation in Federated Learning on Non-IID Data
- 本地模型自蒸馏全局知识,按可信度分级加权
- 在3个公开数据集上减少通信轮次,提升泛化与鲁棒性
- 适合数据异构严重的实际联邦学习场景
联邦学习(FL)允许多个客户端在保持本地数据分散的前提下协同训练全局模型。客户端间的数据异质性(非独立同分布)给FL带来显著挑战,导致本地模型优化至各自局部最优并遗忘全局知识,造成性能下降和收敛变慢。现有方法虽通过引入基于全局模型的正则项缓解该问题,但缺乏自适应机制,难以在深度模型上高效实现高性能。本文提出一种面向联邦学习的自适应选择性自蒸馏方法(FedSSD),通过自蒸馏全局模型知识,并在类别和样本层面评估可信度后选择性加权,对本地更新施加动态约束。理论分析了FedSSD的收敛性,大量实验在三个公开基准数据集上验证:相比其他先进联邦学习方法,FedSSD在更少通信轮次下实现更好泛化能力与鲁棒性。
原文摘要 · Abstract (English)
Federated learning (FL) enables multiple clients to collaboratively train a global model while keeping local data decentralized. Data heterogeneity (non-IID) across clients has imposed significant challenges to FL, which makes local models re-optimize towards their own local optima and forget the global knowledge, resulting in performance degradation and convergence slowdown. Many existing works have attempted to address the non-IID issue by adding an extra global-model-based regularizing item to the local training but without an adaption scheme, which is not efficient enough to achieve high performance with deep learning models. In this paper, we propose a Selective Self-Distillation method for Federated learning (FedSSD), which imposes adaptive constraints on the local updates by self-distilling the global model's knowledge and selectively weighting it by evaluating the credibility at both the class and sample level. The convergence guarantee of FedSSD is theoretically analyzed and extensive experiments are conducted on three public benchmark datasets, which demonstrates that FedSSD achieves better generalization and robustness in fewer communication rounds, compared with other state-of-the-art FL methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。