개인화 추천 시스템 1편 - 유저의 행동은 “언어”일까? : Collaborative Embedding 구축기 (feat. Knowledge Distillation)
명우민 · 컬리 · 컬리 기술 블로그
소개
팀 내에서 범용 개인화 추천 시스템을 구축하며 온라인 A/B 테스트와 CRM 캠페인 테스트에서 비교적 좋은 성과를 확인해, 그 개선 사례와 모델링 과정을 소개하고자 합니다.
AI 핵심 요약
상품 속성만으로는 유저의 구매 맥락을 포착하기 어렵고, 기존 협업 필터링은 콜드 스타트에 취약하며 생성형 순차 추천은 추론 비용이 크다는 문제를 해결하기 위해 컬리는 Collaborative Embedding 기반 추천 시스템을 구축했다. 상품 상세 텍스트를 RQ-VAE로 Semantic ID로 변환하고, Transformer 인코더와 Generative·Retrieval 두 헤드를 함께 학습해 행동 시퀀스의 맥락과 빠른 벡터 검색을 결합했다. Retrieval 성능을 높이기 위해 Generative Head의 확률 분포와 순위 정보를 Knowledge Distillation으로 전달했으며, 주문 단위 위치 인코딩과 인기 보정 샘플링, Hard Negative Mining도 적용했다. 오프라인 평가에서 증류를 적용한 Retrieval 모델은 Teacher 성능에 근접했고, 온라인 A/B 테스트와 CRM 캠페인에서도 기존 방식보다 주요 지표가 개선되어 정확도와 서빙 효율을 함께 고려한 설계의 효과를 확인했다.
- 상품 상세 텍스트를 RQ-VAE로 Semantic ID로 압축하고, 카테고리 예측 보조 손실로 임베딩 공간을 안정화할 수 있다.
- 주문 단위 위치 인코딩을 사용하면 같은 장바구니에 담긴 상품의 동시 구매 맥락을 순차 행동과 구분해 학습할 수 있다.
- 인기 편향은 상품 빈도에 반비례하는 샘플링으로 완화하고, Hard Negative Mining으로 비슷한 상품 간의 선호 차이를 학습시킬 수 있다.
- 대조 학습에서 미반응 상품을 모두 부정 사례로 취급하면 False Negative가 생길 수 있으므로, 교사 모델의 확률 분포와 순위 정보를 증류해 보완할 수 있다.
비슷한 학습 자료
추천 서빙 시스템 아키텍처: 높은 생산성을 위한 아키텍쳐 및 ML Flywheel | 2024 당근 테크 밋업
당근테크 · YouTube
언어 모델 기반의 범용 유저 임베딩과 이를 활용한 추천시스템 및 광고 타겟팅
DEVIEW 2023 · YouTube
[팀네이버 컨퍼런스 DAN25] 실시간 추천-CRM 통합 모델로 완성하는 개인화 UX
DAN 2025 · YouTube
추천 후보는 많을수록 좋을까? TopK를 최적화해 전환율을 높인 방법
구국원 · 토스 기술 블로그
함께 구매하면 좋은 상품이에요! - 장바구니 추천 개발기 2부
최재형, 김수지, 이상협, 구국원 · 컬리 기술 블로그
함께 구매하면 좋은 상품이에요! - 장바구니 추천 개발기 1부
구국원, 박수형 · 컬리 기술 블로그