基于邻域感知图神经网络的会话推荐

doi:10.3778/j.issn.1002-8331.2105-0241

摘要/Abstract

摘要： 在基于会话的推荐中，图神经网络及其改进模型将会话内复杂的交互关系建模为图结构并从中捕获项目特征，是现有推荐模型中性能较好的一类方法。然而大多数模型都忽略了不同会话之间可能存在的有效信息，仅对当前会话建模难以利用其他会话，也无法发挥邻域信息的辅助作用。因此提出基于邻域感知图神经网络的会话推荐（NA-GNN）。该模型构建会话层和全局邻域层的图结构捕获项目表示，结合注意力机制聚合两种项目表征，将会话序列之间的互信息最大化地结合到网络训练中。在真实的数据集Yoochoose和Diginetica上进行实验，与性能最优的基准模型相比，模型P@20在Yoochoose上提高了1.85%，在Diginetica上提升了7.19%；MRR@20分别提升了0.48%和8.36%，证明模型的有效性和合理性。

关键词: 邻域感知, 图神经网络, 注意力机制, 会话推荐

Abstract: Graph neural network and its improved models show better performances in session-based recommendation.They convert session as graph structure, and capture item features from transformation relationship between items in graph structure. However, most of the models ignore that there may be useful informations between different sessions, which can support the prediction task. Therefore, it proposes neighborhood aware graph neural networks for session-based recommendation（NA-GNN）. It builds session layer graph and global neighborhood layer graph, and captures item representations from them. Next it uses the attention mechanism to aggregate item representations. So then, it combines the maximum mutual information between sessions into network training. Experiments on two real datasets show that the prediction accuracy is better than others：P@20 increased by 1.85% on Yoochoose and 7.19% on Diginetica; MRR@20 increased by 0.48% and 8.36%, which proves the model is effective and reasonable.

Key words: neighborhood awareness, graph neural network, attention mechanism, session recommendation

何倩倩, 孙静宇, 曾亚竹. 基于邻域感知图神经网络的会话推荐[J]. 计算机工程与应用, 2022, 58(9): 107-115.

HE Qianqian, SUN Jingyu, ZENG Yazhu. Neighborhood Awareness Graph Neural Networks for Session-Based Recommendation[J]. Computer Engineering and Applications, 2022, 58(9): 107-115.

参考文献

[1] WANG S，HU L，WANG Y，et al.Sequential recommender systems：challenges，progress and prospects[J].arXiv：2001.04830，2020.
[2] WANG S J，WANG Y，SHENG Q Z，et al.A survey on session-based recommender systems[J].ACM Computing Surveys，2021，54（7）：1-38.
[3] WU Z，PAN S，CHEN F，et al.A comprehensive survey on graph neural networks[J].IEEE Transactions on Neural Networks and Learning Systems，2020，32（1）：4-24.
[4] JOAQUIN D，NAOHIR I，TOMOKI U.Content-based collaborative information filtering：actively learning to classify and recommend documents[C]//International Workshop on Cooperative Information Agents.Berlin，Heidelberg：Springer，1998：206-215.
[5] KONSTAN J A，MILLER B N，MALTZ D，et al.GroupLens：applying collaborative filtering to Usenet news[J].Communications of the ACM，1997，40（3）：77-87.
[6] LINDEN G，SMITH B，YORK J.Amazon.com recommendations：item-to-item collaborative filtering[J].IEEE Internet Computing，2003，7（1）：76-80.
[7] KOREN Y，BELL R，VOLINSKY C.Matrix factorization techniques for recommender systems[J].Computer，2009，42（8）：30-37.
[8] KARATZOGLOU A，AMATRIAIN X，BALTRUNAS L，et al.Multiverse recommendation：n-dimensional tensor factorization for context aware collaborative filtering[C]//Proceedings of the Fourth ACM Conference on Recommender Systems，2010：79-86.
[9] RENDLE S.Factorization machines[C]//2010 IEEE International Conference on Data Mining，2010：995-1000.
[10] RENDIE 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.
[11] ZHAN Z X，ZHONG L L，LIN J，et al.Sequence-aware similarity learning for next-item recommendation[J].The Journal of Supercomputing，2021，77：7509-7534.
[12] XU S Y，GE Y Q，LI Y Q，et al.Causal collaborative filtering[J].arXiv：2102.01868，2021.
[13] HIDASI B，KARATZOGLOU A，BALTRUNAS L，et al.Session-based recommendations with recurrent neural networks[J].arXiv：1511.06939，2015.
[14] LI J，REN P J，CHEN Z M，et al.Neural attentive session-based recommendation[C]//Proceedings of the 2017 ACM on Conference on Information and Knowledge Management，2017：1419-1428.
[15] LIU Q，ZENG Y F，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，2018：1831-1839.
[16] 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，2019：345-354.
[17] BACH N X，LONG D H，PHUONG T M.Recurrent convolutional networks for session-based recommendations[J].Neurocomputing，2020，411：247-258.
[18] ZHANG J，MA C，MU X，et al.Recurrent convolutional neural network for session-based recommendation[J].Neurocomputing，2021，437：157-167.
[19] GORI M，MONFARDIN G，SCARSELLI F.A new model for learning in graph domains[C]//Proceedings 2005 IEEE International Joint Conference on Neural Networks，2005：729-734.
[20] SCARSELLI F，GORI M，TSOI A C，et al.The graph neural network model[J].IEEE Transactions on Neural Networks，2008，20（1）：61-80.
[21] GALLICCHIO C，MICHELIi A.Graph echo state networks[C]//The 2010 International Joint Conference on Neural Networks（IJCNN），2010：1-8.
[22] BRUNA J，ZAREMBA W，SZLAM A，et al.Spectral networks and locally connected networks on graphs[J].arXiv：1312.6203，2013.
[23] 白铂，刘玉婷，马驰骋，等.图神经网络[J].中国科学：数学，2020，50（3）：367-384.
BAI B，LIU Y T，MA C C，et al.Graph neural network[J].Science in China：Series A，2020，50（3）：367-384.
[24] ZHANG Z，CUI P，ZHU W.Deep learning on graphs：a survey[J].IEEE Transactions on Knowledge and Data Engineering，2020，34（1）：249-270.
[25] WU S，TANG Y，ZHU Y，et al.Session-based recommend-ation with graph neural networks[J].arXiv：1811.00855，2018.
[26] XU C，ZHAO P，LIU Y，et al.Graph contextualized self-attention network for session-based recommendation[C]//Proceedings of the Twenty-Eighth International Conference on Artificial Intelligence，2019.
[27] WANG X，WANG R，SHI C，et al.Multi-component graph convolutional collaborative filtering[C]//Proceedings of the AAAI Conference on Artificial Intelligence，2020：6267-6274.
[28] XIAN X，FANG L，SUN S.ReGNN：a repeat aware graph neural network for session-based recommendations[J].IEEE Access，2020，8：98518-98525.
[29] YANG L，LUO L，XIN L，et al.DAGNN：demand-aware graph neural networks for session-based recommendation[J].arXiv：2105.14428，2021.
[30] SARWAR B.Item-based collaborative filtering recommend-ation algorithms[C]//Proceedings of the 10th International Conference on World Wide Web，2001：285-295.
[31] MOLCHANOV P，TYREE S，KARRAS T，et al.Pruning convolutional neural networks for resource efficient inference[J].arXiv：1611.06440，2016.