Skip to main content

Message-Passing Inference on a Factor Graph for Collaborative Filtering

Publication ,  Internet Publication
Kim, B-H; Yedla, A; Pfister, HD
April 7, 2010

This paper introduces a novel message-passing (MP) framework for the collaborative filtering (CF) problem associated with recommender systems. We model the movie-rating prediction problem popularized by the Netflix Prize, using a probabilistic factor graph model and study the model by deriving generalization error bounds in terms of the training error. Based on the model, we develop a new MP algorithm, termed IMP, for learning the model. To show superiority of the IMP algorithm, we compare it with the closely related expectation-maximization (EM) based algorithm and a number of other matrix completion algorithms. Our simulation results on Netflix data show that, while the methods perform similarly with large amounts of data, the IMP algorithm is superior for small amounts of data. This improves the cold-start problem of the CF systems in practice. Another advantage of the IMP algorithm is that it can be analyzed using the technique of density evolution (DE) that was originally developed for MP decoding of error-correcting codes.

Duke Scholars

Publication Date

April 7, 2010
 

Citation

APA
Chicago
ICMJE
MLA
NLM
Kim, B.-H., Yedla, A., & Pfister, H. D. (2010). Message-Passing Inference on a Factor Graph for Collaborative Filtering.
Kim, Byung-Hak, Arvind Yedla, and Henry D. Pfister. “Message-Passing Inference on a Factor Graph for Collaborative Filtering,” April 7, 2010.

Publication Date

April 7, 2010