Hybrid Transfer Learning for Classification of Uterine Cervix Images for Cervical Cancer Screening

Kudva, Vidya and Keerthana, Prasad and Guruvare, Shyamala (2020) Hybrid Transfer Learning for Classification of Uterine Cervix Images for Cervical Cancer Screening. Journal of Digital Imaging, 33. pp. 619-631. ISSN 0897-1889

[img] PDF
8972.pdf - Published Version
Restricted to Registered users only

Download (4MB) | Request a copy

Abstract

Transfer learning using deep pre-trained convolutional neural networks is increasingly used to solve a large number of problems in the medical field. In spite of being trained using images with entirely different domain, these networks are flexible to adapt to solve a problem in a different domain too. Transfer learning involves fine-tuning a pre-trained network with optimal values of hyperparameters such as learning rate, batch size, and number of training epochs. The process of training the network identifies the relevant features for solving a specific problem. Adapting the pre-trained network to solve a different problem requires fine-tuning until relevant features are obtained. This is facilitated through the use of large number of filters present in the convolutional layers of pre-trained network. A very few features out of these features are useful for solving the problem in a different domain, while others are irrelevant, use of which may only reduce the efficacy of the network. However, by minimizing the number of filters required to solve the problem, the efficiency of the training the network can be improved. In this study, we consider identification of relevant filters using the pre-trained networks namely AlexNet and VGG-16 net to detect cervical cancer from cervix images. This paper presents a novel hybrid transfer learning technique, in which a CNN is built and trained from scratch, with initial weights of only those filters which were identified as relevant using AlexNet and VGG-16 net. This study used 2198 cervix images with 1090 belonging to negative class and 1108 to positive class. Our experiment using hybrid transfer learning achieved an accuracy of 91.46%.

Item Type: Article
Uncontrolled Keywords: Cervical cancer screening . Deep learning . Transfer learning . Hybrid transfer learning . Machine learning . Medical image classification . Artificial intelligence
Subjects: Information Sciences > MCIS Manipal
Depositing User: MIT Library
Date Deposited: 27 Jun 2020 10:40
Last Modified: 27 Jun 2020 10:40
URI: http://eprints.manipal.edu/id/eprint/155369

Actions (login required)

View Item View Item