arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1004.1003cs.ITcs.LGmath.IT
收藏

协同过滤中基于因子图的消息传递推理

Message-Passing Inference on a Factor Graph for Collaborative Filtering

Texas A&M University(德州农工大学)

Byung-Hak Kim, Arvind Yedla, Henry D. Pfister

摘要:

本文介绍了一种新的消息传递(MP)框架的协同过滤(CF)问题与推荐系统。我们使用概率因子图模型对Netflix奖推广的电影评级预测问题进行建模,并通过导出训练误差方面的泛化误差界来研究该模型。基于该模型,我们开发了一个新的MP算法,称为IMP,学习模型。为了显示IMP算法的优越性,我们将其与密切相关的期望最大化(EM)算法和其他一些矩阵完成算法进行比较。我们对Netflix数据的模拟结果表明,虽然这些方法在大量数据的情况下表现相似,但IMP算法在少量数据的情况下更优越。这在实践中改善了CF系统的冷启动问题。IMP算法的另一个优点是可以使用密度演化(DE)技术来分析它,该技术最初是为纠错码的MP解码而开发的。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