Abstract:Federated learning is a distributed machine learning framework. It facilitates collaborative modeling among participants without sharing raw user data, providing a feasible solution to address data silos in a secure and privacy-preserving manner. However, data heterogeneity is a major challenge of federated learning as it greatly impacts both the convergence speed during model training and the accuracy of predictions. To address this issue, recent federated learning algorithms have incorporated knowledge distillation as a approach of information sharing. Nevertheless, the majority of current approaches only employ logits averaging to combine participants' knowledge, which can have a detrimental impact on the accuracy of the global model, particularly when certain local models show poor performance. besides, some methods rely on public datasets, thereby compromising the privacy-preserving principle of federated learning. To address the aforementioned concerns, we introduces a new federated learning algorithm named Federated Adaptive Knowledge Distillation (FedAdKD), which utilizes knowledge distillation. After the server completes the aggregation of the global model, FedAdKD dynamically allocates distillation weights based on the individual local model's loss on the public dataset, and performs knowledge distillation on the global model. As the use of public datasets violates the privacy-preserving principle of federated learning, we also proposes the incorporation of generative models to generate data that adheres to the original image distribution. By employing federated learning on the client side to train diffusion models, data is generated that adheres to the original image distribution while maintaining privacy. This generated data is then utilized as the distillation dataset. Experimental findings validate the effectiveness of FedAdKD in addressing the obstacles presented by data heterogeneity. FedAdKD not only mitigates the decline in global model accuracy caused by subpar local models and minimizes knowledge forgetting resulting from direct model aggregation but also improves the generalization capacity of the global model.

FedTweet: Two-fold Knowledge Distillation for non-IID Federated Learning

FedDGP: Disentangling Global and Personal Models for Federated Learning

Understanding the Training Dynamics in Federated Deep Learning via Aggregation Weight Optimization

Fine-tuning Global Model Via Data-Free Knowledge Distillation for Non-IID Federated Learning

FedD2S: Personalized Data-Free Federated Knowledge Distillation

Data-free knowledge distillation via generator-free data generation for Non-IID federated learning

The Best of Both Worlds: Accurate Global and Personalized Models through Federated Learning with Data-Free Hyper-Knowledge Distillation

pFedKT: Personalized federated learning with dual knowledge transfer

FedDistill: Global Model Distillation for Local Model De-Biasing in Non-IID Federated Learning

Handling Data Heterogeneity in Federated Learning via Knowledge Distillation and Fusion

FedGKD: Towards Heterogeneous Federated Learning via Global Knowledge Distillation

Local-Global Knowledge Distillation in Heterogeneous Federated Learning with Non-IID Data

Federated learning on non-IID and long-tailed data via dual-decoupling

A Personalized Federated Learning Method Based on Clustering and Knowledge Distillation

A Prototype-Based Knowledge Distillation Framework for Heterogeneous Federated Learning

HyperFed: Hyperbolic Prototypes Exploration with Consistent Aggregation for Non-IID Data in Federated Learning

Fedadkd:heterogeneous federated learning via adaptive knowledge distillation

FedGK: Communication-Efficient Federated Learning through Group-Guided Knowledge Distillation

FedDr+: Stabilizing Dot-regression with Global Feature Distillation for Federated Learning

A Hierarchical Knowledge Transfer Framework for Heterogeneous Federated Learning.

Tailored Federated Learning: Leveraging Direction Regulation & Knowledge Distillation