Using Deep Learning to Classify X-ray Images of Potential Tuberculosis Patients

Deep Learning is widely used for image classification. Its success heavily relies on data which contains a sufficient amount of region of interest (~10%). However, due to the region of interest in medical images being as low as 1% of the entire image, Deep Learning has not been conveniently used for such cases. In this study, we employ recent techniques brought forth in Deep Learning and aim to classify X-ray images of potential Tuberculosis patients. Different types of learning rate enhancement techniques were used. Significant improvement was observed when coarse-to-fine knowledge transfer was employed to fine-tune the model further using multiple data augmentation techniques. We achieved an overall accuracy of 94.89% on the augmented images.

[1]  Joshua Burrill,et al.  Tuberculosis: a radiologic review. , 2007, Radiographics : a review publication of the Radiological Society of North America, Inc.

[2]  Wei Liu,et al.  SSD: Single Shot MultiBox Detector , 2015, ECCV.

[3]  Clement J. McDonald,et al.  Automatic Tuberculosis Screening Using Chest Radiographs , 2014, IEEE Transactions on Medical Imaging.

[4]  Luciano Oliveira,et al.  A coarse-to-fine deep learning for person re-identification , 2016, 2016 IEEE Winter Conference on Applications of Computer Vision (WACV).

[5]  Leslie N. Smith,et al.  Cyclical Learning Rates for Training Neural Networks , 2015, 2017 IEEE Winter Conference on Applications of Computer Vision (WACV).

[6]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[7]  Sebastian Ruder,et al.  Universal Language Model Fine-tuning for Text Classification , 2018, ACL.

[8]  Barbara Caputo,et al.  Frustratingly Easy NBNN Domain Adaptation , 2013, 2013 IEEE International Conference on Computer Vision.

[9]  Jian Sun,et al.  Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification , 2015, 2015 IEEE International Conference on Computer Vision (ICCV).

[10]  Clement J. McDonald,et al.  Lung Segmentation in Chest Radiographs Using Anatomical Atlases With Nonrigid Registration , 2014, IEEE Transactions on Medical Imaging.