Modeling Data Correlations in Recommendation

IEEE Access

Cheng Wang¹  Yuan He¹  Changjun Jiang¹ 

[1] Department of Computer Science and Engineering, Tongji University, Shanghai, China;

关键词: Matrix factorization;    probabilistic graphical model;    recommender systems;   

DOI  :  10.1109/ACCESS.2017.2712196

来源: DOAJ

【 摘 要 】

In the field of recommender systems, the Beer & Nappies is a famous story, which reveals the latent relationships between different categories of items. Though matrix factorization (MF) has demonstrated its great effectiveness in most previous work, it neglects the co-occurrences of items selected by individuals. In most MF-based models, the latent preferences of users (or the latent categories of items) are assumed independent, which thereby leads to the weak correlation between the Beer and the Nappies. It also greatly limits the models' ability in recommendation. In this paper, we propose a pure probabilistic generative model, which applies a Gaussian prior to capture the semantic correlations between the latent factors. We also show that our model theoretically achieves better expressive power than traditional MF-based models. We derive efficient inference and learning algorithms based on variational EM methods. The effectiveness of our proposed model is comprehensively verified on three different public data sets. Experimental results show that our approach achieves significant improvements on prediction quality compared with the current state of the art.

【 授权许可】

Unknown