Robust Heterogeneous Federated Learning Via Data-Free Knowledge Amalgamation.

Jun Ma,Zheng Fan,Chaoyu Fan,Qi Kang
DOI: https://doi.org/10.1007/978-981-97-7184-4_6
2024-01-01
Abstract:Federated Learning (FL) served as distributed learning paradigm, which enables multiple parties to collaboratively train a machine learning model without integrating their private data together for privacy. Akey challenge in federated learning is to handle the heterogeneity of local data distribution across parties. Although many studies have been proposed to address this challenge, we find they made improvements either in local training or in aggregation. In this paper, to overcome these issues, we propose a data-free approach to mitigate heterogeneity and catastrophic forgetting in FL scenario, named FedKA. In the server, it trains the global model by a generator training stage and a knowledge amalgamation stage. In the client, it adds the synthetic data of global generator from the server to realize local model adaptation. Experiments on a variety of real-world datasets demonstrate the superiority of our method.
What problem does this paper attempt to address?