其他分享
首页 > 其他分享> > Dice Loss与mIoU

Dice Loss与mIoU

作者：互联网

Dice系数和mIoU是语义分割的评价指标，在这里进行了简单知识介绍。

一、Dice系数

1.1 概念理解

Dice系数源于二分类，本质上是衡量两个样本的重叠部分，是一种集合相似度度量函数，该指标范围从0到1，其中“1”表示完整的重叠。 其计算公式为：
在这里插入图片描述
其中 |X∩Y| 是X和Y之间的交集，|X|和|Y|分表表示X和Y的元素的个数，其中，分子的系数为2，是因为分母存在重复计算X和Y之间的共同元素的原因。

对于语义分割问题而言，X-GT分割图像(Ground Truth)，Y-Pred分割图像。

看图一直观理解一下
在这里插入图片描述
公式就可以理解为
在这里插入图片描述

1.2 实际计算

首先将 |X∩Y| 近似为预测图pred和label GT 之间的点乘，并将点乘的元素的结果相加：

（1）预测分割图与 GT 分割图的点乘：
在这里插入图片描述
（2）逐元素相乘的结果元素的相加和：
在这里插入图片描述
对于二分类问题，GT分割图是只有0,1两个值的，因此 |X∩Y| 可以有效的将在 Pred 分割图中未在 GT 分割图中激活的所有像素清零. 对于激活的像素，主要是惩罚低置信度的预测，较高值会得到更好的 Dice 系数。
(3）计算|X|和|Y|，这里可以采用直接元素相加，也可以采用元素平方求和的方法：
在这里插入图片描述

1.3 PyTorch代码

def dice_coeff(pred, target):
    smooth = 1.
    num = pred.size(0)
    m1 = pred.view(num, -1)  # Flatten
    m2 = target.view(num, -1)  # Flatten
    intersection = (m1 * m2).sum()
    return (2. * intersection + smooth) / (m1.sum() + m2.sum() + smooth)

二、mIoU

2.1 语义分割元素分类的四种情况

true positive（TP）：预测正确, 预测结果是正类, 真实是正类
false positive（FP）：预测错误, 预测结果是正类, 真实是负类
true negative（TN）：预测错误, 预测结果是负类, 真实是正类
false negative（FN）：预测正确, 预测结果是负类, 真实是负类
TN(真负): 预测正确, 预测结果是负类, 真实是负类 #跟类别1无关,所以不包含在并集中

2.2 MloU定义与单个IoU理解

计算真实值和预测值两个集合的交集和并集之比。这个比例可以变形为TP（交集）比上TP、FP、FN之和（并集）。即：MloU=TP/(FP+FN+TP)。
MloU一般都是基于类进行计算的，将每一类的IoU计算之后累加，再进行平均，得到的就是基于全局的评价。
Mean Intersection over Union(MloU, 均交并比):为语义分割的标准度量。其计算两个集合的交并比，在语义分割的问题中，这两个集合为真实值(ground truth) 和预测值(predicted segmentation) 。计算公式如下:
i表示真实值，j表示预测值，Pij 表示将i预测为j。
在这里插入图片描述等价于：
在这里插入图片描述
直观理解：
在这里插入图片描述MloU：计算两圆交集（橙色部分）与两圆并集（红色+橙色+黄色）之间的比例，理想情况下两圆重合，比例为1。

2.3 MloU计算

(1)首先是计算混淆矩阵(误差矩阵)

(2)计算MloU
参考链接中的代码部分：https://blog.csdn.net/u012370185/article/details/94409933
github：https://github.com/dilligencer-zrj/code_zoo/blob/master/compute_mIOU

三、cross entropy 交叉熵

图像分割中最常用的损失函数是逐像素交叉熵损失。该损失函数分别检查每个像素，将类预测(深度方向的像素向量)与我们的热编码目标向量进行比较。
在这里插入图片描述
由此可见，交叉熵的损失函数单独评估每个像素矢量的类预测，然后对所有像素求平均值，所以我们可以认为图像中的像素被平等的学习了。但是，医学图像中常出现类别不均衡（class imbalance）的问题，由此导致训练会被像素较多的类主导，对于较小的物体很难学习到其特征，从而降低网络的有效性。

有较多的文章对其进行了研究，包括 Long et al. 的 FCN 在每个通道加权该损失，从而抵消数据集中存在的类别不均的问题。同时，Ronneberger et al.提出的 U-Net 提出了新的逐像素损失的加权方案，使其在分割对象的边界处具有更高的权重。该损失加权方案以不连续的方式帮助他们的 U-Net 模型细分生物医学图像中的细胞，使得可以在二元分割图中容易地识别单个细胞。

四、Dice Loss

Dice Loss的计算公式非常简单如下：
在这里插入图片描述
这种损失函数被称为 Soft Dice Loss，因为我们直接使用预测概率而不是使用阈值或将它们转换为二进制mask。

Soft Dice Loss 将每个类别分开考虑，然后平均得到最后结果。比较直观如图所示。
在这里插入图片描述
需要注意的是Dice Loss存在两个问题：

(1) 训练误差曲线非常混乱，很难看出关于收敛的信息。尽管可以检查在验证集上的误差来避开此问题。

(2) Dice Loss比较适用于样本极度不均的情况，一般的情况下，使用 Dice Loss 会对反向传播造成不利的影响，容易使训练变得不稳定。

所以在一般情况下，还是使用交叉熵损失函数。

PyTorch参考代码

import torch.nn as nn
import torch.nn.functional as F
 
 
class SoftDiceLoss(nn.Module):
    def __init__(self, weight=None, size_average=True):
        super(SoftDiceLoss, self).__init__()
 
    def forward(self, logits, targets):
        num = targets.size(0)
        smooth = 1
        probs = F.sigmoid(logits)
        m1 = probs.view(num, -1)
        m2 = targets.view(num, -1)
        intersection = (m1 * m2)
        score = 2. * (intersection.sum(1) + smooth) / (m1.sum(1) + m2.sum(1) + smooth)
        score = 1 - score.sum() / num
        return score

参考链接：

https://zhuanlan.zhihu.com/p/86704421
https://blog.csdn.net/weixin_43346901/article/details/99880278
https://blog.csdn.net/gjk0223/article/details/2314844
https://blog.csdn.net/Biyoner/article/details/84728417
https://www.aiuai.cn/aifarm1159.html
https://zhuanlan.zhihu.com/p/88805121
https://blog.csdn.net/u012370185/article/details/94409933

标签：Loss,分割,Dice,像素,https,mIoU,预测
来源： https://blog.csdn.net/weixin_43593330/article/details/113508561