왜 중요한가
스트리밍 추천 환경에서는 아이템 풀과 사용자 분포가 계속 바뀌어 한 번 학습한 표현이 곧바로 쓸모없어질 위험이 있습니다. KGD는 프리트레인으로 얻은 행동적 지식과 태스크가 요구하는 판별적 기하를 서로 다른 소유권 레이어로 분리하여 프리트레인을 빈번히 갱신하면서도 태스크 적응을 안정적으로 유지하게 합니다. 이 방식은 리프레시로 인한 지식 소멸이나 태스크 최적화의 제약을 동시에 해소해 운영 중인 추천 서비스에 직접 적용 가능한 장점을 제공합니다.
핵심 기여
프리트레인 목적의 노이즈 제거
다음-토큰 방식이 세션 경계에 걸친 인접관계를 잘못 학습하는 문제를 완화하기 위해 BMTP를 도입합니다. BMTP는 협업 유사도와 의미 유사도 두 축에서 후속 아이템을 필터링해 핵심적이고 일반화 가능한 전이만을 양성 표본으로 사용합니다. 이로 인해 프리트레인 임베딩이 세션 노이즈에 덜 민감한 '기초 기하'를 형성하도록 설계했습니다.
지식·기하의 소유권 분리 메커니즘
프리트레인 인코더와 태스크 학습기를 각각 다른 파라미터 집합으로 소유하게 하고 인터페이스는 읽기 전용 크로스어텐션과 입력 수준의 저순위 잔차(ACR)로 한정합니다. 태스크 그라디언트는 인코더로 역전달되지 않으며, ACR의 잔차는 프리트레인 임베딩과 직교하도록 제약해 기하의 보존과 태스크 전이를 동시에 만족합니다. 이 구조는 엔코더 리프레시와 태스크 적응을 독립적 절차로 만듭니다.
산업 규모 스트리밍 실험과 실제 배포
여덟 개 공개 벤치마크에서 기존 강력한 프리트레인-전이 방법 대비 4~12% 개선을 보고하며, 90일간 일별 리프레시 시나리오에서 베이스라인이 이득을 잃을 때도 KGD는 이득을 유지합니다. Shopee 프로덕션 전체 트래픽에 배포되어 홈페이지 검색 A/B에서 사용자당 GMV를 1.75% 올리고 광고 수익을 1.53% 상승시켰습니다.
핵심 아이디어 이해하기
추천 로그는 서로 다른 관심 세션들이 이어붙여진 형태라 시퀀스 내 인접 관계가 항상 의미적 의존을 뜻하지 않습니다. BMTP는 협업 및 의미 유사도로 값비싼 전이를 필터해서 프리트레인이 일시적 전환 대신 지속적 공통 구조를 학습하도록 유도합니다. 그 위에 태스크 전용 기하를 입력 수준의 직교 잔차와 읽기 전용 크로스어텐션으로 얹으면 인코더를 자주 리프레시해도 태스크 적응을 보존할 수 있습니다.
관련 Figure

