Caffe의 Softmax와 SoftmaxWithLoss 층의 차이점은 무엇입니까?

Question

CAFFE에 prototxt을 정의하는 동안, 나는 우리가 입력 데이터는 각 클래스에 속하는 확률을 반환합니다 때로는 우리가, 내가 Softmax 계층을 SoftmaxWithLoss 알고 사용 마지막 레이어 유형으로 Softmax를 사용 때때로 발견,하지만 SoftmaxwithLoss도 반환 것으로 보인다 계급 확률은 무엇입니까? 또는 두 레이어 유형의 사용법을 오해 했습니까?

Answer 1

SoftmaxWithLoss은 모델 예측을 통해 각 대상 클래스의 확률을 반환하지만 softmax 연산을 예측에 적용 할뿐만 아니라 출력으로 반환 된 다항식 논리 손실을 계산합니다. 이는 교육 단계에서 필수적입니다 (손실없이 네트워크 매개 변수를 업데이트하는 데 사용할 수있는 그라디언트가 없음).

는 SoftmaxWithLossLayer 더 많은 정보를 원하시면 Caffe Loss 참조하십시오.