计算机科学与技术

基于深度学习加强的混合推荐方法

  • 张敏 ,
  • 丁弼原 ,
  • 马为之 ,
  • 谭云志 ,
  • 刘奕群 ,
  • 马少平
展开
  • 清华大学 计算机系, 清华信息科学与技术国家实验室(筹), 智能技术与系统国家重点实验室, 北京 100084

收稿日期: 2016-12-13

  网络出版日期: 2017-10-15

Hybrid recommendation approach enhanced by deep learning

  • ZHANG Min ,
  • DING Biyuan ,
  • MA Weizhi ,
  • TAN Yunzhi ,
  • LIU Yiqun ,
  • MA Shaoping
Expand
  • State Key Laboratory of Intelligent Technology and Systems, Tsinghua National Laboratory for Information Science and Technology, Department of Computer Science and Technology, Tsinghua University, Beijing 100084, China

Received date: 2016-12-13

  Online published: 2017-10-15

摘要

近年来基于矩阵分解的协同过滤算法在评分预测上取得了显著成果,但仍未能很好地解决冷启动、数据稀疏等问题。因此,如何将评论信息引入推荐系统以缓解上述问题成为研究的热点之一。该文尝试基于深度学习来加强个性化推荐,提出将层叠降噪自动编码器(stacked denoising auto-encoder,SDAE)与隐含因子模型(latent factor model,LFM)相结合的混合推荐方法,综合考虑评论文本与评分,以提升推荐模型对潜在评分预测的准确性。在常用大规模公开Amazon数据集上进行的测试结果表明:与传统推荐模型相比,该文提出的方法可有效提高评分预测的准确性,性能提升最高可达64.43%。

本文引用格式

张敏 , 丁弼原 , 马为之 , 谭云志 , 刘奕群 , 马少平 . 基于深度学习加强的混合推荐方法[J]. 清华大学学报(自然科学版), 2017 , 57(10) : 1014 -1021 . DOI: 10.16511/j.cnki.qhdxxb.2017.25.039

Abstract

Collaborative filtering based on matrix factorization has been very successful, while cold-start and data sparseness problems have not been well resolved. Hence, many studies have attempted to include review information into rating predictions. This paper presents a hybrid model that introduces deep learning into recommendation system with collaborative filtering. The algorithm combines a stacked denoising auto encoder (SDAE) with a latent factor model (LFM) to make use of both review and rating information to improve the rating predictions. Evaluations on a large, commonly used Amazon dataset show that this approach significantly improves the rating prediction accuracy in comparison with traditional models, with up to 64.43% better predictions.

参考文献

[1] Koren Y, Bell R, Volinsky C. Matrix factorization techniques for recommender systems[J]. IEEE Computer, 2009, 42(8):30-37.[2] Koran Y. Factorization meets the neighborhood:A multifaceted collaborative filtering model[C]//Proceedings of the 14th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (KDD'08). Las Vegas, NV, USA:ACM, 2008:426-434.[3] Lee D D. Algorithms for non-negative matrix factorization[J]. Advances in Neural Information Processing Systems, 2015, 13(6):556-562.[4] Mcauley J, Leskovec J. Hidden factors and hidden topics:Understanding rating dimensions with review text[C]//ACM Conference on Recommender Systems. Hong Kong, China:ACM, 2013:165-172.[5] Almahairi A, Kastner K, Cho K, et al. Learning distributed representations from reviews for collaborative filtering[C]//Proceedings of the 9th ACM Conference on Recommender Systems. Vienna, Austria:ACM, 2015:147-154.[6] Sang S L, Chung T, Mcleod D. Dynamic item recommendation by topic modeling for social networks[C]//Eighth International Conference on Information Technology:New Generations. Las Vegas, NV, USA:IEEE, 2011:884-889.[7] Bao Y, Fang H, Zhang J. TopicMF:Simultaneously exploiting ratings and reviews for recommendation[C]//Proceedings of the 28th AAAI Conference on Artificial Intelligence. Quebec, Canada:AAAI, 2014:2-8.[8] Blei D M, Ng A Y. Latent dirchlet allocation[J]. Journal of Machine Learning Research, 2003, 3(1):993-1022.[9] Almahairi A, Kastner K, Cho K, et al. Learning distributed representations from reviews for collaborative filtering[C]//Proceedings of the 9th ACM Conference on Recommender Systems. Vienna, Austria:ACM, 2015:147-154.[10] Wang C, Blei D M. Collaborative topic modeling for recommending scientific articles[C]//Proceedings of the 17th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. New York, NY, USA:ACM, 2011:448-456.[11] Lawrence S, Giles C L, Tsoi A C, et al. Face recognition:A convolutional neural-network approach[J]. IEEE Transactions on Neural Networks, 1997, 8(1):98-113.[12] Simonyan K, Zisserman A. Very deep convolutional networks for large-scale image recognition. (2015-04-10). http://arxiv.org/abs/1409.1556.[13] Szegedy C, Liu W, Jia Y, et al. Going deeper with convolutions[C]//IEEE Conference on Computer Vision and Pattern Recognition. Boston, MA, USA:IEEE, 2015:1-9.[14] Graves A. Long short-term memory[J]. Neural Computation, 1997, 9(8):1735-1780.[15] Graves A. Supervised sequence labelling with recurrent neural networks[J]. Studies in Computational Intelligence, 2012, 385:1-124.[16] Sutskever I, Vinyals O, Le Q V. Sequence to sequence learning with neural networks[J]. Advances in Neural Information Processing Systems, 2014, 4:3104-3112.[17] 陈达. 基于深度学习的推荐系统研究[D]. 北京:北京邮电大学, 2014.CHEN Da. Analysis on Deep Learning Based Recommendation System[D]. Beijing:Beijing University of Posts and Telecommunications, 2014.(in Chinese)[18] 杨宇. 基于深度学习特征的图像推荐系统[D]. 成都:电子科技大学, 2015.YANG Yu. Image Recommendation System Based on Deep Learning[D]. Chengdu:University of Electronic Science and Technology of China, 2015.(in Chinese)[19] Wang H, Wang N, Yeung D Y. Collaborative deep learning for recommender systems[C]//Proceedings of the 21st ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. New York, NY, USA:ACM, 2014:1235-1244.[20] Vincent P, Larochelle H, Lajoie I, et al. Stacked denoising autoencoders:Learning useful representations in a deep network with a local denoising criterion.[J]. Journal of Machine Learning Research, 2010, 11(6):3371-3408.
文章导航

/