交叉熵损失函数的求导(Logistic回归)

前言

最近有遇到些同学找我讨论sigmoid训练多标签或者用在目标检测中的问题，我想写一些他们的东西，想到以前的博客里躺着这篇文章（2015年读研时机器学课的作业）感觉虽然不够严谨，但是很多地方还算直观，就先把它放过来吧。

说明: 本文只讨论Logistic回归的交叉熵，对Softmax回归的交叉熵类似（Logistic回归和Softmax回归两者本质是一样的，后面我会专门有一篇文章说明两者关系，先在这里挖个坑）。首先，我们二话不说，先放出逻辑回归交叉熵的公式：

$J( heta)=-frac{1}{m}sum_{i=1}^{m}y^{(i)}log(h_ heta(x^{(i)}))+(1-y^{(i)})log(1-h_ heta(x^{(i)})),$

以及 $J( heta)$ 对参数 $heta$ 的偏导数（用于诸如梯度下降法等优化算法的参数更新），如下：

$frac{partial}{partial heta_{j}}J( heta) =frac{1}{m}sum_{i=1}^{m}(h_ heta(x^{(i)})-y^{(i)})x_j^{(i)}$

但是在大多论文或数教程中，也就是直接给出了上面两个公式，而未给出推导过程，这就给初学者造成了一定的困惑。交叉熵的公式可以用多种解释得到，甚至不同领域也会有不同，比如数学系的用极大似然估计，信息工程系的的从信息编码角度，当然更多是联合KL散度来解释。但是我这里假设那些你都不了解的情况下如何用一个更加直白和直观的解释来得到Logistic Regression的交叉熵损失函数，说清楚它存在的合理性就可以解惑（关于交叉熵的所谓"正统"解释后续我会专门写一篇文章来总结，先挖个坑）。因水平有限，如有错误，欢迎指正。

废话不说，下文将介绍一步步得到Logistic Regression的交叉熵损失函数，并推导出其导数，同时给出简洁的向量形式及其导数推导过程。

交叉熵损失函数(Logistic Regression代价函数)

我们一共有 $m$ 组已知样本（ $Batch size = m$ ）， $(x^{(i)},y^{(i)})$ 表示第 $i$ 组数据及其对应的类别标记。其中 $x^{(i)}=(1,x^{(i)}_1,x^{(i)}_2,...,x^{(i)}_p)^T$ 为 $p+1$ 维向量（考虑偏置项）， $y^{(i)}$ 则为表示类别的一个数：

logistic回归（是非问题）中， $y^{(i)}$ 取0或者1；
softmax回归 （多分类问题）中， $y^{(i)}$ 取1,2...k中的一个表示类别标号的一个数（假设共有k类）。

这里，只讨论logistic回归，输入样本数据 $x^{(i)}=(1,x^{(i)}_1,x^{(i)}_2,...,x^{(i)}_p)^T$ ，模型的参数为 $heta=( heta_0, heta_1, heta_2,..., heta_p)^T$ ,因此有

$heta^T x^{(i)}:= heta_0+ heta_1 x^{(i)}_1+dots+ heta_p x^{(i)}_p.$

二元问题中常用sigmoid作为假设函数（hypothesis function），定义为：

$h_ heta(x^{(i)})=frac{1}{1+e^{- heta^T x^{(i)}} }.$

因为Logistic回归问题就是0/1的二分类问题，可以有

$P({hat{y}}^{(i)}=1|x^{(i)}; heta)=h_ heta(x^{(i)}) \ P({hat{y}}^{(i)}=0|x^{(i)}; heta)=1-h_ heta(x^{(i)})$

现在，我们不考虑“熵”的概念，根据下面的说明，从简单直观角度理解，就可以得到我们想要的损失函数：我们将概率取对数，其单调性不变，有

$log P({hat{y}}^{(i)}=1|x^{(i)}; heta)=log h_ heta(x^{(i)})=logfrac{1}{1+e^{- heta^T x^{(i)}} } \ log P({hat{y}}^{(i)}=0|x^{(i)}; heta)=log (1-h_ heta(x^{(i)}))=logfrac{e^{- heta^T x^{(i)}}}{1+e^{- heta^T x^{(i)}} }$

那么对于第 $i$ 组样本，假设函数表征正确的组合对数概率为：

$I{y^{(i)}=1}log P({hat{y}}^{(i)}=1|x^{(i)}; heta)+I{y^{(i)}=0}log P({hat{y}}^{(i)}=0|x^{(i)}; heta)\ =y^{(i)}log P({hat{y}}^{(i)}=1|x^{(i)}; heta)+(1-y^{(i)})log P({hat{y}}^{(i)}=0|x^{(i)}; heta)\ =y^{(i)}log(h_ heta(x^{(i)}))+(1-y^{(i)})log(1-h_ heta(x^{(i)}))$

其中， $I{y^{(i)}=1}$ 和 $I{y^{(i)}=0}$ 为示性函数（indicative function），简单理解为{ }内条件成立时，取1，否则取0，这里不赘言。那么对于一共 $m$ 组样本，我们就可以得到模型对于整体训练样本的表现能力：

$sum_{i=1}^{m}y^{(i)}log(h_ heta(x^{(i)}))+(1-y^{(i)})log(1-h_ heta(x^{(i)}))$

由以上表征正确的概率含义可知，我们希望其值越大，模型对数据的表达能力越好。而我们在参数更新或衡量模型优劣时是需要一个能充分反映模型表现误差的损失函数（Loss function）或者代价函数（Cost function）的，而且我们希望损失函数越小越好。由这两个矛盾，那么我们不妨领代价函数为上述组合对数概率的相反数：

$J( heta)=-frac{1}{m}sum_{i=1}^{m}y^{(i)}log(h_ heta(x^{(i)}))+(1-y^{(i)})log(1-h_ heta(x^{(i)}))$

上式即为大名鼎鼎的交叉熵损失函数。(说明：如果熟悉“信息熵"的概念 $E[-log p_i]=-sum_{i=1}^mp_ilog p_i$ ，那么可以有助理解叉熵损失函数）

交叉熵损失函数的求导

这步需要用到一些简单的对数运算公式，这里先以编号形式给出，下面推导过程中使用特意说明时都会在该步骤下脚标标出相应的公式编号，以保证推导的连贯性。

① $log frac{a}{b}=log a-log b$

② $log a+log b=log (ab)$

③ $a=log e^a$ (为了方便这里 $log$ 指 $log_e$ ，即 $ln$ ，其他底数如2,10等，由换底公式可知，只是前置常数系数不同，对结论毫无影响)

另外，值得一提的是在这里涉及的求导均为矩阵、向量的导数（矩阵微商），这里有一篇教程总结得精简又全面，非常棒，推荐给需要的同学。

下面开始推导：

交叉熵损失函数为：

$J( heta)=-frac{1}{m}sum_{i=1}^{m}y^{(i)}log(h_ heta(x^{(i)}))+(1-y^{(i)})log(1-h_ heta(x^{(i)})) ag{1}$

其中，

$log h_ heta(x^{(i)})=logfrac{1}{1+e^{- heta^T x^{(i)}} }=-log ( 1+e^{- heta^T x^{(i)}} ) ,\ egin{align}log(1- h_ heta(x^{(i)}))&=log(1-frac{1}{1+e^{- heta^T x^{(i)}} })\ &=log(frac{e^{- heta^T x^{(i)}}}{1+e^{- heta^T x^{(i)}} })\&=log (e^{- heta^T x^{(i)}} )-log ( 1+e^{- heta^T x^{(i)}} ) \ &=- heta^T x^{(i)}-log ( 1+e^{- heta^T x^{(i)}} ) _{①③} . end{align}$

由此，得到

$egin{align} J( heta) &=-frac{1}{m}sum_{i=1}^m left[-y^{(i)}(log ( 1+e^{- heta^T x^{(i)}})) + (1-y^{(i)})(- heta^T x^{(i)}-log ( 1+e^{- heta^T x^{(i)}} )) ight]\ &=-frac{1}{m}sum_{i=1}^m left[y^{(i)} heta^T x^{(i)}- heta^T x^{(i)}-log(1+e^{- heta^T x^{(i)}}) ight]\ &=-frac{1}{m}sum_{i=1}^m left[y^{(i)} heta^T x^{(i)}-log e^{ heta^T x^{(i)}}-log(1+e^{- heta^T x^{(i)}}) ight]_{③}\ &=-frac{1}{m}sum_{i=1}^m left[y^{(i)} heta^T x^{(i)}-left(log e^{ heta^T x^{(i)}}+log(1+e^{- heta^T x^{(i)}}) ight) ight] _②\ &=-frac{1}{m}sum_{i=1}^m left[y^{(i)} heta^T x^{(i)}-log(1+e^{ heta^T x^{(i)}}) ight] end{align}$

这次再计算 $J( heta)$ 对第 $j$ 个参数分量 $heta_j$ 求偏导:

$egin{align} frac{partial}{partial heta_{j}}J( heta) &=frac{partial}{partial heta_{j}}left(frac{1}{m}sum_{i=1}^m left[log(1+e^{ heta^T x^{(i)}})-y^{(i)} heta^T x^{(i)} ight] ight)\ &=frac{1}{m}sum_{i=1}^m left[frac{partial}{partial heta_{j}}log(1+e^{ heta^T x^{(i)}})-frac{partial}{partial heta_{j}}left(y^{(i)} heta^T x^{(i)} ight) ight]\ &=frac{1}{m}sum_{i=1}^m left(frac{x^{(i)}_je^{ heta^T x^{(i)}}}{1+e^{ heta^T x^{(i)}}}-y^{(i)}x^{(i)}_j ight)\ &=frac{1}{m}sum_{i=1}^{m}(h_ heta(x^{(i)})-y^{(i)})x_j^{(i)} end{align}$

这就是交叉熵对参数的导数：

$frac{partial}{partial heta_{j}}J( heta) =frac{1}{m}sum_{i=1}^{m}(h_ heta(x^{(i)})-y^{(i)})x_j^{(i)}$

向量形式

前面都是元素表示的形式，只是写法不同，过程基本都是一样的，不过写成向量形式会更清晰，这样就会把 $i$ 和求和符号 $sum$ 省略掉了。我们不妨忽略前面的固定系数项 $1/m$ ，交叉墒的损失函数(1)则可以写成下式：

$J( heta) = -left[ y^T log h_ heta(x)+(1-y^T)log(1-h_ heta(x)) ight] ag{2}$

将 $h_ heta(x)=frac{1}{1+e^{- heta^T x} }$ 带入，得到：

$egin{align} J( heta) &= -left[ y^T log frac{1}{1+e^{- heta^T x} }+(1-y^T)logfrac{e^{- heta^T x}}{1+e^{- heta^T x} } ight] \ &= -left[ -y^T log (1+e^{- heta^T x}) + (1-y^T) log e^{- heta^T x} - (1-y^T)log (1+e^{- heta^T x}) ight] \ &= -left[(1-y^T) log e^{- heta^T x} - log (1+e^{- heta^T x}) ight]\ &= -left[(1-y^T ) (- heta^Tx) - log (1+e^{- heta^T x}) ight] end{align}$

再对 $heta$ 求导，前面的负号直接削掉了，

$egin{align} frac{partial}{partial heta_{j}}J( heta) &= -frac{partial}{partial heta_{j}}left[(1-y^T ) (- heta^Tx) - log (1+e^{- heta^T x}) ight] \ &= (1-y^T)x- frac{e^{- heta^Tx }}{1+e^{- heta^T x} }x \ &= (frac{1}{1+e^{- heta^T x} } - y^T)x \ &= left(h_ heta(x)-y^T ight)x end{align}$

3 梯度下降参数更新

$m i n i m i z e J (w)$

转载请注明出处Jason Zhao的知乎专栏“人工+智能“，文章链接：

Jason Zhao：交叉熵损失函数的求导(Logistic回归)