Two-stage deep learning model for automate detection and classification of lung diseases
Authors: 
M. Ganeshkumar, Vinayakumar Ravi, V. Sowmya, E. A. Gopalakrishnan, K. P. Soman, M. RupeshkumarAuthors Info & Claims
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.
References
[1]
Andreu J, Caceres J, Pallisa E, and Martinez-Rodriguez M Radiological manifestations of pulmonary tuberculosis Eur J Radiol 2004 51 2 139-149
[2]
Anupama MA, Sowmya V, Soman KP (2019). Breast cancer classification using capsule network with preprocessed histology images. In: 2019 International conference on communication and signal processing (ICCSP) (p. 0143-0147). IEEE
[3]
Bekhet S, Hassaballah M, Kenk MA, Hameed MA (2020). An artificial intelligence based technique for COVID-19 diagnosis from chest X-Ray. In: 2020 2nd Novel Intelligent and Leading Emerging Sciences Conference (NILES) (p. 191-195). IEEE
[4]
Bermejo-Peláez D, Ash SY, Washko GR, et al. Classification of Interstitial Lung Abnormality Patterns with an Ensemble of Deep Convolutional Neural Networks Sci Rep 2020 10 338
[5]
Bhosale YH, Patnaik KS (2022). IoT deployable lightweight deep learning application for COVID-19 detection with lung diseases using RaspberryPi. In: 2022 International Conference on IoT and Blockchain Technology (ICIBT) (pp. 1-6). IEEE
[6]
Bhosale YH, Patnaik KS (2022). Application of Deep Learning Techniques in Diagnosis of Covid-19 (Coronavirus): a Systematic Review. Neural Process Lett p. 1-53
[7]
Bhosale YH, Patnaik KS (2022). PulDi-COVID: Chronic Obstructive Pulmonary (Lung) Diseases With COVID-19 Classification Using Ensemble Deep Convolutional Neural Network From Chest X-Ray Images To Minimize Severity And Mortality Rates. Biomedical Signal Processing and Control, p. 104445
[8]
Bhosale YH, Zanwar S, Ahmed Z, Nakrani M, Bhuyar D, Shinde U (2022) Deep Convolutional Neural Network Based Covid-19 Classification From Radiology X-Ray Images For IoT Enabled Devices. In: 2022 8th International Conference on Advanced Computing and Communication Systems (ICACCS) (Vol. 1, p. 1398-1402). IEEE
[9]
Brunese L, Mercaldo F, Reginelli A, and Santone A Explainable deep learning for pulmonary disease and coronavirus COVID-19 detection from X-rays Comput Methods Programs Biomed 2020 196
[10]
Chen KC, Yu HR, Chen WS, et al. Diagnosis of common pulmonary diseases in children by X-ray images and deep learning Sci Rep 2020 10 17374
[11]
Chollet F (2017). Xception: Deep learning with depthwise separable convolutions. In: Proceedings of the IEEE conference on computer vision and pattern recognition (p. 1251-1258)
[12]
Cleverley J, Piper J, and Jones MM The role of chest radiography in confirming covid-19 pneumonia BMJ 2020 370
[13]
Cohen JP, Morrison P, Dao L, Roth K, Duong TQ, Ghassemi M (2020). Covid-19 image data collection: Prospective predictions are the future. arXiv preprint arXiv:2006.11988
[14]
Das D and Howlett DC Chest X-ray manifestations of pneumonia Surgery-Oxford Int Ed 2009 27 10 453-455
[15]
Elaziz MA, Hosny KM, Salah A, Darwish MM, Lu S, and Sahlol AT New machine learning method for image-based diagnosis of COVID-19 PLoS One 2020 15 6
[16]
Etlay JP, Kapoor WN, and Fine MJ Does this patient have community-acquired pneumonia?: Diagnosing pneumonia by history and physical examination JAMA 1997 278 17 1440-1445
[17]
Fridadar M, Amer R, Gozes O, Nassar J, Greenspan H (2021). COVID-19 in CXR: from Detection and Severity Scoring to Patient Disease Monitoring. IEEE J Biomed Health Informat
[18]
Han Z, Wei B, Hong Y, Li T, Cong J, Zhu X, Wei H, and Zhang W Accurate screening of COVID-19 using attention-based deep 3D multiple instance learning IEEE Trans Med Imag 2020 39 8 2584-2594
[19]
Hassaballah M and Awad AI Deep learning in computer vision: principles and applications 2020 USA CRC Press
[20]
He K, Zhang X, Ren S, Sun J (2016). Deep residual learning for image recognition. In: Proceedings of the IEEE conference on computer vision and pattern recognition (pp. 770-778)
[21]
Huang G, Liu Z, Van Der Maaten L, Weinberger KQ (2017) Densely connected convolutional networks. In: Proceedings of the IEEE conference on computer vision and pattern recognition (p. 4700-4708)
[22]
Infante M, Lutman RF, Imparato S, Di Rocco M, Ceresoli GL, Torri V, Morenghi E, Minuti F, Cavuto S, Bottoni E, and Inzirillo F Differential diagnosis and management of focal ground-glass opacities Eur Respirat J 2009 33 4 821-827
[23]
Koch G, Zemel R, Salakhutdinov R (2015). Siamese neural networks for one-shot image recognition. In: ICML deep learning workshop (Vol. 2)
[24]
Kusuma S and Udayan D Machine learning and deep learning methods in heart disease (HD) research Int J Pure Appl Math 2018 119 1483-1496
[25]
Li MD, Chang K, Bearce B, Chang CY, Huang AJ, Campbell JP, Brown JM, Singh P, Hoebel KV, Erdoğmuş D, and Ioannidis S Siamese neural networks for continuous disease severity evaluation and change detection in medical imaging NPJ Digit Med 2020 3 1 1-9
[26]
Liu Y, Wu YH, Ban Y, Wang H, Cheng MM (2020). Rethinking computer-aided tuberculosis diagnosis. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (p. 2646-2655)
[27]
Oh Y, Park S, and Ye JC Deep learning covid-19 features on cxr using limited training data sets IEEE Trans Med Imag 2020 39 8 2688-2700
[28]
Sabour S, Frosst N, Hinton GE (2017). Dynamic routing between capsules. arXiv preprint arXiv:1710.09829
[29]
Selvaraju RR, Cogswell M, Das A, Vedantam R, Parikh D, Batra D (2017) Grad-cam: Visual explanations from deep networks via gradient-based localization. In: Proceedings of the IEEE international conference on computer vision (p. 618-626)
[30]
Simonyan K, Vedaldi A, Zisserman A (2013). Deep inside convolutional networks: visualising image classification models and saliency maps. arXiv preprint arXiv:1312.6034
[31]
Simonyan K, Zisserman A (2014). Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:1409.1556
[32]
Swapna G, Vinayakumar R, Soman KP (2018) Diabetes detection using deep learning algorithms. ICT express 4(4):243–246
[33]
Szegedy C, Liu W, Jia Y, Sermanet P, Reed S, Anguelov D, Erhan D, Vanhoucke V, Rabinovich A (2015). Going deeper with convolutions. In: Proceedings of the IEEE conference on computer vision and pattern recognition (p. 1-9)
[34]
Tan M, Le Q (2019) Efficientnet: Rethinking model scaling for convolutional neural networks. In: International Conference on Machine Learning (p. 6105-6114). PMLR
[35]
Thomas E, Delabat S, Carattini YL, and Andrews DM SARS-CoV-2 and variant diagnostic testing approaches in the United States Viruses 2021 13 12 2492
[36]
Xu S, Wu H, and Bie R CXNet-m1: Anomaly detection on chest X-rays with image-based deep learning IEEE Access 2018 7 4466-4477
[37]
Yadav P, Menon N, Ravi V, Vishvanathan S (2021). Lung-gans: Unsupervised representation learning for lung disease classification using chest ct and x-ray images. IEEE Trans Eng Manag
[38]
Yamaç M, Ahishali M, Degerli A, Kiranyaz S, Chowdhury ME, Gabbouj M (2021). Convolutional Sparse Support Estimator-Based COVID-19 Recognition From X-Ray Images. IEEE Trans Neural Netw Learn Syst
Recommendations
A multichannel EfficientNet deep learning-based stacking ensemble approach for lung disease detection using chest X-ray images
AbstractThis paper proposes a multichannel deep learning approach for lung disease detection using chest X-rays. The multichannel models used in this work are EfficientNetB0, EfficientNetB1, and EfficientNetB2 pretrained models. The features from ...
Computer-assisted lung diseases detection from pediatric chest radiography using long short-term memory networks
AbstractPneumonia is one of the most common lung diseases for children. Computer-aided diagnosis (CAD) can be utilized to achieve automated lung disease detection using artificial intelligence techniques. This paper proposes a hybrid deep ...
Graphical abstractDisplay Omitted
Convolutional neural networks applied in the detection of pneumonia by X-ray images
According to the World Health Organization (WHO), pneumonia kills about 2 million children under the age of 5 and is constantly estimated as the leading cause of child mortality, killing more children than AIDS, malaria, and measles together. The ...
Comments
Please enable JavaScript to view thecomments powered by Disqus.Information & Contributors
Information
Published In
© The Author(s), under exclusive licence to Springer-Verlag GmbH Germany, part of Springer Nature 2023. Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
Publisher
Springer-Verlag
Berlin, Heidelberg
Publication History
Published: 07 September 2023
Accepted: 21 August 2023
Author Tags
Qualifiers
- Research-article
Contributors
Other Metrics
Bibliometrics & Citations
Bibliometrics
Article Metrics
- 0Total Citations
- 0Total Downloads
- Downloads (Last 12 months)0
- Downloads (Last 6 weeks)0
Reflects downloads up to 30 Nov 2024
Other Metrics
Citations
View Options
View options
Login options
Check if you have access through your login credentials or your institution to get full access on this article.
Sign in