Focal Loss

Kullback-Leibler Divergence（相对熵）

KL散度（Kullback-Leibler Divergence，也称相对熵）用来衡量两个分布之间的偏差，其可以用下面数学公式描述：
$$\begin{array}{rl}{D}_{KL}(p(x),q(x))& =\sum _i^N{p}_i(x)\log \left(\frac{p_i(x)}{q_i(x)}\right)\\ & =\sum _i^N{p}_i(x)\log p_i(x)-p_i(x)\log q_i(x)\\ & =-H(p)+H(p,q)\end{array}$$

NLL Loss

Cross Entropy Loss

上面公式中的 $CE(p,q)$ 即为交叉熵 (Cross-Entropy) 函数，通常在机器学习中，$p(x)$ 表示为目标分布（可以理解为label），$q(x)$ 表示模型输出分布（可以理解为prediction），监督学习的目标是尽可能使得 $q(x)$ 接近 $p(x)$ 从而达到最佳性能，即 $q(x)$ 和 $p(x)$ 偏差越大，KL散度值越大，当两者分布相等时，KL散度为零，同时目标分布确定时 $p(x)$ 分布的熵 $H(p)$ 也是确定的，因此通过KL散度进行优化等价于通过交叉熵进行优化。
$$CE\_loss(p,q)=-\sum _i^N{p}_i(x)\log q_i(x)$$

如何计算交叉熵？

以二分类为例：这里有1个样本，$y$表示真实值，$\hat{y}$表示预测为1的概率值，

	$y$	$\hat{y}$
sample 1	0	0.1

$$\begin{array}{rl}C{E}_{loss}& =-\left(0\ast \log (1-0.1)+1\ast \log (0.1)\right)\\ & =2.3025\end{array}$$
使用代码验证如下：

import torch
import math
print(-(0*math.log(0.9)+1*math.log(0.1)))

BCE_loss = torch.nn.BCELoss()
target = torch.tensor([0]).float()
output = torch.tensor([0.9]).float()
print(BCE_loss(output, target))

同理，接入对于一个三分类（多分类）任务，一个样本的预测值为 $(0.1,0.3,0.6)$，其对应的 gt 为 (0, 0, 1)，则其交叉熵损失为：$-(log(1-0.1)+log(1-0.3)+log(0.6))$，但torch.nn.CrossEntropyLoss()的计算方式略显不同（它是 softmax+log+NLLLoss的合体）。

Balanced Cross Entropy Loss

在多类别数据中，如果某类别之间分布不均衡，直接训练会导致模型过多关注样本比例较多的类别，而在样本比例较少的类别上分类性能不足，因此Balanced Cross Entropy Loss提出是为了缓解这个问题，其根据每个类别在数据中的分布比例进行加权，数学表达式如下：
$$BalnacedCE\_loss(p,q)=-\sum _i^N{\alpha }_i{p}_i(x)\log q_i(x)$$

Focal Loss

尽管引入Balanced Cross Entropy Loss缓解了类别分布不均衡的问题，但模型对每种类别的分类能力往往是不同的，因此训练模型的时候就需要模型重点关注难分样本的特征，其数学表达式为：
$$Focal\_loss(p,q)=-\sum _i^N{\alpha }_i(1-p_i(x){)}^{\gamma }\log q_i(x)$$
其中 ${\alpha }_i$ 是类别均衡系数，$\gamma >0$，从公式中可以看到，当 $p_i(x)⟶1$ 表明该样本被分类正确的概率越大，易区分样本，则其对应的loss所占的权重更低，反之则是难分类样本其loss权重更大。

应用场景：

1、车道检测与分类

原文地址：https://blog.csdn.net/weixin_50232758/article/details/142992153

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！