54、Softmax 分类器以及它的底层原理

深度学习中的SoftMax分类器详解
SoftMax分类器主要用于将神经网络的原始输出转换为概率分布,解决分类任务中的得分解释性和相近得分判断问题。本文通过实例介绍了SoftMax的工作原理,包括其指数运算如何放大得分差距,确保概率总和为1,以及在实际应用中的使用情况。

下面开始介绍最后一个算法softmax。在前面介绍全连接算法或其他文章中,或多或少也提到了softmax。

在分类网络里,softmax的作用主要是将模型的原始输出映射到 0~1之间的概率分布。很多时候对于我们初学者而言,只知道softmax可以做概率映射,但并不了解它内部的原理是如何完成这个概率映射的,本节来详细的来阐述一下这个过程。

在详细说明这个算法的原理之前,先通过一个例子看一下这个算法的使用场景。

1.一个例子

假设我们要使用神经网络做图片分类,现在有 3 个类别分别是 猫,狗,人。如果我们使用 Resnet50 这一个神经网络进行分类,那么最后一层全连接会输出三个得分,也就是它输出的shape为[1, 3]。

我们继续假设猫、狗、人这三个分类的得分分别为:

也就是说,猫得了 2 分,狗得了 1 分,人得了 0.1 分,单看这个结果,我们大概知道,因为猫的得分最高,那最终神经网络会认为这张图片是一只猫,这么理解是可以的,但是大概还是有两个地方有些问题。

第一,神经网络最终选择某一分类,依据的不是得分,而是概率。也就是说,最终神经网络会选择一个概率最高的分类作为它识别的结果。

为什么要使用概率该不是得分呢?主要出于以下几点考虑:

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

董董灿是个攻城狮

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值