Title: Transfer learning approach in deep neural networks for uterine fibroid detection

Authors: Sumod Sundar; S. Sumathy

Addresses: School of Computer Science and Engineering, VIT Vellore, Vellore, India ' School of Information Technology and Engineering, VIT Vellore, Vellore, India

Abstract: Convolutional neural network (CNN), a class of deep neural network, takes images as input and automatically extracts features for effective class prediction. The performance of CNN architecture is a major concern while dealing with fewer data. Traditional CNN architectures like ImageNet, AlexNet, GoogleNet are trained with a large quantity of data. Transfer learning is applied in order to combine pre-trained models such as Inception-V4 and support vector machine (SVM) for better performance with fewer data. The goal of the proposed approach is to efficiently detect fibroid presence in uterus MRI images. Features of these images are extracted using initial layers of Inception-V4 and are transferred to SVM during training. Several combinations on various network classifiers are tested and the performance metrics are evaluated. The proposed model attained an accuracy of 81.05% with a U-Kappa score of 0.402 on predicting fibroid images and 0.25% accuracy improvement compared with ResNet based model.

Keywords: convolutional neural network; CNN; transfer learning; deep learning; uterine fibroid; inception-V4; support vector machine; SVM; MRI imaging; medical imaging; feature transfer; object detection.

DOI: 10.1504/IJCSE.2022.120788

International Journal of Computational Science and Engineering, 2022 Vol.25 No.1, pp.52 - 63

Received: 23 Apr 2020
Accepted: 12 Mar 2021
Published online: 08 Feb 2022 *

Full-text access for editors Full-text access for subscribers Purchase this article Comment on this article