Journal of Northeastern University Natural Science ›› 2019, Vol. 40 ›› Issue (11): 1527-1532.DOI: 10.12068/j.issn.1005-3026.2019.11.002

• Information & Control • Previous Articles     Next Articles

Auto-Encoder for Collaborative Filtering Based on Side-Information Enhancement

LIU Hai-bo1,2, FENG Shi1, YU Ge1   

  1. 1. School of Computer Science & Engineering, Northeastern University, Shenyang 110169, China; 2. School of Cyber Security and Computer, Hebei University, Baoding 071002, China.
  • Received:2018-12-20 Revised:2018-12-20 Online:2019-11-15 Published:2019-11-05
  • Contact: LIU Hai-bo
  • About author:-
  • Supported by:
    -

Abstract: In view of inaccurate rating prediction of traditional recommendation model, a model based on side-information enhanced collaborative filtering auto-encoder was proposed, and the design principle, loss function and structure of the model were discussed in detail. The model was implemented by a single hidden layer auto-encoder network with ratings and side-information as input/output data. The model was directly trained by ratings and side-information. The proposed framework not only reduces the scale and complexity of the model, but also amends the user’s interest with side-information. At the same time, the representation ability of the trained model is increased by reasonable division and expansion of training datasets. The comparisons on real datasets show that the proposed model with side-information can effectively ameliorate the user’s interest predicted by the model without side-information, and improve the accuracy of rating prediction, which has a certain practicality.

Key words: auto-encoder, collaborative filtering, side-information, recommender system, rating prediction

CLC Number: