In the field of Federated Learning (FL), the differences in client-side data distribution pose a challenge of poor generalization of local models. Employing knowledge distillation methods allows the transmission of complex knowledge, referred to as dark knowledge, to client models. The global features within dark knowledge can assist clients in learning out-of-distribution knowledge, thereby enhancing the generalization ability of the federated system. To tackle issues caused by the difference in data distribution, we propose a Federated Knowledge Distillation approach that leverages a prompt generator to match data distribution in FL. Firstly, we develop a personalized visual prompt generator for image classification tasks. By implicitly handling the client’s local data and training local models, this generator can effectively release data problems in distributed learning. Secondly, we utilize knowledge distillation to leverage the server’s prediction probabilities, aiding clients in acquiring global out-of-self-distribution knowledge. Experiment results on three datasets with different data distributions across different sites demonstrate a significant improvement in the classification accuracy of FL models using this approach.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Federated Knowledge Distillation Based on Prompt for Matching Data Distribution

  • Yizhang Liu,
  • Wenze Xu,
  • Zixuan Xu,
  • Tongtong Yuan,
  • Weihong Deng

摘要

In the field of Federated Learning (FL), the differences in client-side data distribution pose a challenge of poor generalization of local models. Employing knowledge distillation methods allows the transmission of complex knowledge, referred to as dark knowledge, to client models. The global features within dark knowledge can assist clients in learning out-of-distribution knowledge, thereby enhancing the generalization ability of the federated system. To tackle issues caused by the difference in data distribution, we propose a Federated Knowledge Distillation approach that leverages a prompt generator to match data distribution in FL. Firstly, we develop a personalized visual prompt generator for image classification tasks. By implicitly handling the client’s local data and training local models, this generator can effectively release data problems in distributed learning. Secondly, we utilize knowledge distillation to leverage the server’s prediction probabilities, aiding clients in acquiring global out-of-self-distribution knowledge. Experiment results on three datasets with different data distributions across different sites demonstrate a significant improvement in the classification accuracy of FL models using this approach.