Bayesian nonparametric poisson factorization for recommendation systems

Prem Gopalan, Francisco J.R. Ruiz, Rajesh Ranganath, David M. Blei

Research output: Contribution to journalConference article

Abstract

We develop a Bayesian nonparametric Poisson factorization model for recommendation systems. Poisson factorization implicitly models each user's limited budget of attention (or money) that allows consumption of only a small subset of the available items. In our Bayesian nonparametric variant, the number of latent components is theoretically unbounded and effectively estimated when computing a posterior with observed user behavior data. To approximate the posterior, we develop an efficient variational inference algorithm. It adapts the dimensionality of the latent components to the data, only requires iteration over the user/item pairs that have been rated, and has computational complexity on the same order as for a parametric model with fixed dimensionality. We studied our model and algorithm with large real-world data sets of user-movie preferences. Our model eases the computational burden of searching for the number of latent components and gives better predictive performance than its parametric counterpart.

Original languageEnglish (US)
Pages (from-to)275-283
Number of pages9
JournalJournal of Machine Learning Research
Volume33
StatePublished - 2014
Event17th International Conference on Artificial Intelligence and Statistics, AISTATS 2014 - Reykjavik, Iceland
Duration: Apr 22 2014Apr 25 2014

ASJC Scopus subject areas

  • Software
  • Control and Systems Engineering
  • Statistics and Probability
  • Artificial Intelligence

Fingerprint Dive into the research topics of 'Bayesian nonparametric poisson factorization for recommendation systems'. Together they form a unique fingerprint.

  • Cite this

    Gopalan, P., Ruiz, F. J. R., Ranganath, R., & Blei, D. M. (2014). Bayesian nonparametric poisson factorization for recommendation systems. Journal of Machine Learning Research, 33, 275-283.