Know Your Space - Inlier and Outlier Construction for Calibrating Medical OOD Detectors

Vivek Narayanaswamy * 1

Yamen Mubarka * 1

Rushil Anirudh 1

Deepta Rajan 2

Andreas Spanias 3

Jayaraman J. Thiagarajan 1


1Lawrence Livermore National Laboratory, CA, USA

2Microsoft, WA, USA

3Arizona State University, AZ, USA



Summary

Our primary focus is on developing well-calibrated out-of-distribution (OOD) detectors to ensure the safe deployment of medical image classifiers. The use of synthetic augmentations has become common for specifying regimes of data inliers and outliers. However, our research findings highlight the substantial influence of both the synthesis space and the type of augmentation on the performance of OOD detectors. After conducting an extensive study using medical imaging benchmarks and open-set recognition settings, we recommend employing a combination of virtual inliers in the classifier's latent space and diverse synthetic outliers in the pixel space. This approach proves highly effective in producing OOD detectors with superior performance.

Method

Alt text

Empirical Results

Citation

  @inproceedings{narayanaswamy2023know,
title={Know Your Space: Inlier and Outlier Construction for Calibrating Medical {OOD} Detectors},
author={Vivek Narayanaswamy and Yamen Mubarka and Rushil Anirudh and Deepta Rajan and Andreas Spanias and Jayaraman J. Thiagarajan},
booktitle={Medical Imaging with Deep Learning},
year={2023}}

Contact

If you have any questions, please feel free to contact us via email: narayanaswam1@llnl.gov; jjayaram@llnl.gov