Confidence Calibration in Convolutional Neural Network

합성곱 신경망에서의 신뢰도 보정

  • 심재훈 (서울대학교 전기정보공학부 뉴미디어통신공동연구소) ;
  • 김세윤 ;
  • 조남익 (서울대학교 전기정보공학부 뉴미디어통신공동연구소)
  • Published : 2020.11.28

Abstract

본 논문에서는 합성곱 신경망을 이용한 이미지 분류에서 신뢰도와 실제 예측 정확도가 다른 문제점을 해결하기 위하여 변형된 두 가지 목적 함수를 제안하였다. 첫 번째는 기존 교차 엔트로피 함수에 새로이 신뢰도와 정확도의 차이를 더해준 것이고, 두번째는 예측값의 최댓값을 0.5로 제한한 것이다. 새로운 목적 함수를 통해 학습해본 결과 정확도의 차이는 거의 나지 않았고, 신뢰도와 실제 정확도는 매우 근접하게 되는 결과를 얻을 수 있었다.

Keywords