IC-GAR: item co-occurrence graph augmented session-based recommendation

被引:11
作者
Gwadabe, Tajuddeen Rabiu [1 ,2 ]
Liu, Ying [1 ,2 ,3 ]
机构
[1] Univ Chinese Acad Sci, Sch Comp Sci & Technol, Beijing 101400, Peoples R China
[2] Chinese Acad Sci, Data Min & High Performance Comp Lab, Beijing 101400, Peoples R China
[3] Chinese Acad Sci, Key Lab Big Data Min & Knowledge Management, Beijing 100190, Peoples R China
关键词
Session-based recommendation; Graph neural networks; Sequential recommendation; Item co-occurrence graph; SYSTEMS;
D O I
10.1007/s00521-021-06859-x
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Session-based recommendation aims to recommend the next item of an anonymous user session. Previous models consider only the current session and learn both of the user's global and local preferences. These models fail to consider an important source of information, i.e., the co-occurrence pattern of items in different sessions. The co-occurrence patterns elicit the trajectory of other similar users and can improve the recommendation performance. We propose an Item Co-occurrence Graph Augmented Session-based Recommendation (IC-GAR) model, a novel session-based recommendation model that augments the representations of the current session with session co-occurrence patterns. IC-GAR consists of three modules: Encode Module, Session Co-occurrence Module and Prediction Module. The Encoder Module learns both of the user's global and local preference from the current session using Gate Recurrent Units (GRU). The Session Co-occurrence Module uses a modified variant of Graph Convolutional Network (GCN) to model higher order interactions between the item transition patterns in the training sessions. By aggregating the GCN representation of items of the current session, session co-occurrence representation is learned. The Prediction Module decomposes global preference, local preference and session co-occurrence to predict the probability scores of candidate items. Extensive experiments on three publicly available datasets are conducted to demonstrate the effectiveness of IC-GAR. 8.5-39.2% improvement are achieved across datasets in Precision @5, 10 and MRR@5, 10.
引用
收藏
页码:7581 / 7596
页数:16
相关论文
共 73 条
[31]   Using sequential and non-sequential patterns in predictive web usage mining tasks [J].
Mobasher, B ;
Dai, HH ;
Luo, T ;
Nakagawa, M .
2002 IEEE INTERNATIONAL CONFERENCE ON DATA MINING, PROCEEDINGS, 2002, :669-672
[32]  
O'Brien M., 2006, COGSCI
[33]   Star Graph Neural Networks for Session-based Recommendation [J].
Pan, Zhiqiang ;
Cai, Fei ;
Chen, Wanyu ;
Chen, Honghui ;
de Rijke, Maarten .
CIKM '20: PROCEEDINGS OF THE 29TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT, 2020, :1195-1204
[34]  
Qiu RH, 2020, PROCEEDINGS OF THE 43RD INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL (SIGIR '20), P669, DOI [10.1145/3397271.3401109, 10.1145/3397271.34011O9]
[35]   Rethinking the Item Order in Session-based Recommendation with Graph Neural Networks [J].
Qiu, Ruihong ;
Li, Jingjing ;
Huang, Zi ;
Yin, Hongzhi .
PROCEEDINGS OF THE 28TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT (CIKM '19), 2019, :579-588
[36]  
Rendle S., 2012, P 25 C UNCERTAINTY A, P452, DOI DOI 10.5555/1795114.1795167
[37]  
Rendle S., 2010, P 19 INT C WORLD WID, P811, DOI DOI 10.1145/1772690.1772773
[38]  
Sanchez-Gonzalez A, 2018, PR MACH LEARN RES, V80
[39]   The Graph Neural Network Model [J].
Scarselli, Franco ;
Gori, Marco ;
Tsoi, Ah Chung ;
Hagenbuchner, Markus ;
Monfardini, Gabriele .
IEEE TRANSACTIONS ON NEURAL NETWORKS, 2009, 20 (01) :61-80
[40]  
Shani G, 2005, J MACH LEARN RES, V6, P1265