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 条
[1]  
[Anonymous], 2001, P 17 C UNC ART INT U
[2]  
Bahdanau D, 2016, Arxiv, DOI arXiv:1409.0473
[3]  
Battaglia PW, 2016, ADV NEUR IN, V29
[4]  
Berg, 2017, AXXIV170602263
[5]   Recommender systems survey [J].
Bobadilla, J. ;
Ortega, F. ;
Hernando, A. ;
Gutierrez, A. .
KNOWLEDGE-BASED SYSTEMS, 2013, 46 :109-132
[6]  
Chaney A.J., 2015, P 9 ACM C RECOMMENDE, P43
[7]  
Chen D, 2019, P 20 IEEE INT C MOB
[8]  
Cho K., 2014, ARXIV14061078, DOI [10.48550/arXiv.1406.1078, DOI 10.3115/V1/D14-1179]
[9]   Ad Click Prediction in Sequence with Long Short-Term Memory Networks: an Externality-aware Model [J].
Deng, Weiwei ;
Ling, Xiaoliang ;
Qi, Yang ;
Tan, Tunzi ;
Manavoglu, Eren ;
Zhang, Qi .
ACM/SIGIR PROCEEDINGS 2018, 2018, :1065-1068
[10]   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