An automated tuberculosis detection approach using deep learning and machine learning techniques from chest X-ray images: a step towards effective diagnosis.

PURPOSE: Tuberculosis (TB) is a severe bacterial infectious lung disease. Millions of people die or experience severe health complications due to TB each year. Accurate, automated, and effective detection of TB is key to curing and preventing person-to-person transmission. In this regard, deep learning (DL) and machine learning (ML) techniques applied to chest X-ray (CXR) images have proved effective. Here, we present our DL- and ML-based approach for TB detection using CXR images.

MATERIAL AND

METHODS

We implemented convolutional neural network (CNN)-based pre-trained DL models, such as DenseNet121, DenseNet169, DenseNet201, ResNet152, and VGG19, as feature extractors. ML models, including support vector machine (SVM), XGBoost, logistic regression, and a DL-based custom model, were used as classifiers. A total of 2,391 CXR images from three publicly available datasets were considered.

RESULTS

We found three models achieving the highest value in different evaluation metrics: accuracy (99.91%) with ResNet152 and SVM; recall (99.22%), precision (99.23%), and F1-score (99.22%) with DenseNet169 and the custom classifier; and area under the curve (99.99%) with DenseNet201 and the custom classifier. Among these models, we propose DenseNet169 with the custom classifier as the best performer for potential clinical application, as it reflected a high and well-balanced performance across all evaluation metrics.

CONCLUSIONS

This study evaluated pre-trained CNN-based DL and ML models on pre-processed CXR images for the detection of TB. The DenseNet169 model with a custom classifier stands out with its high and well-balanced performance, offering a significant contribution to the effective and automated detection of TB.