TB Research

Deep pre-trained networks as a feature extractor with XGBoost to detect tuberculosis from chest X-ray

Muhammad Rahman, Yongzhong Cao, Xiaobing Sun, Bin Li, Yameng Hao

Computers & Electrical Engineering · 2021-06

Abstract

Pulmonary Tuberculosis is a plague caused by Mycobacterium tuberculosis or Tubercle bacillus, which kills 1.8 million people worldwide. Tuberculosis is among the top 10 deadly diseases. It can be life-threatening if it does not diagnose at the initial stage. This study identifies Tuberculosis from chest X-ray images, utilizing image preprocessing techniques, deep learning methods and a publicly accessible dataset of 7000 (3500 normal and 3500 tuberculosis infected) chest X-ray images. We used three pre-trained networks (ResNet101, VGG19, and DenseNet201) to extract features from chest X-ray images. The eXtreme Gradient Boosting (XGBoost) model is used to classify tuberculosis and normal cases. The highest results (Area Under Curve of 99.93 ± 0.13%, accuracy 99.92 ± 0.14%, precision 99.85 ± 0.20%, sensitivity 100 ± 0.1%, F1-score 99.92 ± 0.14% and specificity 99.85 ± 0.20%) achieved with DenseNet201-XGBoost architecture, for tuberculosis chest X-ray images classification, as compared to the ResNet101-XGBoost and VGG19-XGBoost architectures. The proposed method bestows hope to radiologists and medical facilities in developing countries to tackle Tuberculosis's early diagnosis problem.

MeSH terms

  • Tuberculosis
  • Mycobacterium tuberculosis
  • Artificial intelligence
  • Preprocessor
  • Deep learning
  • Pulmonary tuberculosis
  • Extractor
  • Medicine
  • Pattern recognition (psychology)
  • Radiology