SS3D: Unsupervised Out-of-Distribution Detection and Localization for Medical Volumes

Doorenbos, Lars Jelte; Sznitman, Raphael; Márquez Neila, Pablo (2022). SS3D: Unsupervised Out-of-Distribution Detection and Localization for Medical Volumes. In: MICCAI 2021: Biomedical Image Registration, Domain Generalisation and Out-of-Distribution Analysis. Proceedings. Lecture Notes in Computer Science: Vol. 13166 (pp. 111-118). Cham: Springer 10.1007/978-3-030-97281-3_17

[img] Text
Doorenbos2022_Chapter_SS3DUnsupervisedOut-of-Distrib.pdf - Published Version
Restricted to registered users only
Available under License Publisher holds Copyright.

Download (904kB) | Request a copy

We present an extension of the self-supervised outlier detection (SSD) framework to the three-dimensional case. We first apply contrastive learning on a network using a general dataset of two-dimensional slices randomly sampled from all the available training data. This network serves as a latent embedding encoder of the input images. We model the in-distribution latent density as a multivariate Gaussian, fitted to the embeddings of the training slices. At test time, each test sample is scored by summing the Mahalanobis distances from all its slices to the means of the learned Gaussians. While mainly meant as a sample-level method, this approach additionally enables coarse localization, scoring each voxel by the minimum Mahalanobis distance among the slices that contain it. On the sample-level task of the 2021 MICCAI Medical Out-of-Distribution Analysis Challenge, our method ranked second on the challenging abdominal dataset, and fourth overall. Moreover, we show that with pretrained features and the right choice of architecture, a further boost in performance can be gained.

Item Type:

Conference or Workshop Item (Paper)

Division/Institute:

10 Strategic Research Centers > ARTORG Center for Biomedical Engineering Research > ARTORG Center - AI in Medical Imaging Laboratory

Graduate School:

Graduate School for Cellular and Biomedical Sciences (GCB)

UniBE Contributor:

Doorenbos, Lars Jelte, Sznitman, Raphael, Márquez Neila, Pablo

ISBN:

978-3-030-97281-3

Series:

Lecture Notes in Computer Science

Publisher:

Springer

Language:

English

Submitter:

Lars Jelte Doorenbos

Date Deposited:

14 Apr 2022 09:52

Last Modified:

06 Dec 2022 16:27

Publisher DOI:

10.1007/978-3-030-97281-3_17

BORIS DOI:

10.48350/168297

URI:

https://boris.unibe.ch/id/eprint/168297

Actions (login required)

Edit item Edit item
Provide Feedback