Department of Mathematics, Nanjing University of Science and Technology, Nanjing, China.
Department of Medical Biophysics, University of Toronto, Toronto, Canada.
Med Image Anal. 2021 Jul;71:102035. doi: 10.1016/j.media.2021.102035. Epub 2021 Mar 19.
The loss function is an important component in deep learning-based segmentation methods. Over the past five years, many loss functions have been proposed for various segmentation tasks. However, a systematic study of the utility of these loss functions is missing. In this paper, we present a comprehensive review of segmentation loss functions in an organized manner. We also conduct the first large-scale analysis of 20 general loss functions on four typical 3D segmentation tasks involving six public datasets from 10+ medical centers. The results show that none of the losses can consistently achieve the best performance on the four segmentation tasks, but compound loss functions (e.g. Dice with TopK loss, focal loss, Hausdorff distance loss, and boundary loss) are the most robust losses. Our code and segmentation results are publicly available and can serve as a loss function benchmark. We hope this work will also provide insights on new loss function development for the community.
损失函数是基于深度学习的分割方法中的一个重要组成部分。在过去的五年中,已经提出了许多用于各种分割任务的损失函数。然而,这些损失函数的实用性缺乏系统的研究。在本文中,我们以一种有组织的方式对分割损失函数进行了全面的回顾。我们还首次对四个典型的 3D 分割任务中的 20 个通用损失函数进行了大规模分析,涉及来自 10 多个医疗中心的六个公共数据集。结果表明,没有一个损失函数可以在四个分割任务上始终取得最佳性能,但复合损失函数(例如,Dice with TopK 损失、焦点损失、Hausdorff 距离损失和边界损失)是最稳健的损失函数。我们的代码和分割结果是公开的,可以作为损失函数基准。我们希望这项工作也能为社区提供关于新损失函数开发的见解。