FedDK: Improving Cyclic Knowledge Distillation for Personalized Healthcare Federated Learning

被引:4
作者
Xu, Yikai [1 ]
Fan, Hongbo [2 ]
机构
[1] Kunming Univ Sci & Technol, Sch Informat Engn & Automat, Kunming 650500, Peoples R China
[2] Kunming Univ Sci & Technol, Sch Fac Modern Agr Engn, Kunming 650500, Peoples R China
来源
IEEE ACCESS | 2023年 / 11卷
关键词
Federated learning; knowledge distillation; personalization; transfer learning; healthcare;
D O I
10.1109/ACCESS.2023.3294812
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
For most healthcare organizations, a significant challenge today is predicting diseases with incomplete data information, often resulting in isolation. Federated learning (FL) solves the issue of data silos by enabling remote local machines to train a globally optimal model collaboratively without the need for sharing data. In this research, we present FedDK, a serverless framework designed to obtain personalized models for each federation through data from local federations using convolutional neural networks and training through FL. Our approach involves using convolutional neural networks (CNNs) to accumulate common knowledge and transfer it using knowledge distillation, which helps prevent common knowledge forgetting. Additionally, the missing common knowledge is filled circularly between each federation, culminating in a personalized model for each group. This novel design leverages federated, deep, and integrated learning methods to produce more accurate machine-learning models. Our federated model exhibits superior performance to local and baseline FL methods, achieving significant advantages.
引用
收藏
页码:72409 / 72417
页数:9
相关论文
共 34 条
[11]   Deep learning [J].
LeCun, Yann ;
Bengio, Yoshua ;
Hinton, Geoffrey .
NATURE, 2015, 521 (7553) :436-444
[12]  
Lee GH, 2022, Arxiv, DOI arXiv:2106.03097
[13]  
Li QB, 2021, Arxiv, DOI arXiv:2010.01017
[14]   Federated Learning on Non-IID Data Silos: An Experimental Study [J].
Li, Qinbin ;
Diao, Yiqun ;
Chen, Quan ;
He, Bing Sheng .
2022 IEEE 38TH INTERNATIONAL CONFERENCE ON DATA ENGINEERING (ICDE 2022), 2022, :965-978
[15]  
Li T., 2020, P MACHINE LEARNING S, V2, P429
[16]  
Li X, 2020, Arxiv, DOI arXiv:1907.02189
[17]  
Li XX, 2021, Arxiv, DOI arXiv:2102.07623
[18]  
Lin T., 2020, ADV NEURAL INFORM PR, V33, P2351
[19]  
McCloskey M., 1989, Psychology of Learning and Motivation, V24, P109, DOI [DOI 10.1016/S0079-7421(08)60536-8, 10.1016/S0079-7421(08)60536-8]
[20]  
McMahan HB, 2017, PR MACH LEARN RES, V54, P1273