1. Deep Learning-Based Methods for Prostate Segmentation in Magnetic Resonance Imaging
- Author
-
Albert Comelli 1, 2, Navdeep Dahiya 3, Alessandro Stefano 2, Federica Vernuccio 4, Marzia Portoghese 4, Giuseppe Cutaia 4, Alberto Bruno 4, Giuseppe Salvaggio 4, Anthony Yezzi 3, Comelli A., Dahiya N., Stefano A., Vernuccio F., Portoghese M., Cutaia G., Bruno A., Salvaggio G., and Yezzi A.
- Subjects
Computer science ,Graphics processing unit ,02 engineering and technology ,Residual ,lcsh:Technology ,Article ,030218 nuclear medicine & medical imaging ,lcsh:Chemistry ,deep learning ,segmentation ,prostate ,MRI ,ENet ,UNet ,ERFNet ,radiomics ,Set (abstract data type) ,03 medical and health sciences ,0302 clinical medicine ,0202 electrical engineering, electronic engineering, information engineering ,General Materials Science ,Segmentation ,lcsh:QH301-705.5 ,Instrumentation ,Settore ING-INF/05 - Sistemi Di Elaborazione Delle Informazioni ,Fluid Flow and Transfer Processes ,Artificial neural network ,lcsh:T ,business.industry ,Process Chemistry and Technology ,Deep learning ,General Engineering ,Process (computing) ,Pattern recognition ,lcsh:QC1-999 ,Computer Science Applications ,lcsh:Biology (General) ,lcsh:QD1-999 ,lcsh:TA1-2040 ,020201 artificial intelligence & image processing ,Artificial intelligence ,Central processing unit ,lcsh:Engineering (General). Civil engineering (General) ,business ,lcsh:Physics - Abstract
Magnetic Resonance Imaging-based prostate segmentation is an essential task for adaptive radiotherapy and for radiomics studies whose purpose is to identify associations between imaging features and patient outcomes. Because manual delineation is a time-consuming task, we present three deep-learning (DL) approaches, namely UNet, efficient neural network (ENet), and efficient residual factorized convNet (ERFNet), whose aim is to tackle the fully-automated, real-time, and 3D delineation process of the prostate gland on T2-weighted MRI. While UNet is used in many biomedical image delineation applications, ENet and ERFNet are mainly applied in self-driving cars to compensate for limited hardware availability while still achieving accurate segmentation. We apply these models to a limited set of 85 manual prostate segmentations using the k-fold validation strategy and the Tversky loss function and we compare their results. We find that ENet and UNet are more accurate than ERFNet, with ENet much faster than UNet. Specifically, ENet obtains a dice similarity coefficient of 90.89% and a segmentation time of about 6 s using central processing unit (CPU) hardware to simulate real clinical conditions where graphics processing unit (GPU) is not always available. In conclusion, ENet could be efficiently applied for prostate delineation even in small image training datasets with potential benefit for patient management personalization.
- Published
- 2021
- Full Text
- View/download PDF