ソフトマックス関数 (Softmax Function)

ソフトマックス関数の役割と特徴

複数の入力値をとり、それぞれの値を0から1の間の数値に変換し、かつそれらの合計が必ず1になるように正規化する役割を持ちます。
多クラス分類問題において、複数のクラスのそれぞれに対して「スコア」を出力し、ソフトマックス関数はそのスコアをそれぞれのクラスに属する確率として解釈できるようにします。
例えば、画像認識で「猫」、「犬」、「鳥」のどれであるかを分類したい場合は、ニューラルネットワークの最後の層はそれぞれのクラスに対する数値を出力します。この数値をソフトマックス関数に通すことで、「猫である確率」「犬である確率」「鳥である確率」という形に変換され、最も確率の高いクラスを予測結果とします。

ソフトマックス関数の利用例：多クラス分類

例えば、画像認識で「猫」、「犬」、「鳥」のどれであるかを分類したい場合は、ニューラルネットワークの最後の層はそれぞれのクラスに対する数値を出力します。この数値をソフトマックス関数に通すことで、「猫である確率」「犬である確率」「鳥である確率」という形に変換され、最も確率の高いクラスを予測結果とします。