Tuberculosis chest X-ray detection using CNN-based hybrid segmentation and classification approach

Abstract

Tuberculosis still significantly impacts the world's population, with more than 10 million people getting sick each year. Researchers have focused on developing computer-aided diagnosis (CAD) technology based on X-ray imaging to support the identification of tuberculosis, and deep learning is a popular and efficient method. However, deep learning-based CAD approaches usually ignore the relationship between the two vision tasks of specific region segmentation and classification. In this research, we introduced a novel TB-UNet, which is based on dilated fusion block (DF) and Attention block (AB) block for accurate segmentation of lungs regions and achieved the highest results in terms of Precision (0.9574), Recall (0.9512), and F1score (0.8988), IoU (0.8168) and Accuracy (0.9770). We also proposed TB-DenseNet which is based on five dual convolution blocks, DenseNet-169 layer, and a feature fusion block for the precise classification of tuberculosis images. The experiments have been performed on three chest X-ray (CXR) datasets, segmented images, and original images are fed to TB-DenseNet for better classification. Furthermore, the proposed method is tested against simultaneously three different diseases, such as Pneumonia, COVID-19, and Tuberculous. The highest results are achieved in terms of Precision (0.9567), Recall (0.9510), F1score (0.9538), and Accuracy (0.9510). The achieved results reflect that our proposed method produces the highest accuracy compared to the state-of-the-art methods. The source code is available at: https://github.com/ahmedeqbal/TB-DenseNet.