1. 程式人生 > >詳解softmax

詳解softmax

在這裡插入圖片描述
這張圖就很形象的表示了softmax的作用。若原來的輸出結果為[3,1,-3],經過softmax後結果變為[0.88,0.12,0],這樣結果之和為1,也能很清晰的看出輸出結果為0.88的類別是我們要的。
在這裡插入圖片描述
假設1 2 3 為中間層,4 5 6是輸出的結果,a4 a5 a6是4 5 6的結果經softmax後得到的結果。
在這裡插入圖片描述
由上面的推導可知,要求Loss對w41 w42 w43的 偏導就要求a4對y4的偏導,w51 w52 w53對應a4和y5,w61 w62 w63對應a4和y6。
在這裡插入圖片描述
在這裡插入圖片描述
在這裡插入圖片描述
從以上的推導可知,梯度與softmax之後的值和引數對應的輸出有關。

Reference:

https://www.zhihu.com/question/23765351/answer/240869755