phat 21(27): e5

Research Article

Cervix Image Classification for Prognosis of Cervical Cancer using Deep Neural Network with Transfer Learning

Download1812 downloads
  • @ARTICLE{10.4108/eai.12-4-2021.169183,
        author={Sanjeev Dhawan and Kulvinder Singh and Mamta Arora},
        title={Cervix Image Classification for Prognosis of Cervical  Cancer using Deep Neural Network with Transfer Learning},
        journal={EAI Endorsed Transactions on Pervasive Health and Technology},
        volume={7},
        number={27},
        publisher={EAI},
        journal_a={PHAT},
        year={2021},
        month={4},
        keywords={Pervasive healthcare, Cancer, Cervical Cancer, Artificial Intelligence, Deep Learning, Convolutional Neural Network, Inception V3, ResNet50, VGG19, Saliency Map},
        doi={10.4108/eai.12-4-2021.169183}
    }
    
  • Sanjeev Dhawan
    Kulvinder Singh
    Mamta Arora
    Year: 2021
    Cervix Image Classification for Prognosis of Cervical Cancer using Deep Neural Network with Transfer Learning
    PHAT
    EAI
    DOI: 10.4108/eai.12-4-2021.169183
Sanjeev Dhawan1, Kulvinder Singh1, Mamta Arora2,3,*
  • 1: Faculty of Computer Science & Engineering, Department of Computer Science & Engineering, University Institute of Engineering & Technology (U.I.E.T), Kurukshetra University, Kurukshetra, Haryana, India
  • 2: Ph.D. Research Scholar, Department of Computer Science & Engineering, University Institute of Engineering & Technology (U.I.E.T), Kurukshetra University, Kurukshetra, Haryana, India
  • 3: Assistant Professor, Department of Computer Science and Technology, Manav Rachna University, Faridabad, Haryana, India
*Contact email: imamta.arora@gmail.com

Abstract

INTRODUCTION: Cervical cancer is the leading cancer among the other female cancers. It develops in the cervix of women. It takes decades in development thus can be preventable if diagnosed at an early stage. The cervix is classified into three types Type I/II/III. The efficacy of the treatment depends on the diagnosis of the right type of cervix. There is a thin line difference between the three types. Thus, identification of the right type of cervix becomes a difficult task for the health care providers too. To aid this problem, we proposed an algorithm based on the standard transfer learning approach used for building a model that classifies cervix images.

OBJECTIVES: The objective of this study is to develop a cervical cancer predictive model based on deep learning and transfer learning techniques that will recognize and classify the cervix images into one of the classes (Type 1/Type2/Type3).

METHODS: Techniques used for carrying out the experimental work includes deep learning and Transfer Learning. The three pertained models namely InceptionV3, ResNet50, and VGG19 are used for creating ConvNet that will classify the cervix images.

RESULTS: The result of the experiment reveals that the Inception v3 model is performing better than Vgg19 and ResNet50 with an accuracy of 96.1% on the cervical cancer dataset.

CONCLUSION: In the future, augmentation techniques can be employed to achieve better accuracy.