[AWS] 머신러닝에서 Top-K 란? 쉽게 정리

 

 

🔍 Top-K란?

Top-K는 머신러닝 모델이 예측한 단어 중 상위 K개만 선택하는 방식이다.

K개만 선택하도록 제한함으로써, 원하는 정보만 얻을 수 있다.

즉, Top-K는 원하는 정보를 얻기위한 LLM의 출력의 하이퍼파라미터이다.

 

 

✅ 쉬운 예시 :  추천 시스템 (Recommendation)

사용자가 좋아할 가능성이 높은 Top 3개의 영화 추천

 

영화 추천 모델의 예측 확률 (Top-3 적용 전)
🎥 "Inception" → 28%
🎥 "Interstellar" → 22%
🎥 "The Matrix" → 15%
🎥 "The Dark Knight" → 12%
🎥 "Avatar" → 10%
🎥 "Titanic" → 7%

 

영화 추천 모델의 예측 확률 (Top-3 적용 전)

1️⃣ "Inception" 🎥 → 50%
2️⃣ "Interstellar" 🎥 → 30%
3️⃣ "The Matrix" 🎥 → 20%