Personalized federated learning based on self-excluding aggregation with knowledge distillation
摘要
As a decentralized machine learning paradigm, federated learning facilitates collaborative model training while inherently preserves client data privacy. However, in the non-IID settings, conventional federated learning approaches encounter poor model generalization. Group-based aggregation can alleviate the issues caused by non-IID data, but the lack of communication among groups leads to low data utilization. To address these challenges, a federated learning framework based on self-excluding aggregation (FedSEA) is proposed by integrating dynamic sparsity-driven client grouping, self-excluding aggregation and cross-group knowledge distillation. The FedSEA consists of three stages. First, privacy-preserving client grouping is achieved through sparsity analysis of feature mapping. Second, intra-group aggregation and self-exclusion aggregation are performed on grouped clients. Finally, cross-group knowledge distillation is conducted using knowledge distillation. Extensive experiments demonstrate that the FedSEA achieves superior performance on the Office-Home and CIFAR-10 datasets, with about 2% improvement in classification accuracy over the state-of-the-art methods.