基于增强记忆网络的会话推荐算法

doi:10.19678/j.issn.1000-3428.0068444

摘要/Abstract

摘要：

作为协助用户从海量数据中找到匹配兴趣和需求内容的关键, 会话推荐系统的目标是基于匿名会话预测用户的下一个行为。目前常见的推荐算法对于用户整体兴趣表示不足, 而且很少考虑物品间的位置关系。提出一种基于增强记忆网络的会话推荐模型SR-MAN, 旨在分析全局用户兴趣表征和物品顺序问题。首先, 在物品嵌入向量生成时引入位置编码, 凸显不同位置对序列的影响, 再借助神经图灵机存储近期会话信息, 并设计注意力网络学习长期偏好, 结合用户末次点击作为当前兴趣偏好。最后, 通过整合长期与当前偏好进行预测, 推荐用户感兴趣的项目。在算法训练的过程中, 使用贝叶斯个性化排序(BPR)来估计模型参数, 并在3个数据集上的实验验证了所提方法的有效性。

关键词: 会话推荐, 记忆网络, 位置编码, 注意力网络, 兴趣提取

Abstract:

Session-based recommendation systems serve as essential tools for assisting users in identifying matching interests and requirements from large volumes of data. These systems aim to predict the next user actions based on anonymous sessions. However, existing methods inadequately represent the overall interests of a user and frequently neglect the positional relationships among items. To address this limitation, an enhanced memory network-based session recommendation model, SR-MAN, is proposed to analyze global user interest representations and item sequence problems. Initially, the method introduces position encoding during the generation of item embedding vectors to emphasize the impact of different positions on the sequence. Subsequently, a neural Turing machine is employed to store recent session information, and an attention network is designed to learn long-term user preferences by integrating the most recent user interaction as the current interest indicator. Finally, the method integrates long-term and current preferences to predict and recommend items of interest. Bayesian Personalized Ranking(BPR) is employed to estimate the model parameters. Experiments on three datasets demonstrate the effectiveness of the proposed method.

Key words: session-based recommendation, memory network, position encoding, attention network, interest extraction

魏星, 孙浩, 曹健, 祝晓斌. 基于增强记忆网络的会话推荐算法[J]. 计算机工程, 2024, 50(12): 83-89.

WEI Xing, SUN Hao, CAO Jian, ZHU Xiaobin. Session-based Recommendation Algorithm Based on Memory Augmented Network[J]. Computer Engineering, 2024, 50(12): 83-89.

https://www.ecice06.com/CN/Y2024/V50/I12/83

图/表 6

参考文献 27

1	LIN Z H, TIAN C X, HOU Y P, et al. Improving graph collaborative filtering with neighborhood-enriched contrastive learning[C]//Proceedings of the 2022 ACM Web Conference. New York, USA: ACM Press, 2022: 2320-2329.
2	YU W , LI S J . Recommender systems based on multiple social networks correlation. Future Generation Computer Systems, 2018, 87, 312- 327. doi: 10.1016/j.future.2018.04.079
3	SARWAR B, KARYPIS G, KONSTAN J, 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.
4	WILHELM F. Matrix factorization for collaborative filtering is just solving an adjoint latent Dirichlet allocation model after all[C]//Proceedings of the 15th ACM Conference on Recommender Systems. New York, USA: ACM Press, 2021: 55-62.
5	ZHANG Z Y , LIU Y , ZHANG Z J , et al. Fused matrix factorization with multi-tag, social and geographical influences for POI recommendation. World Wide Web, 2019, 22 (3): 1135- 1150. doi: 10.1007/s11280-018-0579-9
6	SHANI G , HECKERMAN D , BRAFMAN R I . An MDP-based recommender system. Journal of Machine Learning Research, 2005, 6, 1265- 1295.
7	HIDASI B, KARATZOGLOU A, BALTRUNAS L, et al. Session-based recommendations with recurrent neural networks[C]//Proceedings of the 4th International Conference on Learning Representations. Berlin, Germany: Springer, 2016: 487-498.
8	HE X N, LIAO L Z, ZHANG H W, et al. Neural collaborative filtering[C]//Proceedings of the 26th International Conference on World Wide Web. Republic and Canton of Geneva, Switzerland: International World Wide Web Conferences Steering Committee, 2017: 173-182.
9	吴志强, 解庆, 李琳, 等. 基于多模态融合的图神经网络推荐算法. 计算机工程, 2024, 50 (1): 91- 100. URL
	WU Z Q , XIE Q , LI L , et al. Graph neural network recommendation algorithm based on multimodal fusion. Computer Engineering, 2024, 50 (1): 91- 100. URL
