2021/12/08 by Damien Sileo, Sileo, Damien, Wout Vossen +3 · 7 citations
Computer Science · #Topic Modeling #Recommender Systems and Techniques #Multimodal Machine Learning Applications
paper · doi:10.48550/arxiv.2112.04184
Recommendation is the task of ranking items (e.g. movies or products) according to individual user needs. Current systems rely on collaborative filtering and content-based techniques, which both require structured training data. We propose a framework for recommendation with off-the-shelf pretrained language models (LM) that only used unstructured text corpora as training data. If a user u liked Matrix and Inception, we construct a textual prompt, e.g. \textit"Movies like Matrix, Inception, " to estimate the affinity between u and m with LM likelihood. We motivate our idea with a corpus analysis, evaluate several prompt structures, and we compare LM-based recommendation with standard matrix factorization trained on different data regimes. The code for our experiments is publicly available (https://colab.research.google.com/drive/1f1mlZ-FGaLGdo5rPzxf3vemKllbh2esT?usp=sharing).