IET Image Processing (Apr 2024)

Optimizing chest tuberculosis image classification with oversampling and transfer learning

  • Ali Alqahtani,
  • Qasem Abu Al‐Haija,
  • Abdulaziz A. Alsulami,
  • Badraddin Alturki,
  • Nayef Alqahtani,
  • Raed Alsini

DOI
https://doi.org/10.1049/ipr2.13010
Journal volume & issue
Vol. 18, no. 5
pp. 1109 – 1118

Abstract

Read online

Abstract Tuberculosis (TB) is an extremely contagious illness caused by Mycobacterium tuberculosis. Chest tuberculosis classification is conducted based on a deep convolutional neural network architecture. In this research, a pre‐trained network is utilized to demonstrate the advantage of using the oversampling technique on the classification of TB and compare results with recent research that used the same dataset. Therefore, the dataset consists of 3500 uninfected TB cases and 700 infected with TB. This paper circumvents the imbalance by using the oversampling technique in X‐ray TB images to be fed into several pre‐trained networks for TB classification. The oversampling technique is crucial in enhancing the performance of TB classification compared with other pre‐trained models reported here. Inceptionv3 shows a promising result compared to other pre‐trained models; it achieves 99.94% accuracy, 99.88% precision, 100% recall, and 99.94% F1‐Score.

Keywords