좌측 히트맵은 세션 경계로 인한 노이즈가 프리트레인에서 인접 아이템 관계를 잘못 학습할 수 있음을 시사하며, 우측 시간-축 그래프는 프리트레인과 태스크의 그라디언트가 초기 진동 뒤에 유사성이 거의 0에 머무르는 모습을 보입니다. 이 두 시각 자료는 BMTP로 안정적 전이 가능한 기초 기하를 학습하고 읽기 전용 인터페이스로 기하 충돌을 차단해야 한다는 논리적 동기를 제공해 줍니다.
그림 (a)는 프리트레인 임베딩의 코사인 유사도 히트맵으로 세션 간 무관한 아이템 군이 분리되어 있음을 시사하고, 그림 (b)는 프리트레인과 태스크 손실의 그라디언트 코사인 유사도가 거의 0에 가깝게 나타나 두 목적이 서로 상충함을 보여줍니다.
방법론
입력은 아이템 임베딩 시퀀스이고 프로세스는 두 축의 필터링·프리트레인·태스크 학습으로 이어집니다. 먼저 BMTP가 협업(sim_col)과 의미(sim_sem) 유사도 임계값 τ_col, τ_sem로 다음 관련 아이템만을 양성 표본으로 골라 소프트맥스 기반 NTP 손실로 인코더를 학습합니다. 그다음 태스크는 인코더의 최종 히든 상태를 stop-gradient로 읽어가며 ACR로 조정된 임베딩을 입력으로 별도 트랜스포머를 학습해 점수화를 수행합니다. 온라인 운영에서는 하루 단위로 인코더만 리프레시한 뒤 인코더를 고정하고 태스크를 재적응시키는 방식으로 비용과 안정성 균형을 맞춥니다.
주요 결과
논문은 여덟 개 공개 벤치에서 KGD가 기존 강력한 프리트레인·전이 방식들보다 4~12% 상대 개선을 기록했다고 보고합니다. 90일간의 스트리밍 실험에서는 베이스라인들이 리프레시와 목표 충돌로 이득을 잃는 반면 KGD는 이득을 지속적으로 유지했습니다. 실제 배포 결과로 Shopee Homepage Search에서 사용자당 GMV가 1.75% 증가하고 광고 수익이 1.53% 증가한 A/B 성과가 추가로 보고되어 운영적 가치도 입증되었습니다.
기술 상세
모델은 두 개의 트랜스포머로 구성됩니다. 인코더는 BMTP 목적으로 학습된 self-attention Transformer이고 입력은 아이템 임베딩입니다. BMTP 손실은 필터된 양성 집합 S_t^{col}, S_t^{sem}에 대해 위치별 소프트맥스 로그 우도 합으로 표현됩니다. 태스크 측은 ACR로 수정된 임베딩을 받는 별도 Transformer를 갖고, 인코더의 최종 히든 상태 H̃는 stop-gradient로 공급되어 읽기 전용 크로스어텐션의 키·값이 됩니다. ACR는 태스크별 스칼라 s_k와 직교를 보장하는 저순위 행렬 Z^{(k)}B^{(k)T}으로 구성되어 예측 임베딩을 \tilde{e}_i^{(k)} = s_k * sg(e_i^{pre}) + Δe_i^{(k)} 형태로 만듭니다. 직교 규제로 Δe_i^{(k)}가 pre 임베딩과 충돌하지 않게 하며 학습 시 인코더 파라미터로 그라디언트가 흐르지 않도록 설계했습니다.
한계점
범위가 전통적 행동 시퀀스 기반 추천에 한정되어 LLM 기반 추천(문서·프롬프트 기반 적응)과는 적용 방식이 다릅니다. ACR와 태스크 트랜스포머를 추가하면 밀집 파라미터가 대략 두 배로 늘어나고 임베딩 스토리지도 증가해 대규모 카탈로그 운영에서 비용 영향이 존재합니다. 또한 BMTP는 협업·의미 유사도를 사전 계산·캐시해야 하므로 초기 오프라인 전처리와 임계값 τ_col, τ_sem의 튜닝이 필요합니다.
실무 활용
KGD는 프리트레인된 행동 지식을 자주 리프레시해야 하는 대형 추천 시스템에 바로 적용할 수 있는 설계입니다. 인코더와 태스크 학습기를 분리해 인코더 리프레시가 태스크 적응을 훼손하지 않도록 하며, 운영상 일간 리프레시 파이프라인과 호환됩니다.
- 상품 추천과 랭킹 파이프라인에서 프리트레인 임베딩을 일별로 리프레시하는 경우 기존 전이 방법보다 성능을 안정적으로 유지해야 할 때
- 검색·랭킹 시스템에서 노이즈성 세션 경계가 많은 로그로부터 일반화 가능한 행동 구조만을 학습하고자 할 때
- 대규모 온라인 서비스에서 프리트레인 업데이트 비용을 제한하면서도 태스크 별 캘리브레이션을 지속적으로 적용해야 할 때
코드 공개 여부: 공개
코드 저장소 보기키워드
용어 해설
- 다음 아이템 예측(Next-token prediction)
- — 시퀀스의 각 위치에서 다음에 나올 아이템을 카탈로그 전체를 분모로 하는 소프트맥스 확률로 예측하는 목적함수로, 위치별 컨텍스트 상태와 아이템 임베딩의 내적을 점수로 사용해 교차엔트로피 손실을 최소화합니다.
- 행동 분포 변화(Behavioral distribution drift)
- — 상품 풀과 유저 집단이 시간에 따라 바뀌어 행동 로그의 통계가 변하는 현상으로, 같은 사양의 임베딩이라도 시간이 흐르면 원래 학습한 패턴과 실제 관측 간 괴리가 생겨 지속적인 리프레시가 필요합니다.
- 읽기 전용 크로스어텐션(Read-only cross-attention)
- — 태스크 측 트랜스포머가 프리트레인 인코더의 최종 히든 스테이트를 키·값으로 사용하되 stop-gradient로 연결해 프리트레인 파라미터에 그라디언트가 도달하지 않도록 하는 일방향 인터페이스입니다.
AI 요약 · 북마크 · 개인 피드 설정 — 무료
출처 · 인용 안내
인용 시 "요약 출처: AI Trends (aitrends.kr)"를 표기하고, 사실 확인은 원문 보기 기준으로 진행해 주세요. 자세한 기준은 운영 정책을 참고해 주세요.