Mila > Publication > Graphical Model > Content-based recommendations with Poisson factorization

Content-based recommendations

Graphical Model
Dec 2014

Content-based recommendations with Poisson factorization

Dec 2014

We develop collaborative topic Poisson factorization (CTPF), a generative model of articles and reader preferences. CTPF can be used to build recommender systems by learning from reader histories and content to recommend personalized articles of interest. In detail, CTPF models both reader behavior and article texts with Poisson distributions, connecting the latent topics that represent the texts with the latent preferences that represent the readers. This provides better recommendations than competing methods and gives an interpretable latent space for understanding patterns of readership. Further, we exploit stochastic variational inference to model massive real-world datasets. For example, we can fit CPTF to the full arXiv usage dataset, which contains over 43 million ratings and 42 million word counts, within a day. We demonstrate empirically that our model outperforms several baselines, including the previous state-of-the art approach.

Reference

Prem K. Gopalan, Laurent Charlin, David Blei, Content-based recommendations with Poisson factorization, in: Neural Information Processing Systems (NIPS), 2014

Source code

Linked Profiles