ISSN 1000-1239 CN 11-1777/TP

Journal of Computer Research and Development ›› 2019, Vol. 56 ›› Issue (8): 1661-1669.doi: 10.7544/issn1000-1239.2019.20190178

Special Issue: 2019人工智能前沿进展专题

Previous Articles     Next Articles

An Integrated Recommendation Model Based on Two-stage Deep Learning

Wang Ruiqin1, Wu Zongda2, Jiang Yunliang1, Lou Jungang1   

  1. 1(School of Information Engineering, Huzhou University, Huzhou, Zhejiang 313000);2(School of Oujiang, Wenzhou University, Wenzhou, Zhejiang 325035)
  • Online:2019-08-01

Abstract: 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.

Key words: deep learning, marginalized stacked denoising auto-encoder, deep neural network (DNN), feature extraction

CLC Number: