Softmax函数

Softmax函数是一个在机器学习和深度学习中常用的函数,特别是在分类问题中。通常被用于多分类的输出层,将模型输出的原始分数(也称为logits)转换成概率分布。Softmax函数的主要作用是确保输出的概率总和为1,并且每个输出的概率都在0到1之间。
函数表达式:

Zi_{}表示模型对于第i_{}个类别的原始分数(logit),而K是类别的总数。函数的输出Softmax\left ( Zi_{}\right )就是第i_{}个类别的概率。

  1. 指数函数 e:对于每个类别的原始分数Zi_{}​,Softmax函数首先计算e的指数幂。指数函数能够将所有原始分数转换成正数,并且当分数的值相差很大时,指数函数能够放大这些差异。

  2. 归一化分母:为了确保所有类别的概率总和为1,Softmax函数在每个类别的指数值上加上一个归一化因子,即所有类别指数值的总和。这个分母确保了所有输出概率的和为1,从而可以被视为一个有效的概率分布。

举个例子,假设我们有一个三分类问题,模型输出的原始分数(logits)为Z1_{} = -2,Z2_{} = 1,Z3_{} = 3,应用Softmax函数后,可以得到:

这样,就得到了每个类别的概率分别为0.05、0.25和0.70,总和为1。

Softmax函数的一个重要性质是,当输入的原始分数(logits)差异很大时,输出的概率分布会非常尖锐,即某个类别的概率会非常接近1,而其他类别的概率则会非常接近0。这反映了模型对于某个类别的强烈信心。相反,如果原始分数相差不大,概率分布会更加均匀,表明模型对各个类别的预测信心相近。

  • 7
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值