自监督混合图神经网络的会话推荐模型

doi:10.3778/j.issn.1673-9418.2212043

摘要/Abstract

摘要： 基于会话的推荐旨在利用匿名会话预测用户行为。现有基于图神经网络（GNN）的会话推荐算法大多仅针对当前会话提取用户偏好，却忽略了来自其他会话的高阶多元关系从而影响推荐精度。此外，由于会话推荐所采用的短时交互序列包含的信息非常有限，使其更容易受到数据稀疏性的影响。针对上述问题，提出了自监督混合图神经网络会话推荐模型（SHGN）。该模型首先通过将原始数据构建为三个视图来描述会话与物品关系，然后通过多头图注意力网络捕获会话内部物品的低阶转换信息，提出了残差图卷积网络捕获物品和会话的高阶转换信息；最后融合自监督学习（SSL）作为辅助任务，通过最大化不同通道学习到的会话嵌入的互信息，对原始数据进行数据增强从而提升推荐性能。为了验证该方法的有效性，在Tmall、Diginetica、Nowplaying、Yoochoose四个基准数据集上与SR-GNN、GCE-GNN、DHCN等主流基线模型进行了对比实验，实验结果在P@20、MRR@20等性能指标上均取得了一定提升。

关键词: 会话推荐, 多视图建模, 图神经网络, 自监督学习

Abstract: Session-based recommendation aims to predict user actions based on anonymous sessions. Most of the existing session recommendation algorithms based on graph neural network (GNN) only extract user preferences for the current session, but ignore the high-order multivariate relationships from other sessions, which affects the recommendation accuracy. Moreover, session-based recommendation suffers more from the problem of data sparsity due to the very limited short-term interactions. To solve the above problems, this paper proposes a model named self- supervised hybrid graph neural network (SHGN) for session-based recommendation. Firstly, the model describes the relationship between sessions and objects by constructing the original data into three views. Next, a graph attention network is used to capture the low-order transitions information of items within a session, and then a residual graph convolutional network is proposed to mine the high-order transitions information of items and sessions. Finally, self-supervised learning (SSL) is integrated as an auxiliary task. By maximizing the mutual information of session embeddings learnt from different views, data augmentation is performed to improve the recommendation performance. In order to verify the effectiveness of the proposed method, comparative experiments with mainstream baseline models such as SR-GNN, GCE-GNN and DHCN are carried out on four benchmark datasets of Tmall, Diginetica, Nowplaying and Yoochoose, and the results are improved in P@20, MRR@20 and other performance indices.

Key words: session-based recommendation, multi-view modeling, graph neural network, self-supervised learning

章淯淞, 夏鸿斌, 刘渊. 自监督混合图神经网络的会话推荐模型[J]. 计算机科学与探索, 2024, 18(4): 1021-1031.

ZHANG Yusong, XIA Hongbin, LIU Yuan. Self-supervised Hybrid Graph Neural Network for Session-Based Recommendation[J]. Journal of Frontiers of Computer Science and Technology, 2024, 18(4): 1021-1031.

参考文献

