A Surprisingly Effective Perimeter-based Loss for Medical Image Segmentation - Normandie Université Access content directly
Conference Papers Year : 2021

A Surprisingly Effective Perimeter-based Loss for Medical Image Segmentation

Abstract

Deep convolutional networks recently made many breakthroughs in medical image segmentation. Still, some anatomical artefacts may be observed in the segmentation results, with holes or inaccuracies near the object boundaries. To address these issues, loss functions that incorporate constraints, such as spatial information or prior knowledge, have been introduced. An example of such prior losses are the contour-based losses, which exploit distance maps to conduct point-by-point optimization between ground-truth and predicted contours. However, such losses may be computationally expensive or susceptible to trivial local solutions and vanishing gradient problems. Moreover, they depend on distance maps which tend to underestimate the contour-to-contour distances. We propose a novel loss constraint that optimizes the perimeter length of the segmented object relative to the ground-truth segmentation. The novelty lies in computing the perimeter with a soft approximation of the contour of the probability map via specialized non-trainable layers in the network. Moreover, we optimize the mean squared error between the predicted perimeter length and ground-truth perimeter length. This soft optimization of contour boundaries allows the network to take into consideration border irregularities within organs while still being efficient. Our experiments on three public datasets (spleen, hippocampus and cardiac structures) show that the proposed method outperforms state-of-the-art boundary losses for both single and multi-organ segmentation.
Fichier principal
Vignette du fichier
el-jurdi21a.pdf (729.35 Ko) Télécharger le fichier
Origin : Publisher files allowed on an open archive

Dates and versions

hal-03410709 , version 1 (03-11-2021)

Identifiers

  • HAL Id : hal-03410709 , version 1

Cite

Rosana El Jurdi, Caroline Petitjean, Paul Honeine, Veronika Cheplygina, Fahed Abdallah. A Surprisingly Effective Perimeter-based Loss for Medical Image Segmentation. Fourth conference on Medical Imaging with Deep Learning (MIDL), Jul 2021, Lübeck, Germany. ⟨hal-03410709⟩
75 View
113 Download

Share

Gmail Facebook X LinkedIn More