본문 바로가기

Deep Learning

여러 클래스로 분류할때, 아웃풋노드에서 사용하는 액티베이션함수 소프트맥스

소프트맥스 함수( Softmax Function)

 - 소프트맥스는 세 개 이상으로 분류하는 다중 클래스 분류에서 사용되는 활성화 함수이다.

 - 소프트맥스 함수는 분류될 클래스가 n개라 할 때, n개의 벡터를 입력받아, 각 클래스에 속할 확률을 추정한다.

 

보통 출력층에서만 사용된다. hidden layer에서 사용할 경우, 기울기 소실 문제 등 학습 효율성이 감소한다는 단점이 있다.