10	陈昱瑾, 王晶, 武志昊, 等. 基于图卷积网络融合群组关系的社会化推荐方法. 计算机工程, 2023, 49 (5): 112- 121. URL
	CHEN Y J , WANG J , WU Z H , et al. Social recommendation method integrating group relationships based on graph convolution network. Computer Engineering, 2023, 49 (5): 112- 121. URL
11	TAN Y K, XU X, LIU Y. Improved recurrent neural networks for session-based recommendations[C]//Proceedings of the 1st Workshop on Deep Learning for Recommender Systems. New York, USA: ACM Press, 2016: 17-22.
12	LI J, REN P, CHEN Z, et al. Neural attentive session-based recommendation[C]//Proceedings of the 2017 ACM Conference on Information and Knowledge Management. New York, USA: ACM Press, 2017: 1419-1428.
13	LIU Q, ZENG Y, MOKHOSI R, et al. STAMP: short-term attention/memory priority model for session-based recommendation[C]//Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. New York, USA: ACM Press, 2018: 1831-1839.
14	WU S , TANG Y , ZHU Y , et al. Session-based recommendation with graph neural networks. Proceedings of the AAAI Conference on Artificial Intelligence, 2019, 33 (1): 346- 353. doi: 10.1609/aaai.v33i01.3301346
15	黄震华, 林小龙, 孙圣力, 等. 会话场景下基于特征增强的图神经推荐方法. 计算机学报, 2022, 45 (4): 766- 780.
	HUANG Z H , LIN X L , SUN S L , et al. Feature augmentation based graph neural recommendation method in session scenarios. Chinese Journal of Computers, 2022, 45 (4): 766- 780.
16	陈聪, 张伟, 王骏. 带有时间预测辅助任务的会话式序列推荐. 计算机学报, 2021, 44 (9): 1841- 1853.
	CHEN C , ZHANG W , WANG J . Session-based sequential recommendation with auxiliary time prediction. Chinese Journal of Computers, 2021, 44 (9): 1841- 1853.
17	WESTON J, CHOPRA S, BORDES A. Memory networks[C]//Proceedings of the 3rd International Conference on Learning Representations. [S. l. ]: ICLR, 2015: 1-10.
18	MALEKMOHAMADI F S , SAFI E F , KARIMIAN K M . A review on Neural Turing Machine(NTM). SN Computer Science, 2020, 1 (6): 333.
19	SUKHBAATAR S, WESTON J, FERGUS R. End-to-end memory networks[C]//Proceedings of the 28th International Conference on Neural Information Processing Systems. New York, USA: ACM Press, 2015: 2440-2448.
20	CHEN X, XU H T, ZHANG Y F, et al. Sequential recommendation with user memory networks[C]//Proceedings of the 11th ACM International Conference on Web Search and Data Mining. New York, USA: ACM Press, 2018: 108-116.
21	HUANG J, ZHAO W X, DOU H J, et al. Improving sequential recommendation with knowledge-enhanced memory networks[C]//Proceedings of the 41st International ACM SIGIR Conference on Research & Development in Information Retrieval. New York, USA: ACM Press, 2018: 505-514.
22	CAI Z P , HE Z B , GUAN X , et al. Collective data-sanitization for preventing sensitive information inference attacks in social networks. IEEE Transactions on Dependable and Secure Computing, 2016, 15 (4): 577- 590.
23	曾亚竹, 孙静宇, 何倩倩. 融合BiGRU和记忆网络的会话推荐算法. 计算机工程与设计, 2023, 44 (2): 335- 342.
	ZENG Y Z , SUN J Y , HE Q Q . Session-based recommendation algorithm combined with BiGRU and memory network. Computer Engineering and Design, 2023, 44 (2): 335- 342.
