Top-1 Accuracy, Top-5 Accuracy

Joonlaxy·2023년 5월 10일
0

Top-1, Top-5 어떤 차이?

개, 고양이, 사슴, 금붕어, 사자, 호랑이, 얼룩말, 캥거루, 상어, 코끼리, 돌고래 총 11가지 동물을 분류하는 AI가 있을때 한 동물 사진을 보여주고 AI가 softmax함수를 통해 각 동물일 확률을 계산하였다.
아래와 같이 말이다.

0.  개 0.3
1.  고양이 0.2
2.  사슴 0.1
3.  금붕어 0.05
4.  사자 0.15
5.  호랑이 0.14
6.  얼룩말 0.1
7.  캥거루 0.11
8.  상어 0.05
9.  코끼리 0.1
10. 돌고래 0.05

설명을 위해 적은 점수이므로 총합이 1이 넘더라도 넘어가주세요.

점수가 높은 순서대로 본다면 개, 고양이, 사자, 호랑이, 캥거루, 이하 생략 ... 순서로 이루어진다.

Top-1 Accuracy

우리가 흔히 생각하는 Accuracy의 형태이다. softmax를 통해 가장 높은 수치인 1등 Class를 예측하고 실제 정답과 동일함을 본다.

Top-5 Accuracy

만약 정답이 "호랑이" 였고 Top-5에서 4등 "호랑이"가 존재함을 통해 계산되는 Accuracy이다.

profile
Come in here

0개의 댓글