Probabilistic deep learning for instance segmentation

Item Type:

Article

Title:

Creators Name:

Rumberger, J.L. and Mais, L. and Kainmueller, D.

Abstract:

Probabilistic convolutional neural networks, which predict distributions of predictions instead of point estimates, led to recent advances in many areas of computer vision, from image reconstruction to semantic segmentation. Besides state of the art benchmark results, these networks made it possible to quantify local uncertainties in the predictions. These were used in active learning frameworks to target the labeling efforts of specialist annotators or to assess the quality of a prediction in a safety-critical environment. However, for instance segmentation problems these methods are not frequently used so far. We seek to close this gap by proposing a generic method to obtain model-inherent uncertainty estimates within proposal-free instance segmentation models. Furthermore, we analyze the quality of the uncertainty estimates with a metric adapted from semantic segmentation. We evaluate our method on the BBBC010 C. elegans dataset, where it yields competitive performance while also predicting uncertainty estimates that carry information about object-level inaccuracies like false splits and false merges. We perform a simulation to show the potential use of such uncertainty estimates in guided proofreading.

Keywords:

Instance Segmentation, Probabilistic Deep Learning, Bayesian Inference, Digital Microscopy

Source:

Lecture Notes in Computer Science

Series Name:

Lecture Notes in Computer Science

Title of Book:

Computer Vision - ECCV 2020 Workshops

ISSN:

0302-9743

ISBN:

978-3-030-66415-2

Publisher:

Springer

Volume:

12535

Page Range:

445-457

Date:

2021

Official Publication:

https://doi.org/10.1007/978-3-030-66415-2_29

Related to:

URL	URL Type
https://edoc.mdc-berlin.de/19484/	Preprint version

Repository Staff Only: item control page