위키득계정 ↗
기술

GRU4Rec

GRU4Rec기술
초기 연구 공개
2015년
발표
ICLR 2016
기본 구조
GRU 기반 순환 신경망
예측 대상
세션의 다음 항목
GRU4Rec은 짧은 세션의 행동 순서를 순환 신경망으로 처리해 다음 항목을 추천하는 모델이다. 장기간의 사용자 이력이 없는 세션 기반 추천을 주요 문제로 다뤘다.

세션 전체의 활용

마지막 클릭과 비슷한 항목만 추천하면 그 이전 행동이 주는 문맥을 놓칠 수 있다. 초기 연구는 현재 세션의 행동을 순서대로 처리해 다음 행동의 추천에 활용했다.

세션 병렬 학습

여러 세션을 병렬 미니배치로 학습한다. 한 세션이 끝나면 다음 세션으로 교체하고 해당 은닉 상태를 초기화해 독립적인 세션의 문맥이 섞이지 않도록 구성했다.

추천에 맞춘 손실

추천에서는 관심을 가질 상위 항목의 순위가 중요하므로 순위 손실을 사용했다. 초기 연구는 두 자료에서 기존 방법과 비교하며 일반적인 순환망을 세션 추천에 맞게 조정하는 접근을 평가했다.

관련 문서

GRU / 순환 신경망 / SASRec / BERT4Rec / DSIN / 추천 시스템