24	SONG B, CAO Y, ZHANG W F, et al. Session-based recommendation with hierarchical memory networks[C]//Proceedings of the Proceedings of the 28th ACM International Conference on Information and Knowledge Management. New York, USA: ACM Press, 2019: 2181-2184.
25	WEN Y , KANG S T , ZENG Q T , et al. Session-based recommendation with GNN and time-aware memory network. Mobile Information Systems, 2022, (6): 1879367.
26	RENDLE S, FREUDENTHALER C, GANTNER Z, et al. Bayesian personalized ranking from implicit feedback[EB/OL]. [2023-09-30]. https://arxiv.org/abs/1510.01784v1.
27	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. New York, USA: ACM Press, 2010: 811-820.

[1]	屈潇雅, 李兵, 温立强. 面向行政执法案件文本的事件抽取研究[J]. 计算机工程, 2024, 50(9): 63-71.
[2]	党小超, 刘涧, 董晓辉, 祝忠彦, 李芬芳. 面向不平衡数据的机械设备故障命名实体识别[J]. 计算机工程, 2024, 50(9): 104-112.
[3]	王炼红, 林飞鹏, 李潇瑶, 谌桂枝, 周莉. 融入课程知识图谱的KMAKT预测[J]. 计算机工程, 2024, 50(7): 23-31.
[4]	周炫余, 吴莲华, 郑勤华, 肖天星, 王紫璇, 张思敏. 联合语义提示和记忆增强的弱监督跳绳视频异常检测方法[J]. 计算机工程, 2024, 50(7): 87-95.
[5]	周昭辰, 方清茂, 吴晓红, 胡平, 何小海. 基于MacBERT与对抗训练的机器阅读理解模型[J]. 计算机工程, 2024, 50(5): 41-50.
[6]	丁国辉, 刘宇琪, 王言开, 耿施展, 姜天昊. 基于翻转网络的低相关性序列数据预测研究[J]. 计算机工程, 2024, 50(2): 78-90.
[7]	杜海军, 余粟. 基于时空图注意力网络的服务机器人动态避障[J]. 计算机工程, 2024, 50(2): 105-112.
[8]	卢苡锋, 王霄. 基于二次分解和IDBO-DABiLSTM的短期风电功率预测模型[J]. 计算机工程, 2024, 50(12): 99-109.
[9]	詹泽慧, 钟煊妍, 邹萱萱, 骆丽霞. 基于BERT-HAN增强人机对话的计算思维评估模型[J]. 计算机工程, 2024, 50(12): 110-123.
[10]	史昕, 曹凤腾, 纪艺, 马峻岩. 基于多尺度时空特征与软注意力机制的交通流预测方法[J]. 计算机工程, 2024, 50(12): 346-357.
[11]	袁平宇, 邱林. 基于BERT的多模型融合的Web攻击检测方法[J]. 计算机工程, 2024, 50(11): 197-206.
[12]	唐卓然, 柳毅. 基于词汇融合和依存关系的中文命名实体识别[J]. 计算机工程, 2024, 50(10): 145-153.
[13]	周莎, 申国伟, 郭春. 基于安全知识图谱与逆向特征的弱点信息补全[J]. 计算机工程, 2024, 50(1): 145-155.
[14]	王志宝, 江树涛, 李菲, 高俊涛, 马强, 杨彬. 基于多邻域感知的石油数据资产图谱实体对齐[J]. 计算机工程, 2024, 50(1): 339-347.
[15]	刘晓黎, 王轶彤. 基于自监督学习的多密度图会话推荐[J]. 计算机工程, 2023, 49(9): 60-68, 78.

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

选择包含的内容