Tuberculosis Diagnosis Using Deep Transferred EfficientNet
Chengxi Huang, Wei Wang, Xin Zhang, Shuihua Wang, Yudong Zhang
Abstract
Tuberculosis is a very deadly disease, with more than half of all tuberculosis cases dead in countries and regions with relatively poor health care resources. Fortunately, the disease is curable, and early diagnosis and medication can go a long way toward curing TB patients. Unfortunately, traditional methods of TB diagnosis rely on specialist doctors, which is lacking in areas with high TB mortality rates. Diagnostic methods based on artificial intelligence technology are one of the solutions to this problem. We propose a Deep Transferred EfficientNet with SVM (DTE-SVM), which replaces the pre-trained EfficientNet classification layer with an SVM classifier and achieves auspicious performance on a small dataset. After ten runs of 10-fold Cross-Validation, the DTE-SVM has a sensitivity of 93.89±1.96, a specificity of 95.35±1.31, a precision of 95.30±1.24, an accuracy of 94.62±1.00, and an F1-score of 94.62±1.00. In addition, our study conducted ablation studies on the effect of the SVM classifier on model performance and briefly discussed the results.