Abstract:As a promising distributed machine learning paradigm, federated learning (FL) trains a central model with decentralized data without compromising user privacy, which makes it widely used by Artificial Intelligence Internet of Things (AIoT) applications. However, the traditional FL suffers from model inaccuracy, since it trains local models only using hard labels of data while useful information of incorrect predictions with small probabilities is ignored. Although various solutions try to tackle the bottleneck of the traditional FL, most of them introduce significant communication overhead, making the deployment of large-scale AIoT devices a great challenge. To address the above problem, this article presents a novel distillation-based FL (DFL) method that enables efficient and accurate FL for AIoT applications. By using knowledge distillation (KD), in each round of FL training, our approach uploads both the soft targets and local model gradients to the cloud server for aggregation, where the aggregation results are then dispatched to AIoT devices for the next round of local training. During the DFL local training, in addition to hard labels, the model predictions approximate soft targets, which can improve model accuracy by leveraging the knowledge of soft targets. To further improve our DFL model performance, we design a dynamic adjustment strategy of loss function weights for tuning the ratio of KD and FL, which can maximize the synergy between soft targets and hard labels. Comprehensive experimental results on well-known benchmarks show that our approach can significantly improve the model accuracy of FL without introducing significant communication overhead.

Knowledge Distillation in Federated Learning: a Survey on Long Lasting Challenges and New Solutions

When Federated Learning Meets Knowledge Distillation

Federated Distillation: A Survey

Personalized and privacy-enhanced federated learning framework via knowledge distillation

A Generalized Look at Federated Learning: Survey and Perspectives

Communication-efficient federated learning via knowledge distillation

Knowledge Distillation in Federated Edge Learning: A Survey

Knowledge Selection and Local Updating Optimization for Federated Knowledge Distillation with Heterogeneous Models.

Federated Learning with New Knowledge: Fundamentals, Advances, and Futures

Digital Twin-Assisted Knowledge Distillation Framework for Heterogeneous Federated Learning

Efficient Federated Learning for AIoT Applications Using Knowledge Distillation

Towards Data-Independent Knowledge Transfer in Model-Heterogeneous Federated Learning

Decentralized Federated Learning Via Mutual Knowledge Distillation.

HYDRA-FL: Hybrid Knowledge Distillation for Robust and Accurate Federated Learning

Continual Federated Learning Based on Knowledge Distillation

A Prototype-Based Knowledge Distillation Framework for Heterogeneous Federated Learning

Data-Free Knowledge Filtering and Distillation in Federated Learning

The Recognition of Remote Sensing Image Based on Federated Knowledge Distillation.

Data-Free Knowledge Distillation for Heterogeneous Federated Learning

Handling Data Heterogeneity in Federated Learning via Knowledge Distillation and Fusion