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 条
[1]   The Liver Tumor Segmentation Benchmark (LiTS) [J].
Bilic, Patrick ;
Christ, Patrick ;
Li, Hongwei Bran ;
Vorontsov, Eugene ;
Ben-Cohen, Avi ;
Kaissis, Georgios ;
Szeskin, Adi ;
Jacobs, Colin ;
Mamani, Gabriel Efrain Humpire ;
Chartrand, Gabriel ;
Lohoefer, Fabian ;
Holch, Julian Walter ;
Sommer, Wieland ;
Hofmann, Felix ;
Hostettler, Alexandre ;
Lev-Cohain, Naama ;
Drozdzal, Michal ;
Amitai, Michal Marianne ;
Vivanti, Refael ;
Sosna, Jacob ;
Ezhov, Ivan ;
Sekuboyina, Anjany ;
Navarro, Fernando ;
Kofler, Florian ;
Paetzold, Johannes C. ;
Shit, Suprosanna ;
Hu, Xiaobin ;
Lipkova, Jana ;
Rempfler, Markus ;
Piraud, Marie ;
Kirschke, Jan ;
Wiestler, Benedikt ;
Zhang, Zhiheng ;
Huelsemeyer, Christian ;
Beetz, Marcel ;
Ettlinger, Florian ;
Antonelli, Michela ;
Bae, Woong ;
Bellver, Miriam ;
Bi, Lei ;
Chen, Hao ;
Chlebus, Grzegorz ;
Dam, Erik B. ;
Dou, Qi ;
Fu, Chi-Wing ;
Georgescu, Bogdan ;
Giro-I-Nieto, Xavier ;
Gruen, Felix ;
Han, Xu ;
Heng, Pheng-Ann .
MEDICAL IMAGE ANALYSIS, 2023, 84
[2]  
Chao W. L., 2020, arXiv
[3]  
Chen YQ, 2022, Arxiv, DOI arXiv:2206.08516
[4]   Federated Class-Incremental Learning [J].
Dong, Jiahua ;
Wang, Lixu ;
Fang, Zhen ;
Sun, Gan ;
Xu, Shichao ;
Wang, Xiao ;
Zhu, Qi .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2022, :10154-10163
[5]  
He Y., 2022, PROC 36 AAAI C ARTIF, VVolume 22
[6]   Learning Critically: Selective Self-Distillation in Federated Learning on Non-IID Data [J].
He, Yuting ;
Chen, Yiqiang ;
Yang, XiaoDong ;
Yu, Hanchao ;
Huang, Yi-Hua ;
Gu, Yang .
IEEE TRANSACTIONS ON BIG DATA, 2024, 10 (06) :789-800
[7]  
Hinton G, 2015, Arxiv, DOI arXiv:1503.02531
[8]   Advances and Open Problems in Federated Learning [J].
Kairouz, Peter ;
McMahan, H. Brendan ;
Avent, Brendan ;
Bellet, Aurelien ;
Bennis, Mehdi ;
Bhagoji, Arjun Nitin ;
Bonawitz, Kallista ;
Charles, Zachary ;
Cormode, Graham ;
Cummings, Rachel ;
D'Oliveira, Rafael G. L. ;
Eichner, Hubert ;
El Rouayheb, Salim ;
Evans, David ;
Gardner, Josh ;
Garrett, Zachary ;
Gascon, Adria ;
Ghazi, Badih ;
Gibbons, Phillip B. ;
Gruteser, Marco ;
Harchaoui, Zaid ;
He, Chaoyang ;
He, Lie ;
Huo, Zhouyuan ;
Hutchinson, Ben ;
Hsu, Justin ;
Jaggi, Martin ;
Javidi, Tara ;
Joshi, Gauri ;
Khodak, Mikhail ;
Konecny, Jakub ;
Korolova, Aleksandra ;
Koushanfar, Farinaz ;
Koyejo, Sanmi ;
Lepoint, Tancrede ;
Liu, Yang ;
Mittal, Prateek ;
Mohri, Mehryar ;
Nock, Richard ;
Ozgur, Ayfer ;
Pagh, Rasmus ;
Qi, Hang ;
Ramage, Daniel ;
Raskar, Ramesh ;
Raykova, Mariana ;
Song, Dawn ;
Song, Weikang ;
Stich, Sebastian U. ;
Sun, Ziteng ;
Suresh, Ananda Theertha .
FOUNDATIONS AND TRENDS IN MACHINE LEARNING, 2021, 14 (1-2) :1-210
[9]   Overcoming catastrophic forgetting in neural networks [J].
Kirkpatricka, James ;
Pascanu, Razvan ;
Rabinowitz, Neil ;
Veness, Joel ;
Desjardins, Guillaume ;
Rusu, Andrei A. ;
Milan, Kieran ;
Quan, John ;
Ramalho, Tiago ;
Grabska-Barwinska, Agnieszka ;
Hassabis, Demis ;
Clopath, Claudia ;
Kumaran, Dharshan ;
Hadsell, Raia .
PROCEEDINGS OF THE NATIONAL ACADEMY OF SCIENCES OF THE UNITED STATES OF AMERICA, 2017, 114 (13) :3521-3526
[10]   Gradient-based learning applied to document recognition [J].
Lecun, Y ;
Bottou, L ;
Bengio, Y ;
Haffner, P .
PROCEEDINGS OF THE IEEE, 1998, 86 (11) :2278-2324