Tuberculosis Detection From Chest X-Ray Image Modalities Based on Transformer and Convolutional Neural Network

Abstract

Tuberculosis (TB) is an airborne disease with a high fatality rate that often affects people’s lungs. Early detection of the disease can guarantee a cure, but that is not the case due to the lack of experts and screening facilities. With the advancement in Artificial Intelligence (AI) techniques, publicly available clinical datasets and high computer systems, several solutions have been proposed to automate the diagnosing process based on deep learning (DL) algorithms. This study presents a DL technique based on a combined Data efficient image transformer and the Residual Network-16 (ResNet-16) model for effective TB diagnosis from X-ray images. The TBX11K dataset got utilized for this investigation. The dataset has three categories—Healthy, Sick but non-TB and TB. Relying on only healthy X-rays as the negative category in clinical settings where there are many unwell but non-TB samples can cause considerable false positives in the model prediction, hence the adoption of the sick but non-TB class. The dataset got divided into training, validation, and testing sets for the experiment. The self-attention mechanism within the transformer part of the proposed model learns crucial information and constructs the relationship between image tokens. The ResNet-16 part uses depth-wise convolution to gather local representations and reduce computing costs while increasing the diagnosing accuracy. The global average pooling is applied to the feature maps at the final convolution layer to generate heatmaps based on the class activation map to show exactly where the prediction happened. The suggested model obtained TB diagnostic accuracy, sensitivity, specificity and precision rates of 99.38%, 99.49%, 99.26% and 99.24%, respectively. Additionally, it is lightweight (6.9M) and detects more quickly (4.79ms) than the other comparative cutting-edge versions. The suggested model’s exceptional results demonstrate its effectiveness in TB detection, allowing for practical use.

Keywords

• Convolutional neural network (CNN)
• deep learning
• vision transformer network
• tuberculosis
• X-ray