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 条
[31]   Experimental Investigation on the Influence of Regional Concrete Spalling on Shield Tunnel Segments [J].
Wang, Shimin ;
Liu, Chuankun ;
Ma, Gaoyu ;
Cao, Songyu ;
Zhang, Junbo ;
Lu, Daiyue ;
He, Chuan .
ADVANCES IN CIVIL ENGINEERING, 2019, 2019
[32]   Neural attention model for recommendation based on factorization machines [J].
Wen, Peng ;
Yuan, Weihua ;
Qin, Qianqian ;
Sang, Sheng ;
Zhang, Zhijun .
APPLIED INTELLIGENCE, 2021, 51 (04) :1829-1844
[33]  
Wu S, 2019, AAAI CONF ARTIF INTE, P346
[34]  
Xu K, 2015, PR MACH LEARN RES, V37, P2048
[35]   TAGNN: Target Attentive Graph Neural Networks for Session-based Recommendation [J].
Yu, Feng ;
Zhu, Yanqiao ;
Liu, Qiang ;
Wu, Shu ;
Wang, Liang ;
Tan, Tieniu .
PROCEEDINGS OF THE 43RD INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL (SIGIR '20), 2020, :1921-1924
[36]   Attention-based context-aware sequential recommendation model [J].
Yuan, Weihua ;
Wang, Hong ;
Yu, Xiaomei ;
Liu, Nan ;
Li, Zhenghao .
INFORMATION SCIENCES, 2020, 510 :122-134
[37]  
Zhang JN, 2019, PROCEEDINGS OF THE TWENTY-EIGHTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P4264
[38]   Collaborative filtering recommendation algorithm integrating time windows and rating predictions [J].
Zhang, Pengfei ;
Zhang, Zhijun ;
Tian, Tian ;
Wang, Yigui .
APPLIED INTELLIGENCE, 2019, 49 (08) :3146-3157
[39]   Hypothalamic response to cocaine cues and cocaine addiction severity [J].
Zhang, Sheng ;
Zhornitsky, Simon ;
Angarita, Gustavo A. ;
Li, Chiang-shan R. .
ADDICTION BIOLOGY, 2020, 25 (01)
[40]   Personalized recommendation algorithm for social networks based on comprehensive trust [J].
Zhang, Zhijun ;
Xu, Gongwen ;
Zhang, Pengfei ;
Wang, Yongkang .
APPLIED INTELLIGENCE, 2017, 47 (03) :659-669