基于张量相似度的推荐方法研究

doi:10.3778/j.issn.1002-8331.2112-0149

摘要/Abstract

摘要： 传统推荐方法中将用户建模为向量的建模方式只关注用户单方面偏好，为了弥补此种建模方法的局限性，提出一种将用户建模为矩形的张量建模方法。构建了一个基于融合协同过滤与序列推荐算法的推荐模型，该模型集成了Fastformer模型和键值记忆网络对用户张量进行建模；结合用户张量与目标物品的距离及偏置项对用户张量与目标物品的相似度进行计算。在MovieLens和CiaoDVD数据集上对该模型进行实验验证，实验结果表明，该模型能够关注用户多方面偏好并在推荐结果的精准度上优于基线方法，特别是在HR与NDCG评价指标上分别比现有基线方法平均提高了1.4%、1.95%。

关键词: 推荐系统, 混合推荐, 用户张量, 相似度计算, 用户建模

Abstract: Traditional recommendation methods model users as vectors in a way that focuses only on unilateral user pre-
ferences. In order to compensate for the limitations of this modelling approach, a tensor modelling method is proposed that models the user as a rectangle. Firstly, a recommendation model based on a fusion of collaborative filtering and sequential recommendation algorithms is constructed, which integrates the Fastformer model and key-value memory network to model the user tensor; secondly, the similarity between the user tensor and the target item is calculated by combining the distance between the user tensor and the target item and the bias term. The model is experimentally validated on the MovieLens and CiaoDVD datasets, and the results show that the model is able to focus on multiple user preferences and outperforms the baseline method in terms of accuracy of recommendation results. In particular, the HR and NDCG evaluation metrics are improved on average by 1.4% and 1.95%, respectively, over the existing baseline method.

Key words: recommender systems, hybrid recommendation, user tensor, similarity calculation, user modeling

马蓓欣, 郝斌, 张飞, 高鹭, 任晓颖. 基于张量相似度的推荐方法研究[J]. 计算机工程与应用, 2023, 59(8): 157-166.

MA Beixin, HAO Bin, ZHANG Fei, GAO Lu, REN Xiaoying. Research on Recommendation Method Based on Tensor Similarity[J]. Computer Engineering and Applications, 2023, 59(8): 157-166.

参考文献

[1] 王国霞，刘贺平.个性化推荐系统综述[J].计算机工程与应用，2012，48（7）：66-76.
WANG Guoxia，LIU Heping.Survey of personalized recommendation system[J].Computer Engineering and Applications，2012，48（7）：66-76.
[2] DONG D，ZHENG X，ZHANG R，et al.Recurrent collaborative filtering for unifying general and sequential recommender[C]//Proceedings of IJCAI，2018：3350-3356.
[3] 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，2010：811-820.
[4] NGUYEN T B，TAKASU A.NPE：neural personalized embedding for collaborative filtering[J].arXiv：1805.06563，2018.
[5] COVINGTON P，ADAMS J，SARGIN E.Deep neural networks for youtube recommendations[C]//Proceedings of the 10th ACM Conference on Recommender Systems，2016：191-198.
[6] XUE H J，DAI X，ZHANG J，et al.Deep matrix factori-
zation models for recommender systems[C]//Proceedings of IJCAI，2017：3203-3209.
[7] ZHANG S，LIU H，ZHANG A，et al.Learning user representations with hypercuboids for recommender systems[C]//Proceedings of the 14th ACM International Conference on Web Search and Data Mining，2021：716-724.
[8] REN H，HU W，LESKOVEC J.Query2box：reasoning over knowledge graphs in vector space using box embeddings[J].arXiv：2002.05969，2020.
[9] ABBOUD R，CEYLAN I I，LUKASIEWICZ T，et al.Boxe：a box embedding model for knowledge base completion[J].arXiv：2007.06267，2020.
[10] VASWANI A，SHAZEER N，PARMAR N，et al.Attention is all you need[C]//Advances in Neural Information Processing Systems，2017：5998-6008.
[11] KANG W C，MCAULEY J.Self-attentive sequential recommendation[C]//2018 IEEE International Conference on Data Mining（ICDM），2018：197-206.
[12] CHEN Q，ZHAO H，LI W，et al.Behavior sequence transformer for e-commerce recommendation in alibaba[C]//Proceedings of the 1st International Workshop on Deep Learning Practice for High-Dimensional Sparse Data，2019：1-4.
[13] 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，2019：1441-1450.
[14] 郑诚，王建.联合注意力和自编码器的协同过滤推荐[J].计算机工程与应用，2021，57（10）：139-145.
ZHENG Cheng，WANG Jian.Collaborative filtering recommendation for joint attention and autoencoder[J].Computer Engineering and Applications，2021，57（10）：139-145.
[15] TAY Y，DEHGHANI M，BAHRI D，et al.Efficient transformers：a survey[J].arXiv：2009.06732，2020.
[16] WU C，WU F，QI T，et al.Fastformer：additive attention can be all you need[J].arXiv：2108.09084，2021.
[17] WESTON J，CHOPRA S，BORDES A.Memory networks[J].arXiv：1410.3916，2014.
[18] CHEN X，XU H，ZHANG Y，et al.Sequential recommendation with user memory networks[C]//Proceedings of the Eleventh ACM International Conference on Web Search and Data Mining，2018：108-116.
[19] MILLER A，FISCH A，DODGE J，et al.Key-value memory networks for directly reading documents[J].arXiv：1606.03126，2016.
[20] HUANG J，ZHAO W X，DOU H，et al.Improving sequential recommendation with knowledge-enhanced memory networks[C]//The 41st International ACM SIGIR Conference on Research & Development in Information Retrieval，2018：505-514.
[21] TAY Y，ANH TUAN L，HUI S C.Latent relational metric learning via memory-based attention for collaborative ranking[C]//Proceedings of the 2018 World Wide Web Conference，2018：729-739.
[22] RENDLE S，FREUDENTHALER C，GANTNER Z，et al.BPR：Bayesian personalized ranking from implicit feedback[J].arXiv：1205.2618，2012.
[23] HARPER F M，KONSTAN J A.The movielens datasets：history and context[J].ACM Transactions on Interactive Intelligent Systems，2015，5（4）：1-19.
[24] GUO G，ZHANG J，THALMANN D，et al.Etaf：an extended trust antecedents framework for trust prediction[C]//2014 IEEE/ACM International Conference on Advances in Social Networks Analysis and Mining（ASONAM 2014），2014：540-547.
[25] SHANI G，GUNAWARDANA A.Evaluating recommendation systems[M]//Recommender systems handbook.Boston，MA：Springer，2011：257-297.
[26] DESHPANDE M，KARYPIS G.Item-based top-n recommendation algorithms[J].ACM Transactions on Information Systems，2004，22（1）：143-177.
[27] HE X，LIAO L，ZHANG H，et al.Neural collaborative filtering[C]//Proceedings of the 26th International Conference on World Wide Web，2017：173-182.
[28] 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.
[29] HE R，KANG W C，MCAULEY J.Translation-based recommendation[C]//Proceedings of the Eleventh ACM Conference on Recommender Systems，2017：161-169.
[30] MA C，KANG P，LIU X.Hierarchical gating networks for sequential recommendation[C]//Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining，2019：825-833.
[31] HE R，MCAULEY J.Fusing similarity models with Markov chains for sparse sequential recommendation[C]//2016 IEEE 16th International Conference on Data Mining（ICDM），2016：191-200.