🌈交叉熵损失函数原理详解

  • ❤️ 信息论
  • 1️⃣ 信息量
  • 2️⃣ 熵
  • 3️⃣ 相对熵(KL散度)
  • 4️⃣ 交叉熵
  • Ⓜ️机器学习中交叉熵的应用
  • 1️⃣ 为什么要用交叉熵做loss函数?
  • 2️⃣ 交叉熵在单分类问题中的使用
  • 3️⃣ 交叉熵在多分类问题中的使用
  • ㊗️总结


⚡最近面临找工作,被面试官问得体无完肤。踏入机器学习领域时间较短,此类基本的问题虽然在实际工作和代码开发中不曾深入,但是对于应对面试和后续的一些复杂模型的学习是必不可少的。

⚡尤其是在代码中经常看见交叉熵损失函数(CrossEntropy Loss),只知道它是分类问题中经常使用的一种损失函数,对于其内部的原理总是很呆萌,而且一般使用交叉熵作为损失函数时,输出层会接一个softmax函数或者sigmoid,其中的原因实际并不复杂,但便于日后回忆,以此文进行总结,以便自己和小伙伴们以后翻阅。

❤️ 信息论

先不急,因为交叉熵是信息论中的一个概念,要想了解交叉熵的本质,需要先从最基本的概念讲起(很容易理解)。

1️⃣ 信息量

首先是信息量。假设我们听到了两件事,分别如下:
事件A:巴西队进入了2018世界杯决赛圈。
事件B:中国队进入了2018世界杯决赛圈。
仅凭直觉来说,显而易见事件B的信息量比事件A的信息量要大。究其原因,是因为事件A发生的概率很大,事件B发生的概率很小。所以当越不可能的事件发生了,我们获取到的信息量就越大。越可能发生的事件发生了,我们获取到的信息量就越小。那么信息量应该和事件发生的概率有关。

假设X是一个离散型随机变量,其取值集合为交叉熵损失函数python 交叉熵损失函数是什么_机器学习,概率分布函数 交叉熵损失函数python 交叉熵损失函数是什么_面试_02,则定义事件交叉熵损失函数python 交叉熵损失函数是什么_计算机视觉_03的信息量为:

交叉熵损失函数python 交叉熵损失函数是什么_计算机视觉_04

由于是概率所以交叉熵损失函数python 交叉熵损失函数是什么_机器学习_05的取值范围是 交叉熵损失函数python 交叉熵损失函数是什么_面试_06,绘制为图形如下:

交叉熵损失函数python 交叉熵损失函数是什么_计算机视觉_07


可见该函数符合我们对信息量的直觉,概率越小,信息量越大。

2️⃣ 熵

考虑另一个问题,对于某个事件,有交叉熵损失函数python 交叉熵损失函数是什么_计算机视觉_08种可能性,每一种可能性都有一个概率交叉熵损失函数python 交叉熵损失函数是什么_面试_09。这样就可以计算出某一种可能性的信息量。
Eg:假设你拿出了你的电脑,按下开关,会有三种可能性,下表列出了每一种可能的概率及其对应的信息量:

序号

事件

概率p

信息量

A

电脑正常开机

0.7

交叉熵损失函数python 交叉熵损失函数是什么_计算机视觉_10

B

电脑无法开机

0.2

交叉熵损失函数python 交叉熵损失函数是什么_计算机视觉_11

C

电脑爆炸了

0.1

交叉熵损失函数python 交叉熵损失函数是什么_面试_12

注:文中的对数均为自然对数交叉熵损失函数python 交叉熵损失函数是什么_面试_13

我们现在有了信息量的定义,而熵用来表示所有信息量的期望,即:
交叉熵损失函数python 交叉熵损失函数是什么_交叉熵损失函数python_14
其中n代表所有的n种可能性,所以上面的问题结果就是
交叉熵损失函数python 交叉熵损失函数是什么_损失函数_15
然而有一类比较特殊的问题,比如投掷硬币只有两种可能,字朝上或花朝上。买彩票只有两种可能,中奖或不中奖。我们称之为0-1分布问题(二项分布的特例),对于这类问题,熵的计算方法可以简化为如下算式:
交叉熵损失函数python 交叉熵损失函数是什么_损失函数_16

3️⃣ 相对熵(KL散度)

相对熵又称KL散度,如果我们对于同一个随机变量x有两个单独的概率分布 P(x)Q(x),我们可以使用 KL 散度(Kullback-Leibler (KL) divergence)来衡量这两个分布的差异
即如果用P来描述目标问题,而不是用Q来描述目标问题,得到的信息增量。

