Reliable Tuberculosis Detection using Chest X-ray with Deep Learning, Segmentation and Visualization

Abstract

Tuberculosis (TB) is a chronic lung disease that occurs due to bacterial infection and is one of the top 10 leading causes of death. Accurate and early detection of TB is very important, otherwise, it could be life-threatening. In this work, we have detected TB reliably from the chest X-ray images using image pre-processing, data augmentation, image segmentation, and deep-learning classification techniques. Several public databases were used to create a database of 700 TB infected and 3500 normal chest X-ray images for this study. Nine different deep CNNs (ResNet18, ResNet50, ResNet101, ChexNet, InceptionV3, Vgg19, DenseNet201, SqueezeNet, and MobileNet), which were used for transfer learning from their pre-trained initial weights and trained, validated and tested for classifying TB and non-TB normal cases. Three different experiments were carried out in this work: segmentation of X-ray images using two different U-net models, classification using X-ray images, and segmented lung images. The accuracy, precision, sensitivity, F1-score, specificity in the detection of tuberculosis using X-ray images were 97.07 %, 97.34 %, 97.07 %, 97.14 % and 97.36 % respectively. However, segmented lungs for the classification outperformed than whole X-ray image-based classification and accuracy, precision, sensitivity, F1-score, specificity were 99.9 %, 99.91 %, 99.9 %, 99.9 %, and 99.52 % respectively. The paper also used a visualization technique to confirm that CNN learns dominantly from the segmented lung regions results in higher detection accuracy. The proposed method with state-of-the-art performance can be useful in the computer-aided faster diagnosis of tuberculosis.

15 pages, 12 figure and 5 Tables

References in corpus (7)

• Very Deep Convolutional Networks for Large-Scale Image Recognition
• Convolutional Neural Networks for Medical Image Analysis: Full Training or Fine Tuning?
• CheXNet: Radiologist-Level Pneumonia Detection on Chest X-Rays with Deep Learning
• SmoothGrad: removing noise by adding noise
• Transfer Learning with Deep Convolutional Neural Network (CNN) for Pneumonia Detection using Chest X-ray
• Multi-source Transfer Learning with Convolutional Neural Networks for Lung Pattern Analysis
• Estimating Blood Pressure from Photoplethysmogram Signal and Demographic Features using Machine Learning Techniques

Cited by in corpus (10)

• Recent Progress in Transformer-based Medical Image Analysis
• COV-ECGNET: COVID-19 detection using ECG trace images with deep convolutional neural network
• Improving deep neural network generalization and robustness to background bias via layer-wise relevance propagation optimization
• QUCoughScope: An Artificially Intelligent Mobile Application to Detect Asymptomatic COVID-19 Patients using Cough and Breathing Sounds
• Deep learning in computed tomography pulmonary angiography imaging: a dual-pronged approach for pulmonary embolism detection
• Exploring the Effect of Image Enhancement Techniques on COVID-19 Detection using Chest X-rays Images
• Improving Robustness and Reliability in Medical Image Classification with Latent-Guided Diffusion and Nested-Ensembles
• Empowering Tuberculosis Screening with Explainable Self-Supervised Deep Neural Networks
• Exponential Pixelating Integral transform with dual fractal features for enhanced chest X-ray abnormality detection
• INSITE: labelling medical images using submodular functions and semi-supervised data programming