基于物品嵌入向量的会话型推荐算法

doi:10.19678/j.issn.1000-3428.0058110

摘要/Abstract

摘要： 传统基于会话的推荐算法主要利用点击物品的时序信息进行建模，忽略了挖掘物品的特征信息，且未利用物品之间的相似性。为提升推荐效果，提出一种新的基于会话的推荐算法SR-I2V。通过Skip-gram模型和层次softmax优化方法学习物品的嵌入向量，由意图递进公式对已发生的物品点击提取出意图特征向量，并根据特征向量相似度计算出每个候选项的推荐分数。实验结果表明，与I2I、PoP和S-POP等传统基于会话的推荐算法相比，该算法在Yoochoose和Diginetica两个数据集上的推荐召回率分别提高了至少4.67个百分点和3.97个百分点，平均倒数排名指标也有相应提高。

Abstract: The existing session-based recommendation algorithms mainly leverage the temporal information of the items while ignoring their feature information and the similarity between the items.In order to improve the recommendation effect,this paper proposes a novel session-based recommendation algorithm named Session-based Recommendation with Item2Vec(SR-I2V).The algorithm learns the embedding vectors of the items through the skip-gram model and the hierarchical softmax optimization method.Then it uses the intent progression formula to extract the intent feature vectors from the click events that have occurred,and calculates the recommendation score of each candidate item based on the similarity between the feature vectors.Experimental results show that compared with traditional session-based recommendation algorithms such as I2I,PoP,and S-POP,the proposed algorithm increases the recommendation recall rate by at least 4.67 percentage points on Yoochoose and 3.97 percentage points on Diginetica.In addition,it improves the mean reciprocal rank.

Key words: recommendation algorithm, Recurrent Neural Network(RNN), embedding vector, hierarchical softmax, intent progression

中图分类号:

TP391.3

陈恩华, 方宝富. 基于物品嵌入向量的会话型推荐算法[J]. 计算机工程, 2021, 47(7): 74-80.

CHEN Enhua, FANG Baofu. Session-Based Recommendation Algorithm with Item2Vec[J]. Computer Engineering, 2021, 47(7): 74-80.

http://www.ecice06.com/CN/Y2021/V47/I7/74

图/表 8

20210721090021

20210721090025

20210721090029

20210721090033

20210721090037

20210721090042

20210721090045

20210721090051

参考文献

[1] RENDLS S,FREUDENTHALER C,SCHMIDTTHIEME L,et al.Factorizing personalized Markov chains for next-basket recommendation[C]//Proceedings of the 19th International Conference on World Wide Web.New York,USA:ACM Press,2010:811-820.
[2] 高茂庭,徐彬源.基于循环神经网络的推荐算法[J].计算机工程,2019,45(8):198-202,209. GAO M T,XU B Y.Recommendation algorithm based on recurrent neural network[J].Computer Engineering,2019,45(8):198-202,209.(in Chinese)
[3] BARKAN O,KOENIGSTEIN N.Item2Vec:neural item embedding for collaborative filtering[C]//Proceedings of IEEE International Workshop on Machine Learning for Signal Processing.Washington D.C.,USA:IEEE Press,2016:1-6.
[4] KOREN Y.Factorization meets the neighborhood:a multifaceted collaborative filtering model[C]//Proceedings of the 14th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining.New York,USA:ACM Press,2008:426-434.
[5] SARWAR B M,KARYPIS G,KONSTAN J A,et al.Item-based collaborative filtering recommendation algorithms[C]//Proceedings of the 10th International Conference on World Wide Web.New York,USA:ACM Press,2001:285-295.
[6] HIDASI B,KARATZOGLOU A,BALTRUNAS L,et al.Session-based recommendations with recurrent neural networks[EB/OL].[2020-03-10].https://arxiv.org/abs/1511.06939.
[7] KOREN Y,BELL R M,VOLINSKY C,et al.Matrix factorization techniques for recommender systems[J].IEEE Computer,2009,42(8):30-37.
[8] HIDASI B,TIKK D.Fast ALS-based tensor factorization for context-aware recommendation from implicit feedback[C]//Proceedings of Joint European Conference on Machine Learning and Knowledge Discovery in Databases.Berlin,Germany:Springer,2012:67-82.
[9] LINDEN G,SMITH B R,YORK J C,et al.Amazon.com recommendations:item-to-item collaborative filtering[J].IEEE Internet Computing,2003,7(1):76-80.
[10] SHANI G,HECKERMAN D,BRAFMAN R I,et al.An MDP-based recommender system[J].Journal of Machine Learning Research,2005,21:1265-1295.
[11] MIKOLOV T,SUTSKEVER I,CHEN K,et al.Distributed representations of words and phrases and their compositionality[C]//Proceedings of NIPS'13.Cambridge,USA:MIT Press,2013:3111-3119.
[12] MIKOLOV T,KARAFIAT M,BURGET L,et al.Recurrent neural network based language model[C]//Proceedings of Conference of the International Speech Communication Association.Chiba,Japan:[s.n.],2010:1045-1048.
[13] CHO K,VAN MERRIENBOER B,GULCEHRE C,et al.Learning phrase representations using RNN encoder-decoder for statistical machine translation[EB/OL].[2020-03-10].https://arxiv.org/abs/1406.1078.
[14] SERBAN I V,SORDONI A,BENGIO Y,et al.Building end-to-end dialogue systems using generative hierarchical neural network models[C]//Proceedings of the 30th AAAI Conference on Artificial Intelligence.[S.1.]:AAAI Press,2016:3776-3784.
[15] MAO J H,XU W,YANG Y,et al.Deep captioning with multimodal recurrent neural networks (m-RNN)[EB/OL].[2020-03-10].https://arxiv.org/abs/1412.6632.
[16] ZHANG Yuyu,DAI Hanjun,XU Chang,et al.Sequential click prediction for sponsored search with recurrent neural networks[C]//Proceedings of AAAI Conference on Artificial Intelligence.[S.1.]:AAAI Press,2014:1369-1376.
[17] 丁勇,王翔,蒋翠清.基于Location2vec的地点推荐算法[J].计算机工程,2019,45(7):212-216. DING Y,WANG X,JIANG C Q.Location recommendation algorithm based on kocation2vec[J].Computer Engineering,2019,45(7):212-216.(in Chinese)
[18] LIU Q,WU S,WANG L,et al.Predicting the next location:a recurrent model with spatial and temporal contexts[C]//Proceedings of the 30th AAAI Conference on Artificial Intelligence.[S.l.]:AAAI Press,2016:194-200.
[19] YU F,LIU Q,WU S,et al.A dynamic recurrent model for next basket recommendation[C]//Proceedings of International ACM SIGIR Conference on Research and Development in Information Retrieval.New York,USA:ACM Press,2016:729-732.
[20] GRBOVIC M,CHENG H.Real-time personalization using embeddings for search ranking at airbnb[C]//Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining.New York,USA:ACM Press,2018:311-320.
[21] MIKOLOV T,CHEN K,CORRADO G S,et al.Efficient estimation of word representations in vector space[C]//Proceedings of IEEE International Conference on Learning Representations.Washington D.C.,USA:IEEE Press,2013:156-168.
[22] REDLE S,FREUDENTHALER C,GANTNER Z,et al.BPR:Bayesian personalized ranking from implicit feedback[C]//Proceedings of the 25th IEEE International Conference on Uncertainty in Artificial Intelligence.Washington D.C.,USA:IEEE Press,2009:452-461.

选择文件类型/文献管理软件名称

选择包含的内容