COVID19XrayNet: A Two-Step Transfer Learning Model for the COVID-19 Detecting Problem Based on a Limited Number of Chest X-Ray Images

Interdiscip Sci. 2020 Dec;12(4):555-565. doi: 10.1007/s12539-020-00393-5. Epub 2020 Sep 21.

Abstract

The novel coronavirus severe acute respiratory syndrome coronavirus 2 (SARS-CoV-2) has caused a major pandemic outbreak recently. Various diagnostic technologies have been under active development. The novel coronavirus disease (COVID-19) may induce pulmonary failures, and chest X-ray imaging becomes one of the major confirmed diagnostic technologies. The very limited number of publicly available samples has rendered the training of the deep neural networks unstable and inaccurate. This study proposed a two-step transfer learning pipeline and a deep residual network framework COVID19XrayNet for the COVID-19 detection problem based on chest X-ray images. COVID19XrayNet firstly tunes the transferred model on a large dataset of chest X-ray images, which is further tuned using a small dataset of annotated chest X-ray images. The final model achieved 0.9108 accuracy. The experimental data also suggested that the model may be improved with more training samples being released. COVID19XrayNet, a two-step transfer learning framework designed for biomedical images.

Keywords: COVID19XrayNet; Feature extraction layer (FEL); Feature smoothing layer (FSL); ResNet34; Two-step transfer learning.

MeSH terms

  • Algorithms
  • Betacoronavirus
  • COVID-19
  • COVID-19 Testing
  • Clinical Laboratory Techniques / methods*
  • Coronavirus
  • Coronavirus Infections / complications
  • Coronavirus Infections / diagnosis*
  • Coronavirus Infections / diagnostic imaging
  • Coronavirus Infections / virology
  • Databases, Factual
  • Datasets as Topic
  • Deep Learning*
  • Humans
  • Lung / diagnostic imaging*
  • Machine Learning
  • Models, Biological*
  • Neural Networks, Computer*
  • Pandemics
  • Pneumonia / diagnosis
  • Pneumonia / diagnostic imaging
  • Pneumonia / etiology
  • Pneumonia / virology
  • Pneumonia, Viral / complications
  • Pneumonia, Viral / diagnosis*
  • Pneumonia, Viral / diagnostic imaging
  • Pneumonia, Viral / virology
  • Radiography / methods
  • Reference Values
  • SARS-CoV-2
  • Tomography, X-Ray Computed / methods
  • X-Rays*