Probabilistic deep learning for instance segmentation

Item Type:

Preprint

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 Microscop

Source:

arXiv

Publisher:

Cornell University

Article Number:

2008.10678

Date:

24 August 2020

Official Publication:

https://arxiv.org/abs/arXiv:2008.10678

Related to:

URL	URL Type
https://edoc.mdc-berlin.de/19870/	Final version

Repository Staff Only: item control page