A Multi-Stage Deep Learning Approach to Tuberculosis Detection with Explainable Insights

Accurate and timely diagnosis of Tuberculosis (TB) is crucial, given its continued impact on global health. While chest X-rays are a common diagnostic tool, they are susceptible to subjective interpretation by radiologists. To enhance the reliability of TB detection, we present an AI-powered solution that integrates segmentation, classification, and explainability. Our approach utilizes UNet++ for precise lung region segmentation, which improves the accuracy of subsequent classification. By leveraging EfficientNetB0, our model delivers outstanding accuracy rates: 99.76% on the Tuberculosis Chest X-ray Dataset, 93.4% on the Montgomery and Shenzhen datasets, and 97.4% when all datasets are combined. To further enhance transparency in decision-making, we incorporate Explainable AI (XAI) techniques, including LIME, SHAP, and Grad-CAM. Our approach is more accurate than current works and can be used on a large scale. Due to its precision and interpretability, we believe it can significantly improve TB detection worldwide and ultimately help more patients get the necessary care.

• Publication year

  2025

• Venue

  2025 2nd International Conference on Next-Generation Computing, IoT and Machine Learning (NCIM)

• Publication date

  2025-06-27

• Fields of study

  Not labeled

• Identifiers
  DOI 10.1109/NCIM65934.2025.11159991
• External record

  Open on Semantic Scholar

• Source metadata

  Semantic Scholar

• No claims are published for this paper.

• No concepts are published for this paper.

• An explainable transfer learning framework for multi-classification of lung diseases in chest X-rays

  2024cited by this paper
• Global, regional, and national age-specific progress towards the 2020 milestones of the WHO End TB Strategy: a systematic analysis for the Global Burden of Disease Study 2021

  2024influential reference
• Deep learning models for tuberculosis detection and infected region visualization in chest X-ray images

  2023cited by this paper
• Machine and Deep Learning for Tuberculosis Detection on Chest X-Rays: Systematic Literature Review

  2023cited by this paper
• Interpreting Black-Box Models: A Review on Explainable Artificial Intelligence

  2023cited by this paper
• An Improved Densenet Deep Neural Network Model for Tuberculosis Detection Using Chest X-Ray Images

  2023cited by this paper
• Tuberculosis chest X-ray detection using CNN-based hybrid segmentation and classification approach

  2023cited by this paper
• Tuberculosis detection in chest radiograph using convolutional neural network architecture and explainable artificial intelligence

  2022cited by this paper
• Biomedical Data Analysis and Processing Using Explainable (XAI) and Responsive Artificial Intelligence (RAI)

  2022cited by this paper
• Deep and Hybrid Learning Technique for Early Detection of Tuberculosis Based on X-ray Images Using Feature Fusion

  2022cited by this paper
• Automatic lung segmentation in chest X-ray images using improved U-Net

  2021influential reference
• Deep learning assistance for tuberculosis diagnosis with chest radiography in low-resource settings

  2021cited by this paper
• Dense-Unet: a light model for lung fields segmentation in Chest X-Ray images

  2020influential reference
• Reliable Tuberculosis Detection Using Chest X-Ray With Deep Learning, Segmentation and Visualization

  2020cited by this paper
• Attention U-Net Based Adversarial Architectures for Chest X-ray Lung Segmentation information

  2020cited by this paper
• An automatic method for lung segmentation and reconstruction in chest X-ray using deep neural networks

  2019cited by this paper
• Automated TB classification using ensemble of deep architectures

  2019cited by this paper
• Lung Field Segmentation in Chest Radiographs From Boundary Maps by a Structured Edge Detector

  2018influential reference
• Deep learning for chest radiograph diagnosis: A retrospective comparison of the CheXNeXt algorithm to practicing radiologists

  2018cited by this paper
• Convolutional Neural Networks for Medical Image Analysis: Full Training or Fine Tuning?

  2016cited by this paper
• Two public chest X-ray datasets for computer-aided screening of pulmonary diseases.

  2014cited by this paper
• Tuberculosis: From an incurable scourge to a curable disease - journey over a millennium

  2013cited by this paper

• No citing papers are available for this paper.