基于生成对抗模型的序列推荐算法

陈继伟,汪海涛,姜瑛,陈星

PDF(3075 KB)
PDF(3075 KB)
中文信息学报 ›› 2022, Vol. 36 ›› Issue (7) : 143-153.
信息检索

基于生成对抗模型的序列推荐算法

  • 陈继伟,汪海涛,姜瑛,陈星
作者信息 +

Sequential Recommendation with Generative Adversarial Networks

  • CHEN Jiwei,WANG Haitao,JIANG Ying, CHEN Xing
Author information +
History +

摘要

针对传统序列推荐算法时间信息和项目内容信息运用不充分的问题,该文提出基于生成对抗模型的序列推荐算法。通过生成对抗模型将序列建模与时间、内容信息建模分离,充分挖掘用户项目交互的序列信息和项目内容信息。运用卷积神经网络作为生成对抗模型的生成器,捕获用户项目交互的序列模式。运用注意力机制作为生成对抗模型的判别器,捕获交互序列的时间信息和项目内容信息。针对传统序列推荐算法时间信息建模不充分的问题,提出一种改进的时间嵌入方式,充分建模用户项目交互关于时间的周期性模式。利用生成对抗模型同时建模用户的稳定偏好和动态偏好,提升推荐系统的用户体验,并在公开数据集MovieLens-1M和Amazon-Beauty上与现有的优秀算法做比较。实验证明,该文所提出的算法在评价指标HR@N和NDCG@N上较基线方法均有一定提升。

Abstract

Existing recommendation algorithms use item similarity or user similarity to recommend items without capturing the sequence pattern of user interaction with items. In fact, the interaction sequence between users and items contains important context information, which is of great significance for generating user interaction predictions. In this paper, a sequence recommendation algorithm based on a generative adversarial model is proposed. First, a convolutional neural network is used as a generator to capture the sequence pattern of the user interaction sequence, and then the attention mechanism is used as the discriminator of the generative adversarial networks to capture the time information of the sequence and the item content attribute information. We also use an improved time embedding method to model the time periodic change of the interaction. The generative adversarial networks simultaneously model the user's long-term preferences and short-term preferences. Experiments on the public data sets MovieLens-1M and Amazon-Beauty prove that the proposed algorithm has a significant improvement over all baseline methods according to HR@N and NDCG@N.

关键词

序列推荐 / 卷积神经网络 / 注意力机制

Key words

sequence recommendation / convolutional neural network / attention mechanism

引用本文

导出引用
陈继伟,汪海涛,姜瑛,陈星. 基于生成对抗模型的序列推荐算法. 中文信息学报. 2022, 36(7): 143-153
CHEN Jiwei,WANG Haitao,JIANG Ying, CHEN Xing. Sequential Recommendation with Generative Adversarial Networks. Journal of Chinese Information Processing. 2022, 36(7): 143-153

参考文献

[1] Fang H,Zhang D,Shu Y,et al. Deep learning for sequential recommendation[J].ACM Transactions on Information Systems, 2021,39(1): 1-42.
[2] Wang X, He X, Wang M, et al. Neural graph collaborative filtering[C]//Proceedings of the 42nd International ACM SIGIR Conference on Research and Development in Information Retrieval. 2019: 165-174.
[3] He R, Mcauley J. VBPR: Visual bayesian personalized ranking from implicit feedback[C]//Proceedings of the 13th AAAI Conference on Artificial Intelligence. 2016: 12-17.
[4] 魏晓辉,孙冰怡,崔佳旭.基于图神经网络的兴趣活动推荐算法[J].吉林大学学报(工学版),2021,51(01): 278-284.
[5] Wang J,Caverlee J. Recurrent recommendation with local coherence[C]//Proceedings of the 12th ACM International Conference on Web Search and Data Mining. 2019: 564-572.
[6] Li J,Wang Y, Mcauley J. Time interval aware self-attention for sequential recommendation[C]//Proceedings of the WSDM'20: The 13th ACM International Conference on Web Search and Data Mining. ACM, 2020: 322-330.
[7] Wang Z, Wei W, Cong G, et al. Global context enhanced graph neural networks for session-based recommendation[C]//Proceedings of the 43rd International ACM SIGIR Conference on Research and Development in Information Retrieval. 2020: 169-178.
[8] He X,Liao L,Zhang H, et al. Neural collaborative filtering[J]. International World Wide Web Conferences Steering Committee, 2017 : 345-354.
[9] Bakir C.Collaborative filtering with temporal dynamics with using singular value decomposition[J]. Tehnicki Vjesnik, 2018, 25(1): 130-135.
[10] Tan Q,Zhang J, Liu N, et al. Dynamic memory based attention network for sequential recommendation[C]//Proceedings of the International ACM SIGIR Conference on Research and Development in Information Retrieval. 2021: 165-175.
[11] Wu J, Cai R, Wang H. Déjà vu: A contextualized temporal attention mechanism for sequential recommendation[C]//Proceedings of the Web Conference 2020.199-209.
[12] Shani G,Heckerman D,Brafman R I,et al.An MDP-based recommender system[J]. Journal of Machine Learning Research, 2005, 6(1): 1265-1295.
[13] Rendle S, Freudenthaler C, Schmidt-Thieme L. Factorizing personalized Markov chains for next-basket recommendation[C]//Proceedings of the 19th international conference on World wide web. Raleigh, North Carolina, USA. 2010: 811-820.
[14] Hidasi B,Karatzoglou A,Baltrunas L,et al.Session-based recommendations with recurrent neural networks[J].Computer Ence, 2015: 40-44.
[15] Tang J, Wang K. Personalized Top-N sequential recommendation via convolutional sequence embedding [C]// Proceedings of the Eleventh ACM International Conference on Web Search and Data Mining.2018: 565-573.
[16] Vaswani A,Shazeer N,Parmar N,et al. Attention is all you need[J]. arXiv preprint arXiv, 2017: 345-353.
[17] Kang WC, Mcauley J. Self-Attentive sequential recommendation[C]//Proceedings of the IEEE International Conference on Data Mining. IEEE, 2018: 197-206.
[18] Wang J, Yu L, Zhang W, et al. IRGAN: A minimax game for unifying generative and discriminative information retrieval models[C]//Proceedings of the International ACM SIGIR Conference on Research and Development in Information Retrieval 2017: 234-239.
[19] Sun F,Liu J,Wu J,et al.BERT4Rec: Sequential recommendation with bidirectional encoder representations from transformer[C]//Proceedings of the 28th ACM International Conference on Information and Knowledge Management, New York, USA,2019: 1441-1450.
[20] Krizhevsky A, Sutskever I, Hinton G. ImageNet classification with deep convolutional neural networks[J]. Advances in Neural Information Processing Systems, 2012, 25(2).144-156.
[21] Kim Y. Convolutional neural networks for sentence classification[J].arXiv preprint arXiv: 10.3115,2014.
[22] Zhou C, Meng F, Zhou J, et al. Confidence based bidirectional global context aware training framework for neural machine translation[J]. arXiv e-prints, 2022.
[23] F, Maxwell, Harper, et al. The movielens datasets: History and context[J]. Acm Transactions on Interactive Intelligent Systems, 2015(6): 41-53.
[24] Chen T, Yin H, Zheng Y, et al. Learning elastic embeddings for customizing on-device recommenders[C]//Proceedings of the 27th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, 2021: 138-147.
[25] Rendle S, Freudenthaler C, Gantner Z, et al. BPR: Bayesian personalized ranking from implicit feedback[C]//Proceedings of the 25th Conference on Uncertainty in Artificial Intelligence, Canada,AUAI Press, 2009: 695-705.

基金

国家自然科学基金(61462049)
PDF(3075 KB)

1600

Accesses

0

Citation

Detail

段落导航
相关文章

/