Prostate Segmentation and Lesions Classification in CT Images Using Mask R-CNN

Non-cancerous prostate lesions such as prostate calcification, prostate enlargement, and prostate inflammation cause too many problems for men’s health. This research proposes a novel approach, a combination of image processing techniques and deep learning methods for classification and segmentation of the prostate in CT-scan images by considering the experienced physicians’ reports.

Due to the various symptoms and nature of these lesions, a three-phases innovative approach has been implemented. In the first phase, using Mask R-CNN, in the second phase, considering the age of each patient and comparison with the standard size of the prostate gland, and finally, using the morphology features, the presence of three common non-cancerous lesions in the prostate gland has investigated.

A hierarchical multitask approach is introduced and the final amount of classification, localization, and segmentation loss is 1%, 1%, and 7%, respectively. Eventually, the overall loss ratio of the model is about 9%.
Originality: In this study, a medical assistant approach is introduced to increase diagnosis process accuracy and reduce error using a real dataset of abdominal and pelvics’ CT scans and the physicians’ reports for each image. A multi-tasks convolutional neural network; also presented to perform localization, classification, and segmentation of the prostate gland in CT scans at the same time.