High-order attentive graph neural network for session-based recommendation

被引:10
作者
Sang, Sheng [1 ]
Liu, Nan [1 ]
Li, Wenxuan [2 ]
Zhang, Zhijun [1 ]
Qin, Qianqian [1 ]
Yuan, Weihua [1 ]
机构
[1] Shandong Jianzhu Univ, Sch Comp Sci & Technol, Jinan 250101, Shandong, Peoples R China
[2] Univ Delaware, Coll Engineer, Newark, NJ 19702 USA
基金
中国国家自然科学基金;
关键词
Recommender systems; Session-based recommendation; Graph neural network; Attention mechanism; ALGORITHM;
D O I
10.1007/s10489-022-03170-7
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Recommender systems are becoming a crucial part of several websites. The purpose of session-based recommendations is to predict the next item that users might click based on users' interaction behavior in a session. The latest research on session-based recommendation focuses on using graph neural networks to model transfer relationships between items. However, when the interaction of low-order relationships between adjacent items is insufficient, learning the high-order relationships between non-adjacent items becomes a challenge. Additionally, to distinguish the importance of nodes in the graph, different weights should be assigned to each edge. Therefore, we propose a novel high-order attentive graph neural network (HA-GNN) model for session-based recommendations. In the proposed method, first, we model sessions as graph-structured data. Then, we use the self-attention mechanism to capture the dependencies between items. Next, we use the soft-attention mechanism to learn high-order relationships in a graph. Finally, we update the embeddings of items using a simple fully connected layer. Experiments on two public e-commerce datasets show that HA-GNN has excellent performance.
引用
收藏
页码:16975 / 16989
页数:15
相关论文
共 41 条
[1]  
[Anonymous], 2016, PMLR
[2]  
[Anonymous], 2012, P ACM SIGKDD INT C K, DOI DOI 10.1145/2339530.2339643
[3]  
Ba J. L., 2016, Advances in Neural Information Processing Systems (NeurIPS), P1
[4]  
Bahdanau D, 2016, Arxiv, DOI arXiv:1409.0473
[5]  
Chung J, 2014, ARXIV
[6]  
Devooght, 2017, ARXIV160807400CS
[7]   Graph Neural Networks for Social Recommendation [J].
Fan, Wenqi ;
Ma, Yao ;
Li, Qing ;
He, Yuan ;
Zhao, Eric ;
Tang, Jiliang ;
Yin, Dawei .
WEB CONFERENCE 2019: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2019), 2019, :417-426
[8]  
Feng YF, 2019, PROCEEDINGS OF THE TWENTY-EIGHTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P2301
[9]   Increasing recommended effectiveness with markov chains and purchase intervals [J].
Gu, Wanrong ;
Dong, Shoubin ;
Zeng, Zhizhao .
NEURAL COMPUTING & APPLICATIONS, 2014, 25 (05) :1153-1162
[10]   Recurrent Neural Networks with Top-k Gains for Session-based Recommendations [J].
Hidasi, Balazs ;
Karatzoglou, Alexandros .
CIKM'18: PROCEEDINGS OF THE 27TH ACM INTERNATIONAL CONFERENCE ON INFORMATION AND KNOWLEDGE MANAGEMENT, 2018, :843-852