Detection, Localization of Cardiomegaly and TB Disease of CXR Images using Deep Learning
Keywords:
CNN architectures, Image Classification, Image segmentation, Tuberculosis, Cardiomegaly, Disease LocalizationAbstract
INTRODUCTION: Tuberculosis (TB) continues to pose a significant worldwide public health concern, as it stands as the primary contributor to mortality stemming from infectious illnesses. Cardiomegaly, characterized by an enlarged heart, poses medical concern as well.
OBJECTIVES: Timely identification of Cardiomegaly is vital for effective management. Chest X-ray diagnosis is an easily available method with less radiation exposure to detect several lung infections and heart enlargement. Utilizing computer-aided diagnostic systems can aid in the early detection of lung conditions and the enlargement of the heart.
METHODS: We worked on different state-of-the-art CNN architectures such as VGG, DenseNet and EfficientNet with customization over dataset generated from combination of multiple publicly available datasets, which consists of 12939 annotated images across three different categories, one being normal and other two being TB and cardiomegaly diseases..
RESULTS: EfficientNetB5 with optimization has shown excellent results amongst others in classifying Tuberculosis and Cardiomegaly with a remarkable accuracy of 97%.
CONCLUSION: The proposed model is ready for clinical diagnosis and triaging of X-ray images. Our solution also offers efficient ways to show the presence of the above diseases using Grad-CAM technique.
Downloads
References
S. I. Nafisah and G. Muhammad, “Tuberculosis detection in chest radiograph using convolutional neural network architecture and explainable artificial intelligence,” Neural Comput. Appl., pp. 1–21, 2022.
D. Capellán-Martín et al., “Deep learning-based lung segmentation and automatic regional template in chest X-ray images for pediatric tuberculosis,” arXiv [cs.CV], 2023.
A. Wong, J. R. H. Lee, H. Rahmat-Khah, A. Sabri, A. Alaref, and H. Liu, “TB-Net: A tailored, self-attention deep convolutional neural network design for detection of tuberculosis cases from chest X-ray images,” Front. Artif. Intell., vol. 5, p. 827299, 2022.
S. Stirenko et al., “Chest X-ray analysis of tuberculosis by deep learning with segmentation and augmentation,” in 2018 IEEE 38th International Conference on Electronics and Nanotechnology (ELNANO), 2018.
W. Chokchaithanakul, P. Punyabukkana, and E. Chuangsuwanich, “Adaptive image preprocessing and augmentation for tuberculosis screening on out-of-domain chest X-ray dataset,” IEEE Access, vol. 10, pp. 132144–132152, 2022.
V. Acharya et al., “AI-assisted tuberculosis detection and classification from chest X-rays using a deep learning Normalization-free network model,” Comput. Intell. Neurosci., vol. 2022, p. 2399428, 2022.
T. Rahman et al., “Reliable tuberculosis detection using chest X-ray with deep learning, segmentation and visualization,” IEEE Access, vol. 8, pp. 191586–191601, 2020.
A. Jafar, M. T. Hameed, N. Akram, U. Waqas, H. S. Kim, and R. A. Naqvi, “CardioNet: Automatic semantic segmentation to calculate the cardiothoracic ratio for cardiomegaly and other chest diseases,” J. Pers. Med., vol. 12, no. 6, p. 988, 2022.
S. S. Sarpotdar, “Cardiomegaly detection using Deep convolutional neural network with U-Net,” arXiv [eess.IV], 2022.
I. Chamveha, T. Promwiset, T. Tongdee, P. Saiviroonporn, and W. Chaisangmongkon, “Automated cardiothoracic ratio calculation and cardiomegaly detection using deep learning approach,” arXiv [eess.IV], 2020.
I. Allaouzi and M. Ben Ahmed, “A novel approach for multi-label chest X-ray classification of common thorax diseases,” IEEE Access, vol. 7, pp. 64279–64288, 2019.
R. R. Selvaraju, M. Cogswell, A. Das, R. Vedantam, D. Parikh, and D. Batra, “Grad-CAM: Visual explanations from deep networks via gradient-based localization,” in 2017 IEEE International Conference on Computer Vision (ICCV), 2017.
S. Jaeger et al., “Automatic tuberculosis screening using chest radiographs,” IEEE Trans. Med. Imaging, vol. 33, no. 2, pp. 233–245, 2014.
S. Candemir et al., “Lung segmentation in chest radiographs using anatomical atlases with nonrigid registration,” IEEE Trans. Med. Imaging, vol. 33, no. 2, pp. 577–590, 2014.
J. Shiraishi et al., “Development of a digital image database for chest radiographs with and without a lung nodule: receiver operating characteristic analysis of radiologists’ detection of pulmonary nodules: Receiver operating characteristic analysis of radiologists’ detection of pulmonary nodules,” AJR Am. J. Roentgenol., vol. 174, no. 1, pp. 71–74, 2000.
A. Bustos, A. Pertusa, J.-M. Salinas, and M. de la Iglesia-Vayá, “PadChest: A large chest x-ray image dataset with multi-label annotated reports,” arXiv [eess.IV], 2019.
N. Gaggion, L. Mansilla, C. Mosquera, D. H. Milone, and E. Ferrante, “Improving anatomical plausibility in medical image segmentation via hybrid graph neural networks: applications to chest x-ray analysis,” arXiv [eess.IV], 2022
X. Wang, Y. Peng, L. Lu, Z. Lu, M. Bagheri, and R. M. Summers, “ChestX-Ray8: Hospital-scale chest X-ray database and benchmarks on weakly-supervised classification and localization of common thorax diseases,” in 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2017.
Y. Liu, Y.-H. Wu, Y. Ban, H. Wang, and M.-M. Cheng, “Rethinking computer-aided tuberculosis diagnosis,” in 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 2020.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2024 Ganesh Pradeep P V, Dinesh R, Anwesh Reddy Paduri, Narayana Darapaneni
This work is licensed under a Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License.
This is an open access article distributed under the terms of the CC BY-NC-SA 4.0, which permits copying, redistributing, remixing, transformation, and building upon the material in any medium so long as the original work is properly cited.
