Tuberculosis Diagnosis Using Deep Transferred EfficientNet

Chengxi Huang, Wei Wang, Xin Zhang, Shui Hua Wang, Yu Dong Zhang*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

16 Citations (Scopus)

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.

Original languageEnglish
Pages (from-to)2639-2646
Number of pages8
JournalIEEE/ACM Transactions on Computational Biology and Bioinformatics
Volume20
Issue number5
DOIs
Publication statusPublished - 1 Sept 2023
Externally publishedYes

Keywords

  • SVM
  • Tuberculosis
  • deep learning
  • efficientNet
  • transfer learning

Fingerprint

Dive into the research topics of 'Tuberculosis Diagnosis Using Deep Transferred EfficientNet'. Together they form a unique fingerprint.

Cite this