A Transfer Learning Approach to Leveraging Pre-trained CNN Models to Detect Tuberculosis using Chest X-Rays

Main Article Content

Ms. Sweety Bakyarani E., Dr. H. Srimathi, Dr. P. J. Arul Leena Rose

Abstract

Pre-Trained Deep Learning models1 are Convolutional Neural Networks that has been developed and bench marked for performing image classificationon a very large dataset such as ImageNet2. These pre-trained networks have demonstrated their capability to work well even with images that are not part of the ImageNet2 dataset through the transfer learning approach3. Pre-trained models help us with eliminating the cost and time involved in training a CNN model from scratch. In this paper we evaluate the performance of some ILSVRC2(The ImageNet Large Scale Visual Recognition Challenge) award winning pre-trained CNN models over the publically available Tuberculosis CXR datasets, we also appraise the ability of these pre-trained models to generalize for tuberculosis detection through CXR images. We customize the pre-trained model as per our requirements and fine tune its performance. The pre-trained CNN models that we have chosen to evaluate on the tuberculosis dataset as a part of this studyare:VGG-164 ,VGG-194,AlexNet5, ResNet-506, Inception7 and DenseNet8.

Article Details

Section
Articles