一种基于深度学习的新型协同过滤推荐系统模型

A Novel Deep Learning-Based Collaborative Filtering Model for Recommendation System

IEEE Transactions on Cybernetics · 2018
被引 242 · 同刊同年前 7%
ABS 3

中文导读

提出一种深度学习模型,先分别学习用户和物品的低维向量以捕捉各自内部关联,再用前馈神经网络模拟用户与物品的交互,在MovieLens数据集上表现优于传统方法。

Abstract

The collaborative filtering (CF) based models are capable of grasping the interaction or correlation of users and items under consideration. However, existing CF-based methods can only grasp single type of relation, such as restricted Boltzmann machine which distinctly seize the correlation of user-user or item-item relation. On the other hand, matrix factorization explicitly captures the interaction between them. To overcome these setbacks in CF-based methods, we propose a novel deep learning method which imitates an effective intelligent recommendation by understanding the users and items beforehand. In the initial stage, corresponding low-dimensional vectors of users and items are learned separately, which embeds the semantic information reflecting the user-user and item-item correlation. During the prediction stage, a feed-forward neural networks is employed to simulate the interaction between user and item, where the corresponding pretrained representational vectors are taken as inputs of the neural networks. Several experiments based on two benchmark datasets (MovieLens 1M and MovieLens 10M) are carried out to verify the effectiveness of the proposed method, and the result shows that our model outperforms previous methods that used feed-forward neural networks by a significant margin and performs very comparably with state-of-the-art methods on both datasets.

推荐系统协同过滤深度学习机器学习