Metadata and Image Features Co-Aware Personalized Federated Learning for Smart Healthcare

被引:11
作者
Jin, Tong [1 ]
Pan, Shujia [1 ]
Li, Xue [2 ]
Chen, Siguang [1 ]
机构
[1] Nanjing Univ Posts & Telecommun, Sch Internet Things, Nanjing 210000, Peoples R China
[2] Nanjing Med Univ, Womens Hosp, Nanjing Matern & Child Hlth Care Hosp, Dept Dermatol, Nanjing 210000, Peoples R China
基金
中国国家自然科学基金; 中国博士后科学基金;
关键词
Smart healthcare; federated learning; metadata; intelligent diagnosis;
D O I
10.1109/JBHI.2023.3279096
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Recently, artificial intelligence has been widely used in intelligent disease diagnosis and has achieved great success. However, most of the works mainly rely on the extraction of image features but ignore the use of clinical text information of patients, which may limit the diagnosis accuracy fundamentally. In this paper, we propose a metadata and image features co-aware personalized federated learning scheme for smart healthcare. Specifically, we construct an intelligent diagnosis model, by which users can obtain fast and accurate diagnosis services. Meanwhile, a personalized federated learning scheme is designed to utilize the knowledge learned from other edge nodes with larger contributions and customize high-quality personalized classification models for each edge node. Subsequently, a Naive Bayes classifier is devised for classifying patient metadata. And then the image and metadata diagnosis results are jointly aggregated by different weights to improve the accuracy of intelligent diagnosis. Finally, the simulation results illustrate that, compared with the existing methods, our proposed algorithm achieves better classification accuracy, reaching about 97.16% on PAD-UFES-20 dataset.
引用
收藏
页码:4110 / 4119
页数:10
相关论文
共 32 条
[31]   SU-Net: An Efficient Encoder-Decoder Model of Federated Learning for Brain Tumor Segmentation [J].
Yi, Liping ;
Zhang, Jinsong ;
Zhang, Rui ;
Shi, Jiaqi ;
Wang, Gang ;
Liu, Xiaoguang .
ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING, ICANN 2020, PT I, 2020, 12396 :761-773
[32]   Federated Learning With Non-IID Data in Wireless Networks [J].
Zhao, Zhongyuan ;
Feng, Chenyuan ;
Hong, Wei ;
Jiang, Jiamo ;
Jia, Chao ;
Quek, Tony Q. S. ;
Peng, Mugen .
IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS, 2022, 21 (03) :1927-1942