Proceedings of the Korean Society of Broadcast Engineers Conference (한국방송∙미디어공학회:학술대회논문집)
- 2020.11a
- /
- Pages.76-78
- /
- 2020
Confidence Calibration in Convolutional Neural Network
합성곱 신경망에서의 신뢰도 보정
- Shim, Jae Hoon (Department of ECE, INMC, Seoul National University) ;
- Kim, Seyun (VUNO Inc.) ;
- Cho, Nam Ik (Department of ECE, INMC, Seoul National University)
- Published : 2020.11.28
Abstract
본 논문에서는 합성곱 신경망을 이용한 이미지 분류에서 신뢰도와 실제 예측 정확도가 다른 문제점을 해결하기 위하여 변형된 두 가지 목적 함수를 제안하였다. 첫 번째는 기존 교차 엔트로피 함수에 새로이 신뢰도와 정확도의 차이를 더해준 것이고, 두번째는 예측값의 최댓값을 0.5로 제한한 것이다. 새로운 목적 함수를 통해 학습해본 결과 정확도의 차이는 거의 나지 않았고, 신뢰도와 실제 정확도는 매우 근접하게 되는 결과를 얻을 수 있었다.
Keywords