Segmentation and classification of ovarian cancer based on conditional adversarial image to image translation approach

Abstract

AbstractMedical image analysis and disease diagnosis have significantly improved with the use of AI and Machine Learning algorithms. Automated systems for medical image analysis will help the doctors and radiologists understand the anomaly in a short span of time and with better visualization. Such automated systems will help to reduce the time taken for diagnosis by experts. Recently, Computer Vision is industrialized with the advancements in algorithms and hardware. The proposed study aims to develop a computer vision solution for automatic segmentation and classification of ovarian tumours in discriminating between benign and malignant tumours by image‐to‐image translation approach using Conditional Generative Adversarial Network (cGAN). Our method uses a novel algorithm which segments and classifies the images in a single pipeline which makes the algorithm unique and useful. This research also aims to compare its diagnostic accuracy with that of an expert radiologist. The dataset used by in the present study is formulated with images obtained from a hospital and annotated by doctors from the hospital. The obtained results show the proposed study is promising for ovarian cancer segmentation and classification with an average segmentation score of 0.825 for benign and 0.765 for malignant and classification accuracy of 83% for benign and 79% for malignant, precision score of 85% for benign and 80% for malignant and F1 score of 81% for benign and 80.1% for malignant images. The proposed methodology is evaluated on the existing MRI images to perform segmentation and classification. The results obtained shows that the proposed methodology can perform well on other MRI images. In this study, proposed methodology is convenient as separate segmentation need not be done and is giving good result. The same MRI images are segmented using UNet and classified using RESNET 101 and results are compared with the proposed methodology.