TB Research

From Pixels to Diagnosis: Convolutional Neural Networks in Tuberculosis Screening

Tanvir Mahtab Zihan, Abduz Zami, Mohiuddin Ahmed, Md. Rakibul Islam, Fahim Ahmed

Abstract

Tuberculosis (TB) is a chronic lung disease that constitutes one of the top 10 global causes of death. Timely and precise identification of TB is critical, as untreated cases can pose life-threatening risks. Traditional radiographic interpretation by trained professionals is subjective, time-consuming, and prone to human error. This study proposes an automated deep learning-based approach for TB detection from chest X-ray images, based on EfficientNet-B0 architecture. The Convolutional Neural Network (CNN) extracts discriminative features, allowing it to differentiate between normal and abnormal lung patterns indicative of TB. Our choice of the EfficientNet-B0 architecture played a pivotal role in the success of our TB detection model. Training and evaluation were conducted on three distinct datasets: Tuberculosis (TB) Chest X-ray Database, Shenzhen Hospital CXR Set, and Montgomery County CXR Set. The key breakthrough is attaining an accuracy of 99.29% on the Tuberculosis (TB) Chest X-ray Database and 98.50% on the combined dataset. This study presents a promising approach for automated TB screening using readily available chest X-ray images, potentially improving diagnostic accuracy and efficiency, especially in resource-limited settings. We compared our scores with those of other works and found that ours significantly outperforms them.

MeSH terms

  • Convolutional neural network
  • Tuberculosis
  • Discriminative model
  • Artificial intelligence
  • Computer science
  • Deep learning
  • Pulmonary tuberculosis
  • Set (abstract data type)
  • Machine learning
  • Medicine
  • Key (lock)
  • Identification (biology)
  • Pattern recognition (psychology)