An Enhanced Deep Learning Architecture for Classification of Tuberculosis Types from CT Lung Images

Xiaohong Gao,Richard Comley,Maleika Heenaye-Mamode Khan
DOI: https://doi.org/10.1109/icip40778.2020.9190815
2020-01-01
Abstract:In this work, an enhanced ResNet deep learning network, depth-ResNet, has been developed to classify the five types of Tuberculosis (TB) lung CT images. Depth-ResNet takes 3D CT images as a whole and processes the volumatic blocks along depth directions. It builds on the ResNet-50 model to obtain 2D features on each frame and injects depth information at each process block. As a result, the averaged accuracy for classification is 71.60% for depth-ResNet and 68.59% for ResNet. The datasets are collected from the ImageCLEF 2018 competition with 1008 training data in total, where the top reported accuracy was 42.27%.
What problem does this paper attempt to address?