
Output Layer에서는 Sigmoid, Softmax 함수 중 어떤 걸 사용해야 하는가?
·
이론 공부/AI HARDWARE
이미지를 분류하는 문제의 신경망 출력층에서는 주로 Sigmoid와 Softmax가 사용된다. 두 함수의 특징과 적합한 사용 사례에 대해 알아보자.1. Sigmoid입력값을 0과 1 사이의 값으로 변환하며, 각 출력 노드가 독립적으로 확률을 나타낸다. 각 출력이 서로 독립적이어서, 다른 결과에 신경 쓰지 않고 "예/아니오"를 따로따로 판단할 수 있다. 즉, 주로 이진 분류에서 사용되는 활성화 함수이다. 스팸 메일 분류(스팸/비스팸)질병 진단(양성/음성).2. Softmax출력층의 모든 노드 값을 정규화하여 합이 1이 되도록 만든다. 클래스 간의 상대적인 확률을 나타내며, Multi-class 상황에 적합하다.즉, 여러 선택지 중에서 하나만 골라야 할 때, 각 선택지의 확률을 계산해서 전체 합이 1이 되도록..