CNN 中 softmax 函数的用途是什么?

人工智能 神经网络 卷积神经网络
2021-11-10 07:54:03

softmax函数有什么用?为什么在卷积神经网络的全连接层末尾使用它?

2个回答

softmax 函数的主要目的是转换(未归一化的)输出K单位(例如表示为K将全连接层的元素)转换为概率分布(归一化输出),通常表示为K元素,每个元素都介于01(概率)并且所有这些元素的总和是1(概率分布)。

在分类任务的情况下,isoftmax 函数产生的向量的第 th 元素对应于网络的输入属于ith 类(例如狗)。

简而言之,softmax 函数有助于进行多分类,即输出两种可能性中的一种以上它适用于分类交叉熵。