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 条
  • [1] Ren M., Intelligent Information System: Method and Practice of Optimizing Data Modeling with Relevant Knowledge, (2012)
  • [2] Wang H., Wang N., Yeung D.Y., Collaborative deep learning for recommender systems, Proc of the 15th ACM SIGKDD Conf on Knowledge Discovery and Data Mining, (2015)
  • [3] Zhang S., Yao L., Xu X., AutoSVD++: An efficient hybrid collaborative filtering model via contractive auto-encoders, Roc of the 40th ACM SIGIR Conf on Research and Development in Information Retrieval, (2017)
  • [4] Zhang L., Luo T., Zhang F., Et al., A Recommendation Model Based on Deep Neural network, IEEE Access, 6, pp. 9454-9463, (2018)
  • [5] Wang H., Liang G., Zhang X., Feature Regularization and Deep Learning for Human Resource recommendation, IEEE Access, 6, pp. 39415-39421, (2018)
  • [6] Wu H., Zhang Z., Yue K., Et al., Dual-Regularized Matrix Factorization with Deep Neural Networks for Recommender systems, Knowledge-Based Systems, 145, pp. 46-58, (2018)
  • [7] Covington P., Adams J., Sargin E., Deep neural networks for YouTube recommendations, Proc of the 10th ACM Conf on Recommender Systems, pp. 191-198, (2016)
  • [8] Chen M., Xu Z., Weinberger K.Q., Et al., Marginalized stacked denoising auto-encoders, Proc of the NIPS Special on Learning Workshop, (2012)
  • [9] Cheng H.T., Koc L., Harmsen J., Et al., Wide & deep learning for recommender systems, Proc of the Workshop on Deep Learning for Recommender Systems, (2016)
  • [10] Guo H., Tang R., Ye Y., Et al., DeepFM: A factorization-machine based neural network for CTR prediction, Proc of the 26th Int Joint Conf on Artificial Intelligence, (2017)