Training Convolutional Networks for Prostate Segmentation with Limited Data

Sara L. Saunders, Ethan Leng, Benjamin Spilseth, Neil Wasserman, Gregory J. Metzger, Patrick J. Bolan

Research output: Contribution to journalArticlepeer-review

2 Scopus citations

Abstract

Multi-zonal segmentation is a critical component of computer-aided diagnostic systems for detecting and staging prostate cancer. Previously, convolutional neural networks such as the U-Net have been used to produce fully automatic multi-zonal prostate segmentation on magnetic resonance images (MRIs) with performance comparable to human experts, but these often require large amounts of manually segmented training data to produce acceptable results. For institutions that have limited amounts of labeled MRI exams, it is not clear how much data is needed to train a segmentation model, and which training strategy should be used to maximize the value of the available data. This work compares how the strategies of transfer learning and aggregated training using publicly available external data can improve segmentation performance on internal, site-specific prostate MR images, and evaluates how the performance varies with the amount of internal data used for training. Cross training experiments were performed to show that differences between internal and external data were impactful. Using a standard U-Net architecture, optimizations were performed to select between 2D and 3D variants, and to determine the depth of fine-tuning required for optimal transfer learning. With the optimized architecture, the performance of transfer learning and aggregated training were compared for a range of 5-40 internal datasets. The results show that both strategies consistently improve performance and produced segmentation results that are comparable to that of human experts with approximately 20 site-specific MRI datasets. These findings can help guide the development of site-specific prostate segmentation models for both clinical and research applications.

Original languageEnglish (US)
Article number9499065
Pages (from-to)109214-109223
Number of pages10
JournalIEEE Access
Volume9
DOIs
StatePublished - Jul 27 2021

Bibliographical note

Funding Information:
This work was supported in part by the National Institutes of Health under Grant P41 EB027061, in part by the U.S. Department of Defense under Grant W81XWH-15-1-0477, and in part by the University of Minnesota and NIH MN-REACH Program.

Publisher Copyright:
© 2013 IEEE.

Keywords

  • 3D U-Net
  • Automatic prostate segmentation
  • convolutional neural networks
  • magnetic resonance imaging

Fingerprint

Dive into the research topics of 'Training Convolutional Networks for Prostate Segmentation with Limited Data'. Together they form a unique fingerprint.

Cite this