[1] LOPS P, GEMMIS M, SEMERARO G. Content-based recommender systems: state of the art and trends[M]//Recommender Systems Handbook. Berlin, Heidelberg: Springer, 2011: 73-105.
[2] SCHAFER J B, FRANKOWIKI D, HERLOCKER J, et al. Collaborative filtering recommender systems[M]//The Adaptive Web. Berlin, Heidelberg: Springer, 2007: 291-324.
[3] 于蒙, 何文涛, 周绪川, 等. 推荐系统综述[J]. 计算机应用, 2022, 42(6): 1898-1913.
YU M, HE W T, ZHOU X C, et al. Review of recommendation systems[J]. Journal of Computer Applications, 2022, 42(6):1898-1913.
[4] ZHANG S, YAO L N, SUN A X, et al. Deep learning based recommender system: a survey and new perspectives[J]. ACM Computing Surveys, 2020, 52(1): 1-38.
[5] HE X, LIAO L, ZHANG H, et al. Neural collaborative filtering[C]//Proceedings of the 26th International Conference on World Wide Web, Perth, Apr 3-7, 2017: 173-182.
[6] GUO H, TANG R, YE Y, et al. DeepFM: a factorization-machine based neural network for CTR prediction[J]. arXiv:1703.04247, 2017.
[7] GAO C, ZHENG Y, LI N, et al. Graph neural networks for recommender systems: challenges, methods, and directions[J]. arXiv:2109.12843, 2021.
[8] 朱志国, 李伟玥, 姜盼, 等. 图神经网络会话推荐系统综述[J]. 计算机工程与应用, 2023, 59(5): 55-69.
ZHU Z G, LI W Y, JIANG P, et al. Survey of graph neural networks in session recommender system[J]. Computer Engineering and Applications, 2023, 59(5): 55-69.
[9] XIA X, YIN H, YU J, et al. Self-supervised graph co-training for session-based recommendation[C]//Proceedings of the 30th ACM International Conference on Information & Knowledge Management, Queensland, Nov 1-5, 2021. New York: ACM, 2021: 2180-2190.
[10] 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, North Carolina, Apr 26-30, 2010. New York: ACM, 2010: 811-820.
[11] 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, Hong Kong, China, May 1-5, 2001: 285-295.
[12] WANG M, REN P, MEI L, et al. A collaborative session-based recommendation approach with parallel memory modules[C]//Proceedings of the 42nd International ACM SIGIR Conference on Research and Development in Information Retrieval, Paris, Jun 28-Jul 1, 2009. New York: ACM, 2019: 345-354.
[13] 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, Singapore, Nov 6-10, 2017. New York: ACM, 2017: 1419-1428.
[14] 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 and Data Mining, London, Aug 19-23, 2018. New York: ACM, 2018: 1831-1839.
[15] KANG W C, MCAULEY J. Self-attentive sequential recommendation[C]//Proceedings of the 2018 IEEE International Conference on Data Mining, Singapore, Nov 17-20, 2018. Piscataway: IEEE, 2018: 197-206.
[16] WU S, TANG Y, ZHU Y, et al. Session-based recommendation with graph neural networks[C]//Proceedings of the 2019 AAAI Conference on Artificial Intelligence, Hawaii, Jan 27-Feb 1, 2019. Menlo Park: AAAI, 2019: 346-353.
[17] XU C, ZHAO P, LIU Y, et al. Graph contextualized self-attention network for session-based recommendation[C]//Proceedings of the 28th International Joint Conference on Artificial Intelligence, Macao, China, Aug 10-16, 2019: 3940-3946.
[18] QIU R, LI J, HUANG Z, et al. Rethinking the item order in session-based recommendation with graph neural networks[C]//Proceedings of the 28th ACM International Conference on Information and Knowledge Management, Beijing, Nov 3-7, 2019. New York: ACM, 2019: 579-588.
[19] 郑楠, 过弋, 李智强, 等. 融合交互注意力和参数自适应的商品会话推荐[J]. 中文信息学报, 2022, 36(11): 131-139.
ZHENG N, GUO Y, LI Z Q, et al. Session-based commodity recommendation through interactive attention and parameter self-adaption[J]. Journal of Chinese Information Processing, 2022, 36(11): 131-139.
[20] BACHMAN P, HJELM R D, BUCHWALTER W. Learning representations by maximizing mutual information across views[C]//Advances?in?Neural?Information?Processing?Systems 32,?Vancouver, Dec?8-14,?2019: 15535-15545.
[21] HASSANI K, KHASAHMADIA H. Contrastive multi-view representation learning on graphs[C]//Proceedings of the 37th International Conference on Machine Learning, Jul 13-18, 2020: 4116-4126.
[22] ZHOU K, WANG H, ZHAO W X, et al. S3-Rec: self-supervised learning for sequential recommendation with mutual information maximization[C]//Proceedings of the 29th ACM International Conference on Information & Knowledge Management. New York: ACM, 2020: 1893-1902.
[23] XIA X, YIN H, YU J, et al. Self-supervised hypergraph convolutional networks for session-based recommendation[C]//Proceedings of the 2021 AAAI Conference on Artificial Intelligence, Columbia, Feb 2-9, 2021. Menlo Park: AAAI,2021: 4503-4511.
[24] 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, Xi’an, Jul 25-30, 2020. New York: ACM, 2020: 169-178.
[25] HE X, DENG K, WANG X, et al. LightGCN: simplifying and powering graph convolution network for recommendation[C]//Proceedings of the 43rd International ACM SIGIR Conference on Research and Development in Information Retrieval, Xi’an, Jul 25-30, 2020. New York: ACM, 2020: 639-648.
[26] VASWANI A, SHAZEER N, PARMAR N, et al. Attention is all you need[C]//Advances in Neural Information Processing Systems 30, Long Beach, Dec 4-9, 2017: 5998-6008.
[27] WANG S, CAO L, WANG Y, et al. A survey on session-based recommender systems[J]. ACM Computing Surveys, 2021, 54(7): 1-38.
[28] OORD A, LI Y, VINYALS O. Representation learning with contrastive predictive coding[J]. arXiv:1807.03748, 2018.
[29] LI A, CHENG Z, LIU F, et al. Disentangled graph neural networks for session-based recommendation[J]. arXiv:2201. 03482, 2022.