在机器学习中,P往往用来表示样本的真实分布,比如[1,0,0]表示当前样本属于第一类。Q用来表示模型所预测的分布,比如[0.7,0.2,0.1] 直观的理解就是如果用P来描述样本,那么就非常完美。而用Q来描述样本,虽然可以大致描述,但是不是那么的完美,信息量不足,需要额外的一些“信息增量”才能达到和P一样完美的描述。如果我们的Q通过反复训练,也能完美的描述样本,那么就不再需要额外的“信息增量”,Q等价于P

KL散度的计算公式:
交叉熵损失函数python 交叉熵损失函数是什么_交叉熵损失函数python_17
交叉熵损失函数python 交叉熵损失函数是什么_计算机视觉_08为事件的所有可能性。
交叉熵损失函数python 交叉熵损失函数是什么_面试_19的值越小,表示交叉熵损失函数python 交叉熵损失函数是什么_面试_20分布和交叉熵损失函数python 交叉熵损失函数是什么_机器学习_21分布越接近

4️⃣ 交叉熵

对式3.1变形可以得到:
交叉熵损失函数python 交叉熵损失函数是什么_损失函数_22
等式的前一部分恰巧就是交叉熵损失函数python 交叉熵损失函数是什么_机器学习_21的熵,等式的后一部分,就是交叉熵:
交叉熵损失函数python 交叉熵损失函数是什么_计算机视觉_24
在机器学习中,我们需要评估label和predicts之间的差距,使用KL散度刚刚好,即交叉熵损失函数python 交叉熵损失函数是什么_交叉熵损失函数python_25,由于KL散度中的前一部分交叉熵损失函数python 交叉熵损失函数是什么_交叉熵损失函数python_26不变,故在优化过程中,只需要关注交叉熵就可以了。所以一般在机器学习中直接用用交叉熵做loss,评估模型。

Ⓜ️机器学习中交叉熵的应用

1️⃣ 为什么要用交叉熵做loss函数?

在线性回归问题中,常常使用MSE (Mean Squared Error)作为loss函数,比如:
交叉熵损失函数python 交叉熵损失函数是什么_计算机视觉_27
这里的m表示m个样本的,loss为m个样本的loss均值。
MSE在线性回归问题中比较好用,那么在逻辑分类问题中还是如此么?

2️⃣ 交叉熵在单分类问题中的使用

这里的单类别是指,每一张图像样本只能有一个类别,比如只能是狗或只能是猫。

交叉熵在单分类问题上基本是标配的方法

交叉熵损失函数python 交叉熵损失函数是什么_损失函数_28

上式为一张样本的loss计算方法。式2.1中n代表着n种类别。

举例说明,比如有如下样本

交叉熵损失函数python 交叉熵损失函数是什么_面试_29


对应的标签和预测值

*


青蛙


Label

0

1

0

Prediction

0.3

0.6

0.1

那么
交叉熵损失函数python 交叉熵损失函数是什么_交叉熵损失函数python_30
对应一个batch的loss就是:
交叉熵损失函数python 交叉熵损失函数是什么_面试_31
m为当前batch的样本数

3️⃣ 交叉熵在多分类问题中的使用

这里的多类别是指,每一张图像样本可以有多个类别,比如同时包含一只猫和一只狗

和单分类问题的标签不同,多分类的标签是n-hot

比如下面这张样本图,即有猫,又有狗,所以是一个多分类问题

交叉熵损失函数python 交叉熵损失函数是什么_交叉熵损失函数python_32


对应的标签和预测值

*

青蛙



Label

0

1

1

Prediction

0.1

0.7

0.8

值得注意的是,这里的Prediction不再是通过softmax计算的了,这里采用的是sigmoid。将每一个节点的输出归一化到[0,1]之间。所有Prediction值的和也不再为1。换句话说,就是每一个Label都是独立分布的,相互之间没有影响。所以交叉熵在这里是单独对每一个节点进行计算,每一个节点只有两种可能值,所以是一个二项分布。前面说过对于二项分布这种特殊的分布,熵的计算可以进行简化。

同样的,交叉熵的计算也可以简化,即
交叉熵损失函数python 交叉熵损失函数是什么_损失函数_33
注意,上式只是针对一个节点的计算公式。这一点一定要和单分类loss区分开来。
例子中可以计算为:
交叉熵损失函数python 交叉熵损失函数是什么_损失函数_34
单张样本的loss即为交叉熵损失函数python 交叉熵损失函数是什么_损失函数_35
每一个batch的loss就是:
交叉熵损失函数python 交叉熵损失函数是什么_面试_36
式中m为当前batch中的样本量,n为类别数。

㊗️总结

我只是知识的搬运工,希望对你们多少有点用❤️