Uncertainty Estimation in Medical Imaging

For uncertainty estimation for a machine-learned model prediction in medical imaging, a distribution in latent space is sampled to determine uncertainty in machine-learned model prediction. For example, a segmentation is output by a machine-learned model in response to input of multi-parametric data. A latent space generated by the machine-learned model is used to estimate the uncertainty of the segmentation, such as a segmentation of a prostate lesion. The model of Kohl, et al., may be used as the variational auto encoder generates a latent space representing the distribution of the training data, which latent space may be used to determine uncertainty.