nn.CrossEntropyLoss()和手算交叉熵

文章讲述了计算交叉熵损失的过程,包括softmax转换、标签类别索引化以及考虑nn.CrossEntropyLoss中logsoftmax计算时减去最大值的常数。手动计算与nn模块的结果不同,主要由于这个常数的处理导致。通过调整手动计算的方式,可以消除这种差异,得到0.0226的损失值。

以计算[[17.95,13.84],[14.96,17.59]]的交叉熵数值,label为[0,1] 为例,手动方法为:
在这里插入图片描述

与使用nn.CrossEntropyLoss()函数计算结果不同,可能是由于两种方法的实现细节不同所致。总之计算可以分为3步 :

  1. 首先,将模型输出应用softmax函数,将输出转换为概率分布。
  2. 然后,将标签转换为类别索引。在本例中,标签为[0, 1],将转换为类别索引[0, 1]。
  3. 最后,计算交叉熵损失,公式如下
    . 在这里插入图片描述
    在第二步计算时,nn.CrossEntropyLoss()函数在计算log softmax时会减去一个常数,这个常数的值是输入中的最大值。这个常数的引入会使得最终的交叉熵结果发生偏移。因此和手动算的不一样。
    可以通过
    max_val = torch.max(input, dim=1).values
    log_softmax = F.log_softmax(input, dim=1)
    log_softmax -= max_val.view(-1, 1)
    然后代入公式可以得到结果0.0226
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值