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