AFGSRec：一种自适应融合全局协同特征的社交推荐模型

蔡晓东, 曾志杨

doi:10.12141/j.issn.1000-565X.220180

华南理工大学学报(自然科学版) >

2022 , Vol. 50 >Issue 12: 71 - 79

DOI: https://doi.org/10.12141/j.issn.1000-565X.220180

计算机科学与技术

AFGSRec：一种自适应融合全局协同特征的社交推荐模型

展开

桂林电子科技大学信息与通信学院，广西桂林 541000

蔡晓东（1971-），男，博士，研究员，主要从事大数据挖掘和自然语言处理研究。

收稿日期: 2022-04-06

网络出版日期: 2022-07-15

基金资助

广西创新驱动发展专项(AA20302001)

收起

AFGSRec: A Social Recommendation Model Based on Adaptive Fusion of Global Collaborative Features

Expand

School of Information and Communication，Guilin University of Electronic Technology，Guilin 541000，Guangxi，China

蔡晓东（1971-），男，博士，研究员，主要从事大数据挖掘和自然语言处理研究。

Received date: 2022-04-06

Online published: 2022-07-15

Supported by

the Guangxi Innovation-driven Development Special Project(AA20302001)

Fold

摘要

以往的序列推荐方法通常从近期交易记录中捕获用户的消费偏好，忽略了全局交易信息和好友偏好对用户交易行为的影响，导致模型的推荐结果不够准确。针对以上问题，文中提出了一种自适应融合全局协同特征的社交推荐模型AFGSRec。首先，用异质图神经网络建模社交网络中的用户、历史交易信息，以捕获全局协同特征和好友之间的社交影响；接着，设计了一种基于选择机制的门图神经网络，以有效过滤与当前序列无关的节点转换信息，更准确地捕获用户当前偏好；然后，提出了一种自适应的特征融合方法，以动态捕获全局协同特征对用户偏好的影响，提高系统的推荐准确率；最后，将周期动态学习率用于模型训练，以更好地处理鞍点，提升模型的收敛速度。实验结果表明：AFGSRec具有较好的鲁棒性，命中率（HR）和平均倒数排名（MRR）都优于当前领先模型SERec，在Gowalla数据集上，HR@10、HR@20分别提升了1.91%和1.15%，MRR@10、MRR@20分别提升了5.05%和4.83%；在Delicious数据集上，HR@10、HR@20分别提升了2.45%和1.19%，MRR@10、MRR@20分别提升了4.84%和4.32%。

本文引用格式

蔡晓东, 曾志杨 . AFGSRec：一种自适应融合全局协同特征的社交推荐模型[J]. 华南理工大学学报(自然科学版), 2022 , 50(12) : 71 -79 . DOI: 10.12141/j.issn.1000-565X.220180

Abstract

Previous session-based recommendation systems usually capture users’ consumption preferences from their recent transaction records, and this method ignores the influence of global transaction information and friends’ preferences on users’ transaction behavior, resulting in less accurate recommendation results of the model. To solve the problem, this paper proposed a social recommendation model AFGSRec based on an adaptive fusion of global collaborative features. Firstly, a heterogeneous graph neural network was used to model users and their historical transaction information on the social network for capturing global collaborative features and social influence among friends. Secondly, this paper designed a graph neural network based on a selection mechanism that effectively filters out the node transition features irrelevant to the current session and captures user preferences more accurately. Thirdly, an adaptive fusion method was designed to capture the impact of global collaborative features on users’ current preferences dynamically and improve the model’s recommendation accuracy. Finally, this paper used a dynamic cyclical learning rate to help the model better handle saddle points during the training process to improve the convergence speed of model AFGSRec. The experimental results show that AFGSRec is robust; both the HR (Hit Rate) and MRR (Mean Reciprocal Rank) of AFGSRec outperform the state-of-art model SERec. On the Gowalla dataset, HR@10 and HR@20 are increased by 1.91% and 1.15%, respectively; MRR@10 and MRR@20 are increased by 5.05% and 4.83%, respectively. On the Delicious dataset, HR@10 and HR@20 are increased by 2.45% and 1.19%, respectively; MRR@10 and MRR@20 are increased by 4.84% and 4.32%, respectively.

Key words： graph neural network; recommendation system; social networking; global collaborative feature; dynamic cyclical learning rate

参考文献

1	WANG S， HU L， WANG Y，et al ．Sequential recommender systems：challenges，progress and prospects ［C］∥ Proceedings of the Twenty-Eighth International Joint Conference on Artificial Intelligence Survey Track.Macao：Morgan Kaufmann，2019：6332-6338.
2	HIDASI B， KARATZOGLOU A， BALTRUNAS L，et al ．Session-based recommendations with recurrent neural networks ［C］∥ Proceedings of the 4th International Confe-rence on Learning Representations．San Juan：OpenReview.net，2016：1-10.
3	CHO K， MERRIENBOER B V， GULCEHRE C，et al ．Learning phrase representations using RNN encoder-decoder for statistical machine translation ［C］∥ Proceedings of the 2014 Conference on Empirical Methods in Natural Language Processing．Doha：Association for Computational Linguistics，2014：1724-1734.
4	BAHDANAU D， CHO K， BENGIO Y ．Neural machine translation by jointly learning to align and translate ［C］∥ Proceedings of the 3rd International Conference on Lear-ning Representations．San Diego：OpenReview.net，2015：1-15.
5	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：ACM，2017：1419-1428.
6	REN P， CHEN Z， LI J，et al ．RepeatNet：a repeat aware neural recommendation machine for session-based recommendation ［C］∥ Proceedings of the 33rd AAAI Conference on Artificial Intelligence．Honolulu：AAAI，2019：4806-4813.
7	JI W， WANG K， WANG X，et al ．Sequential recommender via time-aware attentive memory network ［C］∥ Proceedings of the 29th ACM International Conference on Information and Knowledge Management．New York：ACM，2020：565-574.
8	ZHANG J， MA C， MU X，et al ．Recurrent convolutional neural network for session-based recommendation ［J］．Neurocomputing，2021，437：157-167.
9	WU S， TANG Y， ZHU Y，et al ．Session-based recommendation with graph neural networks ［C］∥ Proceedings of the 33rd AAAI Conference on Artificial Intelligence．Honolulu：AAAI，2019：346-353.
10	LI Y， ZEMEL R， BROCKSCHMIDT M，et al ．Gated graph sequence neural networks ［C］∥ Proceedings of the 4th International Conference on Learning Representations．San Juan：OpenReview.net，2016：11-30.
11	WANG Z， WEI W， CONG G，et al ．Global context enhanced graph neural networks for session-based re-commendation ［C］∥ Proceedings of the 43rd International ACM SIGIR Conference on Research and Deve-lopment in Information Retrieval．New York：ACM，2020：169-178.
12	ZHANG C， SONG D， HUANG C，et al ．Heterogeneous graph neural network ［C］∥ Proceedings of the 25th ACM SIGKDD International Conference on Know-ledge Discovery & Data Mining．Anchorage：ACM，2019：793-803.
13	XU L， XI W D， WANG C D ．Session-based recommendation with heterogeneous graph neural networks ［C］∥ Proceedings of 2021 International Joint Confe-rence on Neural Networks．Shenzhen：IEEE，2021：1-8.
14	PANG Y， WU L， SHEN Q，et al ．Heterogeneous global graph neural networks for personalized session-based recommendation ［C］∥ Proceedings of the Fifteenth ACM International Conference on Web Search and Data Mining．New York：ACM，2022：775-783.
15	SONG W， XIAO Z， WANG Y，et al ．Session-based social recommendation via dynamic graph attention networks ［C］∥ Proceedings of the Twelfth ACM International Conference on Web Search and Data Mining．Melbourne：ACM，2019：555-563.
16	GU P， HAN Y， GAO W，et al ．Enhancing session-based social recommendation through item graph embedding and contextual friendship modeling［J］．Neurocomputing，2021，419：190-202.
17	CHEN T， WONG R C W ．An efficient and effective framework for session-based social recommendation ［C］∥ Proceedings of the Forteenth ACM International Conference on Web Search and Data Mining．New York：ACM，2021：400-408.
18	MA C， KANG P， LIU X ．Hierarchical gating networks for sequential recommendation ［C］∥ Proceedings of the 25th ACM SIGKDD International Conference on Know-ledge Discovery & Data Mining．Anchorage：ACM，2019：825-833.
19	SMITH L N ．Cyclical learning rates for training neural networks ［C］∥ Proceedings of 2017 IEEE Winter Conference on Applications of Computer Vision．Santa Rosa：IEEE，2017：464-472.
20	WANG C， ZHU Y， LIU H，et al ．Enhancing user interest modeling with knowledge-enriched itemsets for sequential recommendation ［C］∥ Proceedings of the 30th ACM International Conference on Information & Know-ledge Management．New York：ACM，2021：1889-1898.
21	YU F， ZHU Y， LIU Q，et al ．TAGNN：target attentive graph neural networks for session-based recommendation ［C］∥ Proceedings of the 43rd International ACM SIGIR Conference on Research and Development in Information Retrieval．New York：ACM，2020：1921-1924.
22	PAN Z， CAI F， CHEN W，et al ．Star graph neural networks for session-based recommendation ［C］∥ Proceedings of the 29th ACM International Conference on Information & Knowledge Management．New York：ACM，2020：1195-1204.
23	CHEN T， WONG R C W ．Handling information loss of graph neural networks for session-based recommendation ［C］∥ Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining．New York：ACM，2020：1172-1180.
24	宋美佳，贾鹤鸣，林志兴，等．自适应学习率梯度下降的优化算法［J］．三明学院学报，2021，38（6）：36-44.
24	SONG Meijia， JIA Heming， LIN Zhixing，et al ．Optimization research of gradient descent algorithm based on adaptive learning rate ［J］．Journal of Sanming University，2021，38（6）：36-44.
25	DAUPHIN Y， de VRIES H， BENGIO Y ．Equilibrated adaptive learning rates for non-convex optimization ［C］∥ Proceedings of the 28th International Conference on Neural Information Processing Systems．Cambridge：MIT Press，2015：1504-1512.

Options

文章导航

模态框（Modal）标题

摘要

本文引用格式

Abstract

参考文献