Distance Map Loss Penalty Term for Semantic Segmentation

08/10/2019
by   Francesco Caliva, et al.
0

Convolutional neural networks for semantic segmentation suffer from low performance at object boundaries. In medical imaging, accurate representation of tissue surfaces and volumes is important for tracking of disease biomarkers such as tissue morphology and shape features. In this work, we propose a novel distance map derived loss penalty term for semantic segmentation. We propose to use distance maps, derived from ground truth masks, to create a penalty term, guiding the network's focus towards hard-to-segment boundary regions. We investigate the effects of this penalizing factor against cross-entropy, Dice, and focal loss, among others, evaluating performance on a 3D MRI bone segmentation task from the publicly available Osteoarthritis Initiative dataset. We observe a significant improvement in the quality of segmentation, with better shape preservation at bone boundaries and areas affected by partial volume. We ultimately aim to use our loss penalty term to improve the extraction of shape biomarkers and derive metrics to quantitatively evaluate the preservation of shape.

READ FULL TEXT

page 2

page 3

research
02/04/2021

Active Boundary Loss for Semantic Segmentation

This paper proposes a novel active boundary loss for semantic segmentati...
research
09/23/2020

Learning Non-Unique Segmentation with Reward-Penalty Dice Loss

Semantic segmentation is one of the key problems in the field of compute...
research
07/17/2020

Edge-Preserving Guided Semantic Segmentation for VIPriors Challenge

Semantic segmentation is one of the most attractive research fields in c...
research
10/31/2021

Incorporating Boundary Uncertainty into loss functions for biomedical image segmentation

Manual segmentation is used as the gold-standard for evaluating neural n...
research
04/06/2021

InverseForm: A Loss Function for Structured Boundary-Aware Segmentation

We present a novel boundary-aware loss term for semantic segmentation us...
research
04/09/2023

On the dice loss gradient and the ways to mimic it

In the past few years, in the context of fully-supervised semantic segme...

Please sign up or login with your details

Forgot password? Click here to reset