A federated learning approach for classifying chest diseases from chest X-ray images

被引:1
作者
Satapathy, Sandeep Kumar [1 ]
Cho, Sung-Bae [1 ]
Mishra, Shruti [2 ]
Sah, Sweeti [3 ]
Kamaldeep [4 ]
Mohanty, Sachi Nandan [5 ]
机构
[1] Yonsei Univ, Dept Comp Sci, 50 Yonsei Ro, Seoul 03722, South Korea
[2] Vellore Inst Technol, Ctr Adv Data Sci, Chennai 600127, Tamil Nadu, India
[3] Natl Inst Technol, Dept Comp Engn, Kurukshetra 136119, Haryana, India
[4] Natl Inst Tech Teachers Training & Res, Dept Media Engn, Chandigarh 160019, India
[5] VIT AP Univ, Sch Comp Sci & Engn SCOPE, Amaravati 522237, Andhra Pradesh, India
关键词
Federated Learning; Classification; X; -Ray; Chest Disease; Image Classification; FRAMEWORK;
D O I
10.1016/j.bspc.2024.107107
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
A leading global cause of death and morbidity is chest disorders which demand accurate and timely diagnosis for effective management. Chest X-rays serve as a commonly used and widely employed diagnostic tool for detecting these diseases. The evolution of deep learning algorithms in modern years has been essential for early diagnosis and treatment planning by significantly improving the accuracy of classifying chest diseases from chest X-ray images. However, collecting a huge amount of chest X-ray data from multiple healthcare institutions for model training can be challenging due to patient privacy concerns and data regulation laws. Federated learning unfolded as a promising finding to this problem that enables several healthcare institutions to conjointly train a model in balance with patient data decentralized and private. In this study, we propose an approach of federated learning for classifying chest diseases from chest X-ray images employing the NIH Chest X-ray dataset. Our approach involves training the model on the data distributed over multiple healthcare institutions, and aggregating the model weights to obtain a global model. We assess and compare the suggested method with conventional centralized learning techniques using a large-scale chest X-ray dataset. Our outcome shows that federated learning achieves a training and validation accuracy of 80% and 69.86% respectively, while preserving patient privacy and security.
引用
收藏
页数:12
相关论文
共 27 条
[1]   A State-of-the-Art Survey on Deep Learning Theory and Architectures [J].
Alom, Md Zahangir ;
Taha, Tarek M. ;
Yakopcic, Chris ;
Westberg, Stefan ;
Sidike, Paheding ;
Nasrin, Mst Shamima ;
Hasan, Mahmudul ;
Van Essen, Brian C. ;
Awwal, Abdul A. S. ;
Asari, Vijayan K. .
ELECTRONICS, 2019, 8 (03)
[2]  
Bonawitz K., 2019, P MACH LEARN SYST, DOI 10.48550/arXiv.1902.01046
[3]   Efficient and Privacy-Enhanced Federated Learning for Industrial Artificial Intelligence [J].
Hao, Meng ;
Li, Hongwei ;
Luo, Xizhao ;
Xu, Guowen ;
Yang, Haomiao ;
Liu, Sen .
IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2020, 16 (10) :6532-6542
[4]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[5]   Densely Connected Convolutional Networks [J].
Huang, Gao ;
Liu, Zhuang ;
van der Maaten, Laurens ;
Weinberger, Kilian Q. .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :2261-2269
[6]   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
[7]   Reliable Federated Learning for Mobile Networks [J].
Kang, Jiawen ;
Xiong, Zehui ;
Niyato, Dusit ;
Zou, Yuze ;
Zhang, Yang ;
Guizani, Mohsen .
IEEE WIRELESS COMMUNICATIONS, 2020, 27 (02) :72-80
[8]   ImageNet Classification with Deep Convolutional Neural Networks [J].
Krizhevsky, Alex ;
Sutskever, Ilya ;
Hinton, Geoffrey E. .
COMMUNICATIONS OF THE ACM, 2017, 60 (06) :84-90
[9]   Blockchain for federated learning toward secure distributed machine learning systems: a systemic survey [J].
Li, Dun ;
Han, Dezhi ;
Weng, Tien-Hsiung ;
Zheng, Zibin ;
Li, Hongzhi ;
Liu, Han ;
Castiglione, Arcangelo ;
Li, Kuan-Ching .
SOFT COMPUTING, 2022, 26 (09) :4423-4440
[10]   A review of applications in federated learning [J].
Li, Li ;
Fan, Yuxi ;
Tse, Mike ;
Lin, Kuo-Yi .
COMPUTERS & INDUSTRIAL ENGINEERING, 2020, 149