Attention Collaborative Autoencoder for Explicit Recommender Systems

被引:8
作者
Chen, Shuo [1 ]
Wu, Min [1 ]
机构
[1] East China Normal Univ, Sch Software Engn, Shanghai 200062, Peoples R China
关键词
recommender systems; neural networks; collaborative filtering; deep learning;
D O I
10.3390/electronics9101716
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Recently, various deep learning-based models have been applied in the study of recommender systems. Some researches have combined the classic collaborative filtering method with deep learning frameworks in order to obtain more accurate recommendations. However, these models either add additional features, but still recommend in the original linear manner, or only extract the global latent factors of the rating matrices in a non-linear way without considering some local special relationships. In this paper, we propose a deep learning framework for explicit recommender systems, named Attention Collaborative Autoencoder (ACAE). Based on the denoising autoencoder, our model can extract the global latent factors in a non-linear fashion from the sparse rating matrices. In ACAE, attention units are introduced during back propagation, enabling discovering potential relationships between users and items in the neighborhood, which makes the model obtain better results in the rating prediction tasks. In addition, we propose how to optimize the training process of the model by proposing a new loss function. Experiments on two public datasets demonstrate the effectiveness of ACAE and its outperformance of competitive baselines.
引用
收藏
页码:1 / 12
页数:12
相关论文
共 24 条
  • [1] [Anonymous], 2016, PREPRINTS
  • [2] [Anonymous], 2017, ARXIV171209043
  • [3] Bahdanau D., 2014, Neural machine translation
  • [4] Bahdanau D., 2014, ABS14090473 CORR
  • [5] Attentive Collaborative Filtering: Multimedia Recommendation with Item- and Component-Level Attention
    Chen, Jingyuan
    Zhang, Hanwang
    He, Xiangnan
    Nie, Liqiang
    Liu, Wei
    Chua, Tat-Seng
    [J]. SIGIR'17: PROCEEDINGS OF THE 40TH INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL, 2017, : 335 - 344
  • [6] Chen TQ, 2012, J MACH LEARN RES, V13, P3619
  • [7] Goodfellow I, 2016, ADAPT COMPUT MACH LE, P1
  • [8] Neural Collaborative Filtering
    He, Xiangnan
    Liao, Lizi
    Zhang, Hanwang
    Nie, Liqiang
    Hu, Xia
    Chua, Tat-Seng
    [J]. PROCEEDINGS OF THE 26TH INTERNATIONAL CONFERENCE ON WORLD WIDE WEB (WWW'17), 2017, : 173 - 182
  • [9] Koren Y., 2008, P 14 ACM SIGKDD INT, P426
  • [10] Lee J., 2013, INT C MACH LEARN, P82