Collaborative Filtering Recommendation Algorithm Based on Attention GRU and Adversarial Learning

被引:12
|
作者
Xia, Hongbin [1 ,2 ]
Li, Jing Jing [1 ]
Liu, Yuan [1 ,2 ]
机构
[1] Jiangnan Univ, Sch Artificial Intelligence & Comp Sci, Wuxi 214122, Jiangsu, Peoples R China
[2] Jiangnan Univ, Jiangsu Key Lab Media Design & Software Technol, Wuxi 214122, Jiangsu, Peoples R China
来源
IEEE ACCESS | 2020年 / 8卷
基金
中国国家自然科学基金;
关键词
Perturbation methods; Feature extraction; Logic gates; Convolutional neural networks; Robustness; Data mining; Context modeling; Adversarial learning; attention mechanism; gated recurrent unit; convolutional neural network; probabilistic matrix factorization; collaborative filtering;
D O I
10.1109/ACCESS.2020.3038770
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Aiming at the problem that the traditional collaborative filtering algorithm using shallow models cannot learn the deep features of users and items, and the recommendation model is very susceptible to the counter-interference of its parameters; this paper proposes a matrix-factorization recommendation model that combines adversarial learning and attention-gated recurrent units (AGAMF). Firstly, the gated recurrent unit based on the attention mechanism is used to extract the user's latent vector from the user's auxiliary side information. Secondly, the convolutional neural network is used to extract the item's latent vector from the item's auxiliary side information. Finally, adversarial disturbances are introduced on the latent factors of users and items to quantify the loss of the model under parameter disturbances, and the latent vectors of users and items are integrated into the probability matrix factorization to predict the user's rating of the item. Experiments were performed on two real data sets MovieLens-1M and MovieLens-10M, and the RMSE, MAE and Recall indicators were used for evaluation. Experiments prove that the model proposed in this paper is robust and can effectively alleviate the problem of data sparsity. Compared with other related recommendation algorithms, our model has a significant improvement in recommendation performance.
引用
收藏
页码:208149 / 208157
页数:9
相关论文
共 50 条
  • [1] A collaborative filtering recommendation algorithm based on DeepWalk and self-attention
    Guo, Jiaming
    Wen, Hong
    Huang, Weihong
    Yang, Ce
    [J]. INTERNATIONAL JOURNAL OF COMPUTATIONAL SCIENCE AND ENGINEERING, 2023, 26 (03) : 296 - 304
  • [2] A recommendation algorithm based on Collaborative Filtering Technology in distance learning
    Zhao, Ji-chun
    Chen, Lei
    Guo, Jian Xin
    [J]. 2017 IEEE 2ND ADVANCED INFORMATION TECHNOLOGY, ELECTRONIC AND AUTOMATION CONTROL CONFERENCE (IAEAC), 2017, : 2320 - 2323
  • [3] Collaborative Filtering Recommendation Algorithm based on Item Similarity Learning
    Liu, Feng
    Li, Huan
    Ma, Zhu-juan
    Zhu, Er-zhou
    [J]. CURRENT TRENDS IN COMPUTER SCIENCE AND MECHANICAL AUTOMATION, VOL 1, 2017, : 322 - 335
  • [4] Recommendation Model Based on Collaborative Filtering Recommendation Algorithm
    Huang, Jun
    [J]. Proceedings of the 2016 4th International Conference on Mechanical Materials and Manufacturing Engineering (MMME 2016), 2016, 79 : 67 - 70
  • [5] Collaborative Filtering Recommendation Algorithm Based on Graph Convolution Attention Neural Network
    Wang, Wei
    Du, Yuxuan
    Zheng, Xiaoli
    Zhang, Chuang
    [J]. Computer Engineering and Applications, 2023, 59 (13) : 247 - 258
  • [6] Movie recommendation based on ALS collaborative filtering recommendation algorithm with deep learning model
    Li, Ni
    Xia, Yinshui
    [J]. ENTERTAINMENT COMPUTING, 2024, 51
  • [7] Collaborative Filtering Recommendation Algorithm for MOOC Resources Based on Deep Learning
    Wu, Lili
    [J]. COMPLEXITY, 2021, 2021
  • [8] Collaborative filtering with a deep adversarial and attention network for cross-domain recommendation
    Liu, Huiting
    Guo, Lingling
    Li, Peipei
    Zhao, Peng
    Wu, Xindong
    [J]. INFORMATION SCIENCES, 2021, 565 : 370 - 389
  • [9] Collaborative Filtering Recommendation Algorithm Based on Cluster
    Li, Xingyuan
    [J]. 2011 INTERNATIONAL CONFERENCE ON COMPUTER SCIENCE AND NETWORK TECHNOLOGY (ICCSNT), VOLS 1-4, 2012, : 2682 - 2685
  • [10] Logistic recommendation algorithm based on collaborative filtering
    Zhang Xiaoyu
    Dai Chaofan
    Zhao yanpeng
    [J]. PROCEEDINGS OF THE 2015 2ND INTERNATIONAL WORKSHOP ON MATERIALS ENGINEERING AND COMPUTER SCIENCES (IWMECS 2015), 2015, 33 : 865 - 868