Epistemic and heteroscedastic uncertainty estimation in retinal blood vessel segmentation

0Citations
Citations of this article
7Readers
Mendeley users who have this article in their library.

Abstract

Current state-of-the-art medical image segmentation methods require high quality datasets to obtain good performance. However, medical specialists often disagree on diagnosis, hence, datasets contain contradictory annotations. This, in turn, leads to difficulties in the optimization process of Deep Learning models and hinder performance. We propose a method to estimate uncertainty in Convolutional Neural Network (CNN) segmentation models, that makes the training of CNNs more robust to contradictory annotations. In this work, we model two types of uncertainty, heteroscedastic and epistemic, without adding any additional supervisory signal other than the ground-truth segmentation mask. As expected, the uncertainty is higher closer to vessel boundaries, and on top of thinner and less visible vessels where it is more likely for medical specialists to disagree. Therefore, our method is more suitable to learn from datasets created with heterogeneous annotators. We show that there is a correlation between the uncertainty estimated by our method and the disagreement in the segmentation provided by two different medical specialists. Furthermore, by explicitly modeling the uncertainty, the Intersection over Union of the segmentation network improves 5.7 percentage points.

Cite

CITATION STYLE

APA

Costa, P., Smailagic, A., Cardoso, J. S., & Campilho, A. (2021). Epistemic and heteroscedastic uncertainty estimation in retinal blood vessel segmentation. U.Porto Journal of Engineering, 7(3), 93–100. https://doi.org/10.24840/2183-6493_007.003_0008

Register to see more suggestions

Mendeley helps you to discover research relevant for your work.

Already have an account?

Save time finding and organizing research with Mendeley

Sign up for free