计算机工程与应用 ›› 2021, Vol. 57 ›› Issue (22): 153-159.DOI: 10.3778/j.issn.1002-8331.2006-0440

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

基于变分自编码器的评分预测模型

陈海,钱付兰,陈洁,赵姝,张燕平   

  1. 1.安徽大学 计算机科学与技术学院,合肥 230601
    2.安徽大学 计算智能与信号处理教育部重点实验室,合肥 230601
  • 出版日期:2021-11-15 发布日期:2021-11-16

Rating Prediction Model Based on Variational Auto-Encoder

CHEN Hai, QIAN Fulan, CHEN Jie, ZHAO Shu, ZHANG Yanping   

  1. 1.School of Computer Science and Technology, Anhui University, Hefei 230601, China
    2.Key Laboratory of Intelligent Computing & Signal Processing(Anhui University), Ministry of Education, Hefei 230601, China
  • Online:2021-11-15 Published:2021-11-16

摘要:

深度学习模型具有鲁棒性差的局限性,常见的如在图片中增加特定的噪声会影响到图片的分类和预测结果。近期有学者将深度学习引入到推荐系统中,因此在推荐系统中也存在噪声对推荐精度影响的问题。针对深度推荐模型的鲁棒性问题,基于变分自编码器(Variational Auto-Encoder,VAE)提出了新的评分预测模型REVAE (REcommender Variational Auto-Encoder)。该模型为了训练模型对噪声干扰的鲁棒性,在传统的VAE上增加了一层隐层表示,利用后验分布对隐层表示进行约束,并在该隐层上增加了噪声,通过重构输入数据,训练得到具有抗噪能力的推荐算法模型。在公开的Movielens数据集上进行的实验结果表明,REVAE可以有效降低噪声对模型的干扰,使得整个模型更具有健壮性,相比其他评分预测算法具有更好的推荐效果。

关键词: 深度学习, 推荐系统, 变分自编码器(VAE), 评分预测

Abstract:

The deep learning model has the limitation of poor robustness. For example, the addition of specific noise in the picture will affect the classification and prediction results of the picture. Recently, some scholars have introduced deep learning into the recommendation system, so the influence of noise on the recommendation accuracy is also a problem in the recommendation system. Regarding the robustness of the deep recommendation model, this paper proposes a new rating prediction model REcommender Variational Auto-Encoder(REVAE) based on Variational Auto-Encoder(VAE). In order to train the robustness of the model to noise interference, a layer of hidden layer representation is added on the traditional VAE, the posterior distribution is used to constrain the hidden layer representation, and the noise is added on the hidden layer. By reconstructing the input data, the recommended algorithm model with anti-noise ability is trained. The experimental results on the public Movielens dataset show that REVAE can effectively reduce the noise interference to the model, making the whole model more robust and having better recommendation effect than other rating prediction algorithms.

Key words: deep learning, recommendation system, Variational Auto-Encoder(VAE), rating prediction