2020 Posts

이 논문을 처음 알게 된 것은 저번달에 Google Brain에서 Tensorflow Recommenders 라는 라이브러리를 공개하면서 입니다. Youtube라는 거대한 추천시스템을 운영하고 있는 구글이 추천 시스템 관련 코드를 공개한다고 해서 집중해서 보게 되었습니다. 전체적인 내용은 Tensorflow Blog에 더 자세히 나와있으니 읽어보시기 바랍니다. TFRS(TensorFlow Recommeners)의 목표는 다음과 같습니다. 추천 후보군을 빠르고 유연하게 빌드 Item, User, Context 정보를 자유롭게 사용하는 구조 다양한 objective를 동시에 학습하는 multi-task 구조 학습된 모델은 TF Serving으로 효율적으로 서빙 사실 코드 자체는 크게 다양한 내용들이 있지는 않았지만, 제일 인상 깊었던 것은 코드에서 기본 모델로 소개한 Two Tower Model이었습니다. 바로 User와 Item을 아예 독립적으로 학습시켜 마지막 단에서 dot product로만 click / unclick을 예측하는 것인데, 생각하면 생각할 수록 좋은 구조더라구요. 비록 학습하는 단에서 user tower와 item tower가 interact 못하기 때문에 엄청난 성능을 낼 지는 미지수였지만, 구조 자체가 input feature의 제약이 없어서 가능한 feature를 자유롭게 넣을 수 있었고, inference할 때는 user별 embedding, item별 embedding으로 가지고 있다가 dot product로만 similarity를 계산해서 serving할 수 있기 때문에 ANN(Approximate Nearest Neighbors) 라이브러리와의 호환성도 좋아 보였습니다. ...