Open Access
1 December 2021 Deep learning classification of COVID-19 in chest radiographs: performance and influence of supplemental training
Author Affiliations +
Abstract

Purpose: Accurate classification of COVID-19 in chest radiographs is invaluable to hard-hit pandemic hot spots. Transfer learning techniques for images using well-known convolutional neural networks show promise in addressing this problem. These methods can significantly benefit from supplemental training on similar conditions, considering that there currently exists no widely available chest x-ray dataset on COVID-19. We evaluate whether targeted pretraining for similar tasks in radiography labeling improves classification performance in a sample radiograph dataset containing COVID-19 cases.

Approach: We train a DenseNet121 to classify chest radiographs through six training schemes. Each training scheme is designed to incorporate cases from established datasets for general findings in chest radiography (CXR) and pneumonia, with a control scheme with no pretraining. The resulting six permutations are then trained and evaluated on a dataset of 1060 radiographs collected from 475 patients after March 2020, containing 801 images of laboratory-confirmed COVID-19 cases.

Results: Sequential training phases yielded substantial improvement in classification accuracy compared to a baseline of standard transfer learning with ImageNet parameters. The test set area under the receiver operating characteristic curve for COVID-19 classification improved from 0.757 in the control to 0.857 for the optimal training scheme in the available images.

Conclusions: We achieve COVID-19 classification accuracies comparable to previous benchmarks of pneumonia classification. Deliberate sequential training, rather than pooling datasets, is critical in training effective COVID-19 classifiers within the limitations of early datasets. These findings bring clinical-grade classification through CXR within reach for more regions impacted by COVID-19.

CC BY: © 2022 Society of Photo-Optical Instrumentation Engineers (SPIE)
Rafael B. Fricks, Francesco Ria, Hamid Chalian, Pegah Khoshpouri, Ehsan Abadi, Lorenzo Bianchi, William P. Segars, and Ehsan Samei "Deep learning classification of COVID-19 in chest radiographs: performance and influence of supplemental training," Journal of Medical Imaging 8(6), 064501 (1 December 2021). https://doi.org/10.1117/1.JMI.8.6.064501
Received: 25 June 2020; Accepted: 8 November 2021; Published: 1 December 2021
Lens.org Logo
CITATIONS
Cited by 2 scholarly publications.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
COVID 19

Education and training

Chest imaging

Radiography

Deep learning

Image classification

Performance modeling

Back to Top