A Surprisingly Effective Perimeter-based Loss for Medical Image Segmentation

Rosana El Jurdi (1, 2) , Caroline Petitjean (1) , Paul Honeine (1) , Veronika Cheplygina (3, 4) , Fahed Abdallah (2, 5)

1 DocApp - LITIS - Equipe Apprentissage
2 Université Libanaise
3 ITU - IT University of Copenhagen
4 TU/e - Eindhoven University of Technology [Eindhoven]
5 LM2S - Laboratoire Modélisation et Sûreté des Systèmes

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.

Keywords: Convolutional Neural Networks; Medical Image Segmentation; Prior-based Losses; Perimeter length constraint

Domains: Statistics [stat]; Machine Learning [stat.ML]; Engineering Sciences [physics]; Signal and Image processing; Mathematics [math]; Statistics [math.ST]; Computer Science [cs]; Signal and Image Processing; Computer Science [cs]; Neural and Evolutionary Computing [cs.NE]; Computer Science [cs]; Machine Learning [cs.LG]; Computer Science [cs]; Computers and Society [cs.CY]; Computer Science [cs]; Computer Vision and Pattern Recognition [cs.CV]; Computer Science [cs]; Artificial Intelligence [cs.AI]

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⟩