Multi-Channel Transfer Learning of Chest X-ray Images for Screening of COVID-19

2020/05/12 by Sampa Misra, Misra, Sampa, Seungwan Jeon +6 · 1 citation
Computer Science · Medicine · #AI in cancer detection #COVID-19 diagnosis using AI #Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences #FOS: Electrical engineering #Image and Video Processing (eess.IV) #Machine Learning (cs.LG) #Radiomics and Machine Learning in Medical Imaging #electronic engineering #information engineering

paper · pdf · doi:10.48550/arxiv.2005.05576

openalex publication_date 2020/05/12 · openalex created_date 2020/05/21 · openalex updated_date 2026/07/28

Abstract

The 2019 novel coronavirus (COVID-19) has spread rapidly all over the world and it is affecting the whole society. The current gold standard test for screening COVID-19 patients is the polymerase chain reaction test. However, the COVID-19 test kits are not widely available and time-consuming. Thus, as an alternative, chest X-rays are being considered for quick screening. Since the presentation of COVID-19 in chest X-rays is varied in features and specialization in reading COVID-19 chest X-rays are required thus limiting its use for diagnosis. To address this challenge of reading chest X-rays by radiologists quickly, we present a multi-channel transfer learning model based on ResNet architecture to facilitate the diagnosis of COVID-19 chest X-ray. Three ResNet-based models (Models a, b, and c) were retrained using Dataset_A (1579 normal and 4429 diseased), Dataset_B (4245 pneumonia and 1763 non-pneumonia), and Dataset_C (184 COVID-19 and 5824 Non-COVID19), respectively, to classify (a) normal or diseased, (b) pneumonia or non-pneumonia, and (c) COVID-19 or non-COVID19. Finally, these three models were ensembled and fine-tuned using Dataset_D (1579 normal, 4245 pneumonia, and 184 COVID-19) to classify normal, pneumonia, and COVID-19 cases. Our results show that the ensemble model is more accurate than the single ResNet model, which is also re-trained using Dataset_D as it extracts more relevant semantic features for each class. Our approach provides a precision of 94 % and a recall of 100%. Thus, our method could potentially help clinicians in screening patients for COVID-19, thus facilitating immediate triaging and treatment for better outcomes.

Citations

• A Survey on Transfer Learning
• Deep Convolutional Neural Networks for Image Classification: A Comprehensive Review
• Pathological findings of COVID-19 associated with acute respiratory distress syndrome
• COVID-19 Image Data Collection
• COVID-ResNet: A Deep Learning Framework for Screening of COVID19 from Radiographs

Cited by

• AquaSignal: An Integrated Framework for Robust Underwater Acoustic Analysis

Related

• COVID-19 Detection Based on Self-Supervised Transfer Learning Using Chest X-Ray Images
• COVID-ResNet: A Deep Learning Framework for Screening of COVID19 from Radiographs
• Advancing Diagnostic Precision: Leveraging Machine Learning Techniques for Accurate Detection of Covid-19, Pneumonia, and Tuberculosis in Chest X-Ray Images
• Deep-COVID: Predicting COVID-19 From Chest X-Ray Images Using Deep\n Transfer Learning
• Empowering COVID-19 Detection: Optimizing Performance Through Fine-Tuned EfficientNet Deep Learning Architecture