Title :
Generalized Probabilistic Matrix Factorizations for Collaborative Filtering
Author :
Shan, Hanhuai ; Banerjee, Arindam
Author_Institution :
Dept. of Comput. Sci. & Eng., Univ. of Minnesota, Twin Cities, MN, USA
Abstract :
Probabilistic matrix factorization (PMF) methods have shown great promise in collaborative filtering. In this paper, we consider several variants and generalizations of PMF framework inspired by three broad questions: Are the prior distributions used in existing PMF models suitable, or can one get better predictive performance with different priors? Are there suitable extensions to leverage side information? Are there benefits to taking into account row and column biases? We develop new families of PMF models to address these questions along with efficient approximate inference algorithms for learning and prediction. Through extensive experiments on movie recommendation datasets, we illustrate that simpler models directly capturing correlations among latent factors can outperform existing PMF models, side information can benefit prediction accuracy, and accounting for row/column biases leads to improvements in predictive performance.
Keywords :
groupware; inference mechanisms; information filtering; learning (artificial intelligence); matrix decomposition; pattern clustering; approximate inference algorithm; collaborative filtering; learning; probabilistic matrix factorization; probabilistic matrix factorization; topic models; variational inference;
Conference_Titel :
Data Mining (ICDM), 2010 IEEE 10th International Conference on
Conference_Location :
Sydney, NSW
Print_ISBN :
978-1-4244-9131-5
Electronic_ISBN :
1550-4786
DOI :
10.1109/ICDM.2010.116