FedDK: Improving Cyclic Knowledge Distillation for Personalized Healthcare Federated Learning

被引:5
作者
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
关键词
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 条
[21]  
Regulation P., 2016, Regulation, V679
[22]   Introducing a New Benchmarked Dataset for Activity Monitoring [J].
Reiss, Attila ;
Stricker, Didier .
2012 16TH INTERNATIONAL SYMPOSIUM ON WEARABLE COMPUTERS (ISWC), 2012, :108-109
[23]   FEDAUX: Leveraging Unlabeled Auxiliary Data in Federated Learning [J].
Sattler, Felix ;
Korjakow, Tim ;
Rischke, Roman ;
Samek, Wojciech .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 34 (09) :5531-5543
[24]  
Shen Y., 2022, P IEEE CVF C COMP VI, P10031
[25]   Deep learning for sensor-based activity recognition: A survey [J].
Wang, Jindong ;
Chen, Yiqiang ;
Hao, Shuji ;
Peng, Xiaohui ;
Hu, Lisha .
PATTERN RECOGNITION LETTERS, 2019, 119 :3-11
[26]  
Wu C., 2022, Nature Commun., V13
[27]   Efficient Multiple Organ Localization in CT Image Using 3D Region Proposal Network [J].
Xu, Xuanang ;
Zhou, Fugen ;
Liu, Bo ;
Fu, Dongshan ;
Bai, Xiangzhi .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2019, 38 (08) :1885-1898
[28]   MedMNIST v2-A large-scale lightweight benchmark for 2D and 3D biomedical image classification [J].
Yang, Jiancheng ;
Shi, Rui ;
Wei, Donglai ;
Liu, Zequan ;
Zhao, Lin ;
Ke, Bilian ;
Pfister, Hanspeter ;
Ni, Bingbing .
SCIENTIFIC DATA, 2023, 10 (01)
[29]   MEDMNIST CLASSIFICATION DECATHLON: A LIGHTWEIGHT AUTOML BENCHMARK FOR MEDICAL IMAGE ANALYSIS [J].
Yang, Jiancheng ;
Shi, Rui ;
Ni, Bingbing .
2021 IEEE 18TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (ISBI), 2021, :191-195
[30]   A Gift from Knowledge Distillation: Fast Optimization, Network Minimization and Transfer Learning [J].
Yim, Junho ;
Joo, Donggyu ;
Bae, Jihoon ;
Kim, Junmo .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :7130-7138