Automated detection of COVID-19 and pneumonia diseases using data mining and transfer learning algorithms with focal loss from chest X-ray images

Abstract

Pneumonia is a lung infection that produces severe inflammation, which can be caused by viruses, bacteria, or fungi. The global occurrence of pneumonia cases has been notably affected by the outbreak of Coronavirus Disease 2019 (COVID-19). Recent studies have demonstrated the capability of Deep Learning (DL) algorithms in detecting COVID-19 and pneumonia in medical images like X-rays. In this study, we employ four different Chest X-ray (CXR) datasets to evaluate four pre-trained Convolutional Neural Network (CNN) models: InceptionResNet V2, Inception V3, Xception, and MobileNet. These models were fine-tuned using either the Cross-Entropy (CE) loss function only for balanced datasets or separately using both the CE and Focal Loss (FL) functions for imbalanced datasets. For the first dataset, the InceptionResNet V2 model achieved the highest multi-classification accuracy of 88.63%. The superior model for the second and fourth datasets is Inception V3, with 94.35% and 97.67% multi-classification accuracy, respectively. For the third dataset, the best model is Xception, with a binary classification accuracy of 100.00%. Our results emphasize the significance of using the FL function in solving class imbalance problems in DL models. Additionally, it highlights the effectiveness of individual DL models in detecting different pneumonia infections using different CXR datasets.