Optimizing chest tuberculosis image classification with oversampling and transfer learning

Abstract

AbstractTuberculosis (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.