FedACL: Federated Multi-Distillation with Auxiliary Classification Layers in IoT.

Zhipeng Gao,Futeng Feng,Zijia Mo,Zijian Xiong,Yang Yang
DOI: https://doi.org/10.1109/HPCC-DSS-SmartCity-DependSys57074.2022.00190
2022-01-01
Abstract:Federated learning is a popular privacy-preserving machine learning paradigm that can train a robust centralized model while guaranteeing the privacy of client data. However, traditional federated learning cannot cope with the problem of slow convergence and low accuracy due to model heterogeneity and statistical heterogeneity. To solve the problem, we propose a federated multi-distillation with auxiliary classification layers in IoT (FedACL). In FedACL, we add auxiliary classification layers after certain hidden layers in the local heterogeneous model, thus implementing a multi-distillation framework consisting of self-distillation in the local update and co-distillation in the communication. Besides, in order to improve the stability of model training, we propose a double-weighted method based on the test accuracy of the clients to measure the contribution of each client. A large number of experiments are conducted for different numbers of client and communication volumes under the Non-IID settings of CIFAR100 and FEMNIST datasets, and all of them prove that FedACL has fast convergence speed, high accuracy and good stability.
What problem does this paper attempt to address?