交叉熵损失cross entropy loss

除了均方差,在神经网络中很常用的一个损失函数是交叉熵损失cross entropy loss。这里记录一下它的概念。
Entropy
熵,衡量不确定性。
信息熵越大,混乱程度越大,信息量越大。
当概率均等的时候信息熵最大,这时系统对取什么态没有偏向性,所以混乱度最大。
在这里插入图片描述
Cross Entropy
交叉熵,是与两个分布有关的概念
在这里插入图片描述
KL Divergence 散度,衡量两个分布的差别大小:
在这里插入图片描述
对于0-1的情况H§=0所以优化H(p,q)就是直接优化散度

举例
五分类问题
真实分布:P1[1 0 0 0 0]
在这里插入图片描述
模型分布:Q1=[0.4 0.3 0.05 0.05 0.2]

计算交叉熵:
在这里插入图片描述
如果模型分布更接近真实分布:Q1=[0.98 0.01 0 0 0.01]

在这里插入图片描述
总结
在分类问题中,相比于均方差,交叉熵损失的梯度会更大。(上例用均方差计算,只会下降0.3-0.4,而用交叉熵可以下降接近0.9)
但是MSE的梯度在计算的时候更加简单。所以有时要根据实际情况进行选择。
在pytorch中,计算softmax和交叉熵损失一般是合在一起的。

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值