在数学,尤其是概率论和相关领域中,Softmax函数,或称归一化指数函数[1]:198,是逻辑斯谛函数的一种推广。它能将一个含任意实数的K维向量
“压缩”到另一个K维实向量
中,使得每一个元素的范围都在
之间,并且所有元素的和为1(也可視為一個 (k-1)維的hyperplane或subspace)。该函数的形式通常按下面的式子给出:
for j = 1, …, K.
Softmax函数实际上是有限项离散概率分布的梯度对数归一化。因此,Softmax函数在包括 多项逻辑回归[1]:206–209 ,多项线性判别分析,朴素贝叶斯分类器和人工神经网络等的多种基于機率的多分类问题方法中都有着广泛应用。[2] 特别地,在多项逻辑回归和线性判别分析中,函数的输入是从K个不同的線性函數得到的结果,而样本向量 x 属于第 j 个分类的機率为:
![{\displaystyle P(y=j|\mathbf {x} )={\frac {e^{\mathbf {x} ^{\mathsf {T}}\mathbf {w} _{j}}}{\sum _{k=1}^{K}e^{\mathbf {x} ^{\mathsf {T}}\mathbf {w} _{k}}}}}](https://wikimedia.org/api/rest_v1/media/math/render/svg/001ce4c2c74e78a66a4d7d04ab92cbd0d0fdec02)
这可以被视作K个线性函数
Softmax函数的复合(![{\displaystyle \mathbf {x} ^{\mathsf {T}}\mathbf {w} }](https://wikimedia.org/api/rest_v1/media/math/render/svg/ed370c04ba00660e67f31aa882d791a742a0da9c)
![{\displaystyle \mathbf {x} }](https://wikimedia.org/api/rest_v1/media/math/render/svg/32adf004df5eb0a8c7fd8c0b6b7405183c5a5ef2)
)。
实际实现的时候,为了防止溢出,会先把每个元素减去原先的最大值
。
![{\displaystyle \sigma (\mathbf {z} )_{i}={\frac {e^{x_{i}}}{\sum _{j}e^{x_{j}}}}={\frac {e^{-m}}{e^{-m}}}{\frac {e^{x_{i}}}{\sum _{j}e^{x_{j}}}}={\frac {e^{x_{i}-m}}{\sum _{j}e^{x_{j}-m}}}}](https://wikimedia.org/api/rest_v1/media/math/render/svg/8df063fc8303b236e9f4b9e8678ad4f629dc9c24)
例子
输入向量
对应的Softmax函数的值为
。输出向量中拥有最大权重的项对应着输入向量中的最大值“4”。这也显示了这个函数通常的意义:对向量进行归一化,凸显其中最大的值并抑制远低于最大值的其他分量。
下面是使用Python进行函数计算的範例程式碼:
import math
z = [1.0, 2.0, 3.0, 4.0, 1.0, 2.0, 3.0]
z_exp = [math.exp(i) for i in z]
print(z_exp) # Result: [2.72, 7.39, 20.09, 54.6, 2.72, 7.39, 20.09]
sum_z_exp = sum(z_exp)
print(sum_z_exp) # Result: 114.98
softmax = [round(i / sum_z_exp, 3) for i in z_exp]
print(softmax) # Result: [0.024, 0.064, 0.175, 0.475, 0.024, 0.064, 0.175]
Python使用numpy计算的示例代码:
import numpy as np
z = np.array([1.0, 2.0, 3.0, 4.0, 1.0, 2.0, 3.0])
print(np.exp(z)/sum(np.exp(z)))
Julia 的範例:
julia> A = [1.0, 2.0, 3.0, 4.0, 1.0, 2.0, 3.0]
7-element Array{Float64,1}:
1.0
2.0
3.0
4.0
1.0
2.0
3.0
julia> exp.(A) ./ sum(exp.(A))
7-element Array{Float64,1}:
0.0236405
0.0642617
0.174681
0.474833
0.0236405
0.0642617
0.174681
參考資料
- ^ 1.0 1.1 Bishop, Christopher M. (2006).
- ^ ai-faq What is a softmax activation function? (页面存档备份,存于互联网档案馆)
可微分计算 |
---|
| 概论 | |
---|
| 概念 | |
---|
| 应用 | |
---|
| 硬件 | - TPU
- VPU
- IPU(英语:Graphcore)
- 憶阻器
- SpiNNaker(英语:SpiNNaker)
|
---|
| 软件库 | |
---|
| 实现 | |
---|
| 人物 | |
---|
| 组织 | |
---|
| 架构 | - 多层感知器(MLP)
- 循环神经网络(RNN)
- 長短期記憶(LSTM)
- 门控循环单元(英语:Gated recurrent unit)(GRU)
- 卷积神经网络(CNN)
- 残差神经网络(ResNet)
- 变换器
- 自编码器
- 变分自编码器(VAE)
- 生成对抗网络(GAN)
- 图神经网络(英语:Graph neural network)(GNN)
- 回响状态网络(英语:Echo state network)(ESN)
- 神经图灵机(NTM)
- 可微分神经计算机(英语:Differentiable neural computer)(DNC)
|
---|
| 主题 分类 |
|