TB-CXRNet: Tuberculosis and Drug-Resistant Tuberculosis Detection Technique Using Chest X-ray Images

Tawsifur Rahman,Amith Khandakar,Ashiqur Rahman,Susu M. Zughaier,Muna Al Maslamani,Moajjem Hossain Chowdhury,Anas M. Tahir,Md. Sakib Abrar Hossain,Muhammad E. H. Chowdhury

DOI: https://doi.org/10.1007/s12559-024-10259-3

IF: 4.89

2024-02-17

Cognitive Computation

Abstract:Abstract Tuberculosis (TB) is a chronic infectious lung disease, which caused the death of about 1.5 million people in 2020 alone. Therefore, it is important to detect TB accurately at an early stage to prevent the infection and associated deaths. Chest X-ray (CXR) is the most popularly used method for TB diagnosis. However, it is difficult to identify TB from CXR images in the early stage, which leads to time-consuming and expensive treatments. Moreover, due to the increase of drug-resistant tuberculosis, the disease becomes more challenging in recent years. In this work, a novel deep learning-based framework is proposed to reliably and automatically distinguish TB, non-TB (other lung infections), and healthy patients using a dataset of 40,000 CXR images. Moreover, a stacking machine learning-based diagnosis of drug-resistant TB using 3037 CXR images of TB patients is implemented. The largest drug-resistant TB dataset will be released to develop a machine learning model for drug-resistant TB detection and stratification. Besides, Score-CAM-based visualization technique was used to make the model interpretable to see where the best performing model learns from in classifying the image. The proposed approach shows an accuracy of 93.32% for the classification of TB, non-TB, and healthy patients on the largest dataset while around 87.48% and 79.59% accuracy for binary classification (drug-resistant vs drug-sensitive TB), and three-class classification (multi-drug resistant (MDR), extreme drug-resistant (XDR), and sensitive TB), respectively, which is the best reported result compared to the literature. The proposed solution can make fast and reliable detection of TB and drug-resistant TB from chest X-rays, which can help in reducing disease complications and spread.

computer science, artificial intelligence,neurosciences

What problem does this paper attempt to address?

The paper primarily aims to address two key issues: 1. **Automatic Detection of Tuberculosis (TB)**: The study proposes a new deep learning-based framework (referred to as TB-CXRNet) for reliably and automatically distinguishing between TB patients, non-TB patients (other lung infections), and healthy individuals in chest X-ray images. This framework was trained and tested on a dataset containing 40,000 chest X-ray images, achieving a high classification accuracy of 93.32%. 2. **Diagnosis of Drug-Resistant Tuberculosis**: Additionally, the study developed a machine learning-based cascade model to diagnose drug-resistant tuberculosis (including multidrug-resistant TB, MDR-TB, and extensively drug-resistant TB, XDR-TB). This model was trained on a dataset containing 3,037 chest X-ray images of TB patients, achieving an accuracy of 87.48% for binary classification (drug-resistant vs. drug-sensitive TB) and 79.59% for three-class classification (MDR-TB, XDR-TB, and drug-sensitive TB). To achieve these goals, the research team first created a large benchmark dataset (QU-MLG-TB) containing a substantial number of chest X-ray images with ground truth labels. They then utilized various preprocessing techniques, such as gamma correction to enhance image quality, and performed precise lung segmentation using a Feature Pyramid Network (FPN) combined with a DenseNet121 encoder. Finally, after feature extraction through CheXNet, a Multi-Layer Perceptron (MLP) was used for classification. Additionally, Score-CAM visualization techniques were employed to improve the model's interpretability, helping to understand how the model makes classification decisions. In summary, this study contributes to improving the speed and accuracy of TB diagnosis by proposing an efficient method to identify TB and its different types of drug resistance, thereby reducing the complications and spread of the disease.

TB-CXRNet: Tuberculosis and Drug-Resistant Tuberculosis Detection Technique Using Chest X-ray Images

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

TB-Net: A Tailored, Self-Attention Deep Convolutional Neural Network Design for Detection of Tuberculosis Cases From Chest X-Ray Images

Tuberculosis Detection Using Chest X-Ray with Deep Learning and Visualization

Proposing a novel multi-instance learning model for tuberculosis recognition from chest X-ray images based on CNNs, complex networks and stacked ensemble

A Lightweight, Rapid and Efficient Deep Convolutional Network for Chest X-Ray Tuberculosis Detection

AI-Assisted Tuberculosis Detection and Classification from Chest X-Rays Using a Deep Learning Normalization-Free Network Model

A novel dense-net deep neural network with enhanced feature selection method for classification of different stages of tuberculosis using chest X-ray images

Deep Learning Methods for Screening Pulmonary Tuberculosis Using Chest X-rays

An Efficient Mixture of Deep and Machine Learning Models for COVID-19 and Tuberculosis Detection Using X-Ray Images in Resource Limited Settings

A Robust Tuberculosis Diagnosis Using Chest X-Rays Based on a Hybrid Vision Transformer and Principal Component Analysis

Deep Learning for Automated Screening of Tuberculosis from Indian Chest X-rays: Analysis and Update

A robust ensemble deep learning framework for accurate diagnoses of tuberculosis from chest radiographs

TX-CNN: Detecting Tuberculosis in Chest X-ray Images Using Convolutional Neural Network.

Deep learning-based automatic detection of tuberculosis disease in chest X-ray images

Revisiting Computer-Aided Tuberculosis Diagnosis

Computer-aided diagnosis using embedded ensemble deep learning for multiclass drug-resistant tuberculosis classification

Tuberculosis detection in chest radiograph using convolutional neural network architecture and explainable artificial intelligence

Self-Trained Convolutional Neural Network (CNN) for Tuberculosis Diagnosis in Medical Imaging

CovidAID: COVID-19 Detection Using Chest X-Ray

Artificial Intelligence-based Deep Learning Architecture for Tuberculosis Detection