Deep learning in magnetic resonance prostate segmentation: A review and a new perspective

Abstract

Prostate radiotherapy is a well established curative oncology modality, which in future will use Magnetic Resonance Imaging (MRI)-based radiotherapy for daily adaptive radiotherapy target definition. However the time needed to delineate the prostate from MRI data accurately is a time consuming process. Deep learning has been identified as a potential new technology for the delivery of precision radiotherapy in prostate cancer, where accurate prostate segmentation helps in cancer detection and therapy. However, the trained models can be limited in their application to clinical setting due to different acquisition protocols, limited publicly available datasets, where the size of the datasets are relatively small. Therefore, to explore the field of prostate segmentation and to discover a generalisable solution, we review the state-of-the-art deep learning algorithms in MR prostate segmentation; provide insights to the field by discussing their limitations and strengths; and propose an optimised 2D U-Net for MR prostate segmentation. We evaluate the performance on four publicly available datasets using Dice Similarity Coefficient (DSC) as performance metric. Our experiments include within dataset evaluation and cross-dataset evaluation. The best result is achieved by composite evaluation (DSC of 0.9427 on Decathlon test set) and the poorest result is achieved by cross-dataset evaluation (DSC of 0.5892, Prostate X training set, Promise 12 testing set). We outline the challenges and provide recommendations for future work. Our research provides a new perspective to MR prostate segmentation and more importantly, we provide standardised experiment settings for researchers to evaluate their algorithms. Our code is available at https://github.com/AIEMMU/MRI\_Prostate.

10 pages

References in corpus (14)

• Very Deep Convolutional Networks for Large-Scale Image Recognition
• mixup: Beyond Empirical Risk Minimization
• Attention U-Net: Learning Where to Look for the Pancreas
• MultiResUNet : Rethinking the U-Net Architecture for Multimodal Biomedical Image Segmentation
• Confounding variables can degrade generalization performance of radiological deep learning models
• Understanding deep learning requires rethinking generalization
• A large annotated medical image dataset for the development and evaluation of segmentation algorithms
• On the Variance of the Adaptive Learning Rate and Beyond
• Manifold Mixup: Better Representations by Interpolating Hidden States
• Recurrent Residual Convolutional Neural Network based on U-Net (R2U-Net) for Medical Image Segmentation
• nnU-Net: Self-adapting Framework for U-Net-Based Medical Image Segmentation
• Deep learning to achieve clinically applicable segmentation of head and neck anatomy for radiotherapy
• Mish: A Self Regularized Non-Monotonic Activation Function
• Semantic-guided Encoder Feature Learning for Blurry Boundary Delineation

Cited by in corpus (2)

• The use of deep learning in interventional radiotherapy (brachytherapy): a review with a focus on open source and open data
• 3D-OOCS: Learning Prostate Segmentation with Inductive Bias