【知识星球】softmax损失相关的小问题

本文深入探讨了图像分类任务中Softmax与Sigmoid损失函数的应用区别,解释了两者在多类别与二分类场景下的输入通道数差异,为读者提供了清晰的理解和学习资料。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

欢迎大家来到《知识星球》专栏,这里是AI1000问小专题。

作者&编辑 | 言有三

1 Softmax相关的问题

Softmax在图像分类,分割任务中是很常见的,关于它有些常见的小问题,供大家思考学习。

有三AI1000问-机器学习篇

为什么分类任务softmax损失输入是2通道,sigmoid损失是1通道

640?wx_fmt=jpeg

想必大家都玩过图像分类了,为什么同样是2分类任务,softmax损失函数的输入是2通道而sigmoid损失函数的输入是1通道?

作者/编辑 言有三

归根结底还是因为这两个损失根本就不是用于同一个任务,尽管在二分类的时候它们恰好是同一个问题。

我们首先看一下softmax损失函数定义,i表示第i个样本,k表示第k类,C表示总类别数,yik是第i个样本第k类的标签,f(xik)即样本i属于第k类的概率。

640?wx_fmt=png

在k=0到C中,只有一个yik非零,这个损失函数的定义是用于多类别分类任务的,即一个样本只属于其中一个类别,比如手写数字的1,2,3,4,5,6,7。在具体实现损失的时候,就需要先取到非零的通道,按照上式计算该通道损失,所以输入通道数目为2。

再看sigmoid损失函数定义:

640?wx_fmt=png

从这个式子可以看出,根本就没有类别C的概念,yi即是否为正样本,f(xi)即样本i是否为正样本的概率,所以单个的sigmoid损失函数只能用于二分类问题。假如有一个分类任务为男,女,老,少,就需要2个sigmoid损失函数。在具体实现损失的时候,每一类问题只需要知道一个正样本概率直接带入上式即可,所以输入通道数目为1。

640?wx_fmt=jpeg

2 更多AI1000问

更多的AI1000,可以移步知识星球阅读!

640?wx_fmt=jpeg

有三AI知识星球的内容非常多,大家可以预览一些内容如下。

640?wx_fmt=png

640?wx_fmt=jpeg

640?wx_fmt=jpeg

640?wx_fmt=jpeg

640?wx_fmt=jpeg

640?wx_fmt=jpeg

640?wx_fmt=jpeg

640?wx_fmt=jpeg

640?wx_fmt=jpeg


以上所有内容

加入有三AI知识星球即可获取

来日方长

点击加入

不见不散

更多精彩

每日更新


640?wx_fmt=jpeg

转载文章请后台联系

侵权必究

640?wx_fmt=png

640?wx_fmt=png

640?wx_fmt=png

往期精选

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

言有三

三人行必有AI

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

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

打赏作者

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

抵扣说明:

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

余额充值