A Lightweight Hybrid Deep Learning Model for Tuberculosis Detection from Chest X-Rays.
- 1. Faculty of Artificial Intelligence and Data Science, UNESCO Chair in Data Science for Sustainable Development, Arab American University, Ramallah P600, Palestine.
- 2. Faculty of Artificial Intelligence and Data Science, Arab American University, Ramallah P600, Palestine.
- 3. Department of Natural, Engineering and Technology Sciences, Arab American University, Ramallah P600, Palestine.
- 4. Information Technologies Group, School of Telecommunication Engineering, University of Vigo, 36310 Vigo, Spain.
Description
Background/Objectives: Tuberculosis remains a significant global health problem, particularly in resource-limited environments. Its mortality and spread can be considerably decreased by early and precise detection via chest X-ray imaging. This study introduces a novel approach based on hybrid deep learning for Tuberculosis detection from chest X-ray images. Methods: The introduced approach combines GhostNet, a lightweight convolutional neural network tuned for computational efficiency, and MobileViT, a transformer-based model that can capture both local spatial patterns and global contextual dependencies. Through such integration, the model attains a balanced trade-off between classification accuracy and computational efficiency. The architecture employs feature fusion, where spatial features from GhostNet and contextual representations from MobileViT are globally pooled and concatenated, which allows the model to learn discriminative and robust feature representations. Results: The suggested model was assessed on two publicly available chest X-ray datasets and contrasted against several cutting-edge convolutional neural network architectures. Findings showed that the introduced hybrid model surpasses individual baselines, attaining 99.52% accuracy on dataset 1 and 99.17% on dataset 2, while keeping low computational cost (7.73M parameters, 282.11M Floating Point Operations). Conclusions: These outcomes verify the efficacy of feature-level fusion between a convolutional neural network and transformer branches, allowing robust tuberculosis detection with low inference overhead. The model is ideal for clinical deployment and resource-constrained contexts due to its high accuracy and lightweight design.
Open Access
Licence Attribution (CC BY)
Publisher Website
Access full text
References
Bhuria, R., and Gupta, S. (2024, January 25\u201326). Automated Tuberculosis Det...
Read more
Stirenko, S., Kochura, Y., Alienin, O., Rokovyi, O., Gang, P., Zeng, W., and Gor...
Read more
Xie . Computer-Aided System for the Detection of Multicategory Pulmonary Tubercu...
Read more
Sivaramakrishnan, R., Antani, S., Candemir, S., Xue, Z., Abuya, J., Kohli, M., A...
Read more
Shaw, P., Uszkoreit, J., and Vaswani, A. (2018). Self-attention with relative po...
Read more
Showing first 5 of 59 references.