Two-stage deep learning model for automate detection and classification of lung diseases
Soft Computing(2023)
Abstract
Around the world, lung disease is a prevalent cause of death and illness. In this article, we propose a lung disease detection system for the automated identification of critical lung diseases: 1. Tuberculosis 2. Regular Pneumonia 3. COVID Pneumonia and healthy individuals from chest X-ray (CXR) images. The proposed system incorporates a two-stage classification method using an ensemble of convolutional neural networks (CNNs). We evaluated the performance of the proposed system using various evaluation metrics with a test set containing 200 normal, 100 Tuberculosis, 200 Regular Pneumonia, and 100 COVID Pneumonia (a total of 600) CXR images. The proposed system achieves an average macro precision, recall, and F1-score of 0.99 and an accuracy of 0.99. The performance of the proposed system based on F1-Score (0.99) for three classes, namely Tuberculosis, Pneumonia, and COVID, is better than the following existing classification frameworks: ensemble of K-ResNets for the same three classes (F1-Score: 0.84), modified ResNet50 (F1-Score:0.89), KNN classifier with fractional multi-channel exponent moments (F1-Score: 0.89), VGG-16 (F1-Score: 0.89), and convolutional sparse support estimator network (F1-Score for COVID: 0.78). The proposed system can also determine the confidence associated with the challenging classification of Regular Vs. COVID Pneumonia. We also utilized visualization techniques like Grad-CAM and Saliency maps to analyze the features learned by the CNNs, to create transparency in the prediction.
MoreTranslated text
Key words
Lung disease,Deep learning,COVID-19,Pneumonia,X-ray,Computer aided diagnosis
AI Read Science
Must-Reading Tree
Example
![](https://originalfileserver.aminer.cn/sys/aminer/pubs/mrt_preview.jpeg)
Generate MRT to find the research sequence of this paper
Chat Paper
Summary is being generated by the instructions you defined