Document Type
Conference Paper
Publication Date
1-1-2020
Publication Source
Progress in Biomedical Optics and Imaging - Proceedings of SPIE
Abstract
Approximately two million pediatric deaths occur every year due to Pneumonia. Detection and diagnosis of Pneumonia plays an important role in reducing these deaths. Chest radiography is one of the most commonly used modalities to detect pneumonia. In this paper, we propose a novel two-stage deep learning architecture to detect pneumonia and classify its type in chest radiographs. This architecture contains one network to classify images as either normal or pneumonic, and another deep learning network to classify the type as either bacterial or viral. In this paper, we study and compare the performance of various stage one networks such as AlexNet, ResNet, VGG16 and Inception-v3 for detection of pneumonia. For these networks, we employ transfer learning to exploit the wealth of information available from prior training. For the second stage, we find that transfer learning with these same networks tends to overfit the data. For this reason we propose a simpler CNN architecture for classification of pneumonic chest radiographs and show that it overcomes the overfitting problem. We further enhance the performance of our system in a novel way by incorporating lung segmentation using a U-Net architecture. We make use of a publicly available dataset comprising 5856 images (1583 - Normal, 4273 - Pneumonic). Among the pneumonia patients, 2780 patients are identified as bacteria type and the rest belongs to virus category. We test our proposed algorithm(s) on a set of 624 images and we achieve an area under the receiver operating characteristic curve of 0.996 for pneumonia detection. We also achieve an accuracy of 97.8% for classification of pneumonic chest radiographs thereby setting a new benchmark for both detection and diagnosis. We believe the proposed two-stage classification of chest radiographs for pneumonia detection and its diagnosis would enhance the workflow of radiologists.
ISBN/ISSN
1605-7422
Publisher
SPIE
Keywords
Computer Aided Detection, Computer Aided Diagnosis, Convolutional Neural Networks, Lung Segmentation, Pneumonia Detection, University of Dayton Electro-optics and Photonics
eCommons Citation
Barath Narayanan Narayanan, Venkata Salini Priyamvada Davuluru, Russell C. Hardie, "Two-stage deep learning architecture for pneumonia detection and its diagnosis in chest radiographs," Proc. SPIE 11318, Medical Imaging 2020: Imaging Informatics for Healthcare, Research, and Applications, 113180G (2 March 2020); https://doi.org/10.1117/12.2547635
Comments
The document is provided in compliance with the publisher's policy on self-archiving. Permission documentation is on file. To view the paper on the publisher's website, use the DOI: https://doi.org/10.1117/12.2547635