1. Inter-site Variability in Prostate Segmentation Accuracy Using Deep Learning
- Author
-
Dean C. Barratt, Caroline M. Moore, Hashim U. Ahmed, Henkjan J. Huisman, Mark Emberton, Yipeng Hu, Eli Gibson, and Nooshin Ghavami
- Subjects
Training set ,Computer science ,business.industry ,Deep learning ,education ,Pattern recognition ,030218 nuclear medicine & medical imaging ,03 medical and health sciences ,0302 clinical medicine ,Medical imaging ,Segmentation ,Artificial intelligence ,business ,030217 neurology & neurosurgery ,Prostate segmentation - Abstract
Deep-learning-based segmentation tools have yielded higher reported segmentation accuracies for many medical imaging applications. However, inter-site variability in image properties can challenge the translation of these tools to data from ‘unseen’ sites not included in the training data. This study quantifies the impact of inter-site variability on the accuracy of deep-learning-based segmentations of the prostate from magnetic resonance (MR) images, and evaluates two strategies for mitigating the reduced accuracy for data from unseen sites: training on multi-site data and training with limited additional data from the unseen site. Using 376 T2-weighted prostate MR images from six sites, we compare the segmentation accuracy (Dice score and boundary distance) of three deep-learning-based networks trained on data from a single site and on various configurations of data from multiple sites. We found that the segmentation accuracy of a single-site network was substantially worse on data from unseen sites than on data from the training site. Training on multi-site data yielded marginally improved accuracy and robustness. However, including as few as 8 subjects from the unseen site, e.g. during commissioning of a new clinical system, yielded substantial improvement (regaining 75% of the difference in Dice score).
- Published
- 2018
- Full Text
- View/download PDF