An Integrated Recommendation Model Based on Two-stage Deep Learning

被引:0
|
作者
Wang R. [1 ]
Wu Z. [2 ]
Jiang Y. [1 ]
Lou J. [1 ]
机构
[1] School of Information Engineering, Huzhou University, Huzhou, 313000, Zhejiang
[2] School of Oujiang, Wenzhou University, Wenzhou, 325035, Zhejiang
基金
中国国家自然科学基金;
关键词
Deep learning; Deep neural network (DNN); Feature extraction; Marginalized stacked denoising auto-encoder;
D O I
10.7544/issn1000-1239.2019.20190178
中图分类号
学科分类号
摘要
In recent years, deep learning technology has been widely used in the field of recommendation systems and has achieved great success. However, the input quality of the deep learning models has a great influence on the learning results. A sparse input feature vector will not only increase the difficulty of subsequent model training, but also will lead to the learning results falling into local optimum. In this article, an integrated recommendation model based on two-stage deep learning is proposed. Firstly, two individual marginal stacked denoising auto-encoders (mSDA) models with closed-form parameter calculation are used to extract the high-level abstract features of the users and the items. Then the resulted user abstract feature and the item abstract feature are connected as the input vector of the deep neural network (DNN) model, and the parameter learning and model optimization are performed through joint training. In addition, in order to model low-order feature interactions, a logistic regression model based on original feature vector is also integrated into the recommendation model. Extensive experiments with two real-world datasets indicate that the proposed recommendation model shows excellent recommendation performance compared with the state-of-the-art methods, especially in the data sparse and the cold start environments. © 2019, Science Press. All right reserved.
引用
收藏
页码:1661 / 1669
页数:8
相关论文
共 17 条
  • [11] Lian J., Zhou X., Zhang F., Et al., xDeepFM: Combining explicit and implicit feature interactions for recommender systems, Proc of the 24th ACM SIGKDD Int Conf, (2018)
  • [12] He X., Liao L., Zhang H., Et al., Neural collaborative filtering, Proc of the 26th Int Conf on World Wide Web, (2017)
  • [13] Chen M., Xu Z., Weinberger K., Et al., Marginalized denoising auto-encoders for domain adaptation, Proc of the 29th Int Conf on Machine Learning, (2012)
  • [14] Wu Y., Dubois C., Zheng A.X., Et al., Collaborative denoising auto-encoders for top-n recommender systems, Proc of the ACM Int Conf on Web Search and Data Mining, (2016)
  • [15] Li S., Kawale J., Fu Y., Deep collaborative filtering via marginalized denoising auto-encoder, Proc of the ACM Int Conf on Information and Knowledge Management, (2015)
  • [16] Ioffe S., Szegedy C., Batch normalization: Accelerating deep network training by reducing internal covariate shift, Proc of the 32nd Int Conf on Machine Learning, (2015)
  • [17] Abadi M., Barham P., Chen J., Et al., Tensorflow: A system for large-scale machine learning, Proc of the 12th USENIX Conf on Operating Systems Design and Implementation, pp. 165-283, (2016)