DMFL_Net: A Federated Learning-Based Framework for the Classification of COVID-19 from Multiple Chest Diseases Using X-rays

• Published in: Sensors (Basel, Switzerland) Vol. 23; no. 2; p. 743
• Main Authors: Malik, Hassaan, Naeem, Ahmad, Naqvi, Rizwan Ali, Loh, Woong-Kee
• Format: Journal Article
• Language: English
• Published: Switzerland MDPI AG 09-01-2023; MDPI
• Subjects: Algorithms; Artificial intelligence; Classification; Communication; Confidentiality; Coronaviruses; COVID-19; COVID-19 - diagnostic imaging; data privacy; Datasets; Deep learning; Disease transmission; Energy consumption; federated learning; Hospitals; Humans; Image analysis; Lung cancer; Lung Neoplasms; machine learning; Medical research; Model accuracy; Pneumothorax; Privacy; Public health; Radiography; Servers; Thorax - diagnostic imaging; X-Rays; COVID-19; lung cancer; data privacy; machine learning; TB; federated learning
• ISSN: 1424-8220; 1424-8220
• DOI: 10.3390/s23020743

Coronavirus Disease 2019 (COVID-19) is still a threat to global health and safety, and it is anticipated that deep learning (DL) will be the most effective way of detecting COVID-19 and other chest diseases such as lung cancer (LC), tuberculosis (TB), pneumothorax (PneuTh), and pneumonia (Pneu). However, data sharing across hospitals is hampered by patients' right to privacy, leading to unexpected results from deep neural network (DNN) models. Federated learning (FL) is a game-changing concept since it allows clients to train models together without sharing their source data with anybody else. Few studies, however, focus on improving the model's accuracy and stability, whereas most existing FL-based COVID-19 detection techniques aim to maximize secondary objectives such as latency, energy usage, and privacy. In this work, we design a novel model named decision-making-based federated learning network (DMFL_Net) for medical diagnostic image analysis to distinguish COVID-19 from four distinct chest disorders including LC, TB, PneuTh, and Pneu. The DMFL_Net model that has been suggested gathers data from a variety of hospitals, constructs the model using the DenseNet-169, and produces accurate predictions from information that is kept secure and only released to authorized individuals. Extensive experiments were carried out with chest X-rays (CXR), and the performance of the proposed model was compared with two transfer learning (TL) models, i.e., VGG-19 and VGG-16 in terms of accuracy (ACC), precision (PRE), recall (REC), specificity (SPF), and F1-measure. Additionally, the DMFL_Net model is also compared with the default FL configurations. The proposed DMFL_Net + DenseNet-169 model achieves an accuracy of 98.45% and outperforms other approaches in classifying COVID-19 from four chest diseases and successfully protects the privacy of the data among diverse clients.