Back to Search
Start Over
Hybrid Transfer Learning for Classification of Uterine Cervix Images for Cervical Cancer Screening
- Source :
- J Digit Imaging
- Publication Year :
- 2019
- Publisher :
- Springer International Publishing, 2019.
-
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%.
- Subjects :
- Computer science
Uterine Cervical Neoplasms
Machine learning
computer.software_genre
Convolutional neural network
Field (computer science)
030218 nuclear medicine & medical imaging
Domain (software engineering)
Machine Learning
03 medical and health sciences
0302 clinical medicine
Humans
Radiology, Nuclear Medicine and imaging
Early Detection of Cancer
Hyperparameter
Original Paper
Radiological and Ultrasound Technology
business.industry
Deep learning
Class (biology)
Computer Science Applications
Identification (information)
Female
Artificial intelligence
Neural Networks, Computer
Transfer of learning
business
computer
030217 neurology & neurosurgery
Subjects
Details
- Language :
- English
- Database :
- OpenAIRE
- Journal :
- J Digit Imaging
- Accession number :
- edsair.doi.dedup.....abf0c65826e80cbfb8ab1f71b15261f5