计算机工程与应用 ›› 2020, Vol. 56 ›› Issue (14): 169-175.DOI: 10.3778/j.issn.1002-8331.1903-0362

• 模式识别与人工智能 • 上一篇    下一篇

HRS-DC:基于深度学习的混合推荐模型

刘振鹏,尹文召,王文胜,孙静薇   

  1. 1.河北大学 电子信息工程学院,河北 保定 071002
    2.河北大学 信息技术中心,河北 保定 071002
  • 出版日期:2020-07-15 发布日期:2020-07-14

HRS-DC:Hybrid Recommendation Model Based on Deep Learning

LIU Zhenpeng, YIN Wenzhao, WANG Wensheng, SUN Jingwei   

  1. 1.College of Electronic Information Engineering, Hebei University, Baoding, Hebei 071002, China
    2.Information Technology Center, Hebei University, Baoding, Hebei 071002, China
  • Online:2020-07-15 Published:2020-07-14

摘要:

针对传统的矩阵分解算法,仅利用评分信息作为推荐依据,当评分数据稀疏时,不能准确获取隐式反馈,影响推荐的准确性,充分利用辅助信息进行隐式特征的提取成为研究热点之一,提出一种基于深度学习的推荐模型HRS-DC,利用深度神经网络和卷积神经网络从辅助信息中分别提取出用户和项目的隐性特征向量,再将特征向量经过改进的神经协同过滤得出新的评分矩阵。通过在三个真实的数据集上进行验证,与概率矩阵分解(PMF)、协同过滤主题回归(CTR)、协同过滤深度学习(CDL)、卷积矩阵分解ConvMF算法相比提高了评分预测的准确性,也在一定程度上缓解了冷启动问题。

关键词: 深度学习, 神经网络, 矩阵分解, 辅助信息, 协同过滤

Abstract:

For the traditional matrix factorization algorithm, only the scoring information is used as the recommendation basis, when the scoring data is sparse, the implicit factor vector cannot be accurately extracted, making full use of auxiliary information has become one of the research hotspot, and a recommendation based on deep learning is proposed, the model HRS-DC uses the deep neural network and the convolutional neural network to extract the recessive feature vectors of the user and the project from the auxiliary information, then transforms the feature vector through the improved neural collaborative filtering to obtain a new scoring matrix. Through verification on three real data sets, the accuracy of scoring prediction is improved compared with Probability Matrix Factorization(PMF), Collaborative filtering Topic Regression(CTR), Collaborative filtering Deep Learning(CDL) and Convolution Matrix Factorization(ConvMF), and the cold start problem is alleviated to some extent.

Key words: deep learning, neural network, matrix factorization, side information, collaborative filtering