Transfer Learning with Deep Convolutional Neural Network (CNN) for Pneumonia Detection using Chest X-ray

Abstract

Pneumonia is a life-threatening disease, which occurs in the lungs caused by either bacterial or viral infection. It can be life-endangering if not acted upon in the right time and thus an early diagnosis of pneumonia is vital. The aim of this paper is to automatically detect bacterial and viral pneumonia using digital x-ray images. It provides a detailed report on advances made in making accurate detection of pneumonia and then presents the methodology adopted by the authors. Four different pre-trained deep Convolutional Neural Network (CNN)- AlexNet, ResNet18, DenseNet201, and SqueezeNet were used for transfer learning. 5247 Bacterial, viral and normal chest x-rays images underwent preprocessing techniques and the modified images were trained for the transfer learning based classification task. In this work, the authors have reported three schemes of classifications: normal vs pneumonia, bacterial vs viral pneumonia and normal, bacterial and viral pneumonia. The classification accuracy of normal and pneumonia images, bacterial and viral pneumonia images, and normal, bacterial and viral pneumonia were 98%, 95%, and 93.3% respectively. This is the highest accuracy in any scheme than the accuracies reported in the literature. Therefore, the proposed study can be useful in faster-diagnosing pneumonia by the radiologist and can help in the fast airport screening of pneumonia patients.

13 Figures, 5 tables. arXiv admin note: text overlap with arXiv:2003.13145

References in corpus (6)

• ChestX-ray8: Hospital-scale Chest X-ray Database and Benchmarks on Weakly-Supervised Classification and Localization of Common Thorax Diseases
• Convolutional Neural Networks for Medical Image Analysis: Full Training or Fine Tuning?
• Multi-source Transfer Learning with Convolutional Neural Networks for Lung Pattern Analysis
• Learning Activation Functions: A new paradigm for understanding Neural Networks
• Chester: A Web Delivered Locally Computed Chest X-Ray Disease Prediction System
• Fast Feature Extraction with CNNs with Pooling Layers

Cited by in corpus (8)

• Can AI help in screening Viral and COVID-19 pneumonia?
• Reliable Tuberculosis Detection using Chest X-ray with Deep Learning, Segmentation and Visualization
• Pneumonia Detection on chest X-ray images Using Ensemble of Deep Convolutional Neural Networks
• COV-ECGNET: COVID-19 detection using ECG trace images with deep convolutional neural network
• QUCoughScope: An Artificially Intelligent Mobile Application to Detect Asymptomatic COVID-19 Patients using Cough and Breathing Sounds
• An ensemble-based approach by fine-tuning the deep transfer learning models to classify pneumonia from chest X-ray images
• Exploring the Effect of Image Enhancement Techniques on COVID-19 Detection using Chest X-rays Images
• Deep Learning-based Anomaly Detection on X-ray Images of Fuel Cell Electrodes