On the dice loss gradient and the ways to mimic it
Not on the map. Why?
Abstract (arXiv)
In the past few years, in the context of fully-supervised semantic segmentation, several losses -- such as cross-entropy and dice -- have emerged as de facto standards to supervise neural networks. The Dice loss is an interesting case, as it comes from the relaxation of the popular Dice coefficient; one of the main evaluation metric in medical imaging applications. In this paper, we first study theoretically the gradient of the dice loss, showing that concretely it is a weighted negative of the ground truth, with a very small dynamic range. This enables us, in the second part of this paper, to mimic the supervision of the dice loss, through a simple element-wise multiplication of the network output with a negative of the ground truth. This rather surprising result sheds light on the practical supervision performed by the dice loss during gradient descent. This can help the practitioner to understand and interpret results while guiding researchers when designing new losses.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 7 of its 16 references that are in the map. First posted on arXiv 9 Apr 2023.
Builds on
7 of 16 references.
References · 7
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2016V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentationcited by 2,292
- 2017Generalised Dice overlap as a deep learning loss function for highly unbalanced segmentationscited by 620
- 2020Calibrating Deep Neural Networks using Focal Losscited by 357
- 2016ENet: A Deep Neural Network Architecture for Real-Time Semantic Segmentationcited by 291
- 2018Boundary loss for highly unbalanced segmentationcited by 190
- 2020Optimization for Medical Image Segmentation: Theory and Practice when evaluating with Dice Score or Jaccard Indexcited by 52
Built on by
Not cited yet.