1,引言
对于图像分割,通过模型的优化提高准确率一直是大家推进的重点,目标损失函数作为算法求解重要的一部分,在帮助模型快速收敛方面发挥着重要的作用。
文章参考:

图像分割可以定义为像素级别的分类任务。图像由各种像素组成,这些像素组合在一起定义了图像中的不同元素,因此将这些像素分类为一类元素的方法称为语义图像分割。在设计基于复杂图像分割的深度学习架构时,通常会遇到了一个至关重要的选择,即选择哪个损失/目标函数,因为它们会激发算法的学习过程。损失函数的选择对于任何架构学习正确的目标都是至关重要的。

这是一篇关于图像分割损失函数的总结,一共总结了14种用于分割的损失函数。并将他们分成4类。
基于分布的损失函数,基于区域的损失函数,基于边界的损失函数和基于复合的损失函数( Distribution-based,Region-based, Boundary-based, and Compounded)。

论文地址:https://arxiv.org/pdf/2006.14822.pdf
代码地址:https://github.com/shruti-jadon/Semantic-Segmentation-Loss-Functions
项目推荐:https://github.com/JunMa11/SegLoss
2,基于分布
2.1 二进制交叉熵损失函数 Binary Cross-Entropy
交叉熵定义为对给定随机变量或事件集的两个概率分布之间的差异的度量。它被广泛用于分类任务,并且由于分割是像素级分类,因此效果很好。在多分类任务中,经常采用 softmax 激活函数+交叉熵损失函数,因为交叉熵描述了两个概率分布的差异,然而神经网络输出的是向量,并不是概率分布的形式。所以需要 softmax激活函数将一个向量进行“归一化”成概率分布的形式,再采用交叉熵损失函数计算 loss。
交叉熵的损失函数单独评估每个像素矢量的类预测,然后对所有像素求平均值,所以我们可以认为图像中的像素被平等的学习了。但是,医学图像中常出现类别不均衡(class imbalance)的问题,由此导致训练会被像素较多的类主导,对于较小的物体很难学习到其特征,从而降低网络的有效性。
最好的使用场景是不同的类别间有均匀分布的数据,是基于伯努利分布的损失函数。
2.2 加权二元交叉熵Weighted Binary Cross-Entropy
**加权二元交叉熵(WCE)**是二元交叉熵变量的一个变种。在这个例子中,正的例子被一些系数加权。它广泛用于倾斜数据的情况。
加入的权值用于调整假阴性和假阳性。如果你想减少假阴性的数量就设置权值大于1,同样地,如果想要减少假阳性的数量就设置权值小于1.权值是用于正例的系数。
应用场景:广泛应用于偏斜数据集,通过系数来衡量
2.3 平衡交叉熵 Balanced Cross-Entropy
平衡交叉熵(BCE)类似于加权交叉熵。唯一的区别是,除了正例子,我们还对负示例进行了加权。
应用场景:与加权交叉熵类似,广泛应用于倾斜数据集,它分别对正样本和负样本进行加权。
2.4 焦点损失Focal Loss
焦损(FL)也可以看作二元交叉熵的变化。它降低了简

文章总结了14种图像分割的损失函数,包括二元交叉熵、加权二元交叉熵、平衡交叉熵、焦点损失等,并按照分布、区域、边界和复合四类进行分类。这些损失函数在处理像素级别分类任务,尤其是面对类别不平衡问题时有不同的效果和应用场景。例如,FocalLoss适合高度不平衡的数据集,而DiceLoss和TverskyLoss则考虑了真阳性和真阴性的平衡。
983

被折叠的 条评论
为什么被折叠?



