softmax的损失函数不应该是交叉熵吗？

Answer 1

正十七 2020-04-15 21:52:13

损失函数可以是交叉熵，也可以是其他的。softmax的输出你可以认为是概率分布，然后要拟合的也是一个概率分布。比如有三类，softmax输出是[0.1, 0.8, 0.1]， 类别ground truth是[0, 1, 0]。

对于这个case，交叉熵可以拟合，MSE也可以拟合。分类一般用交叉熵。

• 提问者 Shuren_Yu #1

  非常感谢！