Incorporating Boundary Uncertainty into loss functions for biomedical image segmentation

10/31/2021
by   Michael Yeung, et al.
0

Manual segmentation is used as the gold-standard for evaluating neural networks on automated image segmentation tasks. Due to considerable heterogeneity in shapes, colours and textures, demarcating object boundaries is particularly difficult in biomedical images, resulting in significant inter and intra-rater variability. Approaches, such as soft labelling and distance penalty term, apply a global transformation to the ground truth, redefining the loss function with respect to uncertainty. However, global operations are computationally expensive, and neither approach accurately reflects the uncertainty underlying manual annotation. In this paper, we propose the Boundary Uncertainty, which uses morphological operations to restrict soft labelling to object boundaries, providing an appropriate representation of uncertainty in ground truth labels, and may be adapted to enable robust model training where systematic manual segmentation errors are present. We incorporate Boundary Uncertainty with the Dice loss, achieving consistently improved performance across three well-validated biomedical imaging datasets compared to soft labelling and distance-weighted penalty. Boundary Uncertainty not only more accurately reflects the segmentation process, but it is also efficient, robust to segmentation errors and exhibits better generalisation.

READ FULL TEXT

page 3

page 4

research
10/31/2021

Focal Attention Networks: optimising attention for biomedical image segmentation

In recent years, there has been increasing interest to incorporate atten...
research
10/31/2021

Calibrating the Dice loss to handle neural network overconfidence for biomedical image segmentation

The Dice similarity coefficient (DSC) is both a widely used metric and l...
research
04/12/2021

Spatially Varying Label Smoothing: Capturing Uncertainty from Expert Annotations

The task of image segmentation is inherently noisy due to ambiguities re...
research
11/08/2022

A kinetic approach to consensus-based segmentation of biomedical images

In this work, we apply a kinetic version of a bounded confidence consens...
research
08/10/2019

Distance Map Loss Penalty Term for Semantic Segmentation

Convolutional neural networks for semantic segmentation suffer from low ...
research
11/14/2019

Give me (un)certainty – An exploration of parameters that affect segmentation uncertainty

Segmentation tasks in medical imaging are inherently ambiguous: the boun...
research
12/22/2019

Learning Improved Representations by Transferring Incomplete Evidence Across Heterogeneous Tasks

Acquiring ground truth labels for unlabelled data can be a costly proced...

Please sign up or login with your details

Forgot password? Click here to reset