Get our free extension to see links to code for papers anywhere online!

Chrome logo  Add to Chrome

Firefox logo Add to Firefox

Disentangled Uncertainty and Out of Distribution Detection in Medical Generative Models

Add code

Nov 11, 2022
Kumud Lakara, Matias Valdenegro-Toro

Share this with someone who'll enjoy it:

Trusting the predictions of deep learning models in safety critical settings such as the medical domain is still not a viable option. Distentangled uncertainty quantification in the field of medical imaging has received little attention. In this paper, we study disentangled uncertainties in image to image translation tasks in the medical domain. We compare multiple uncertainty quantification methods, namely Ensembles, Flipout, Dropout, and DropConnect, while using CycleGAN to convert T1-weighted brain MRI scans to T2-weighted brain MRI scans. We further evaluate uncertainty behavior in the presence of out of distribution data (Brain CT and RGB Face Images), showing that epistemic uncertainty can be used to detect out of distribution inputs, which should increase reliability of model outputs.

* 3.5 pages, with appendix. Medical Imaging Meets NeurIPS Workshop 2022 Camera Ready 

   Access Paper Source

Share this with someone who'll enjoy it: