Global Context Enhanced Graph Neural Networks for Session-based Recommendation

被引:391
作者
Wang, Ziyang [1 ]
Wei, Wei [1 ]
Cong, Gao [2 ]
Li, Xiao-Li [3 ]
Mao, Xian-Ling [4 ]
Qiu, Minghui [5 ]
机构
[1] Huazhong Univ Sci & Technol, Sch Comp Sci & Technol, Cognit Comp & Intelligent Informat Proc CCIIP Lab, Wuhan, Peoples R China
[2] Nanyang Technol Univ, Sch Comp Engn, Singapore, Singapore
[3] Inst Infocomm Res, Singapore, Singapore
[4] Beijing Inst Technol, Sch Comp Sci & Technol, Beijing, Peoples R China
[5] Alibaba Grp, Hangzhou, Peoples R China
来源
PROCEEDINGS OF THE 43RD INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL (SIGIR '20) | 2020年
基金
中国国家自然科学基金;
关键词
Recommendation system; Session-based recommendation; Graph neural network;
D O I
10.1145/3397271.3401142
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Session-based recommendation (SBR) is a challenging task, which aims at recommending items based on anonymous behavior sequences. Almost all the existing solutions for SBR model user preference only based on the current session without exploiting the other sessions, which may contain both relevant and irrelevant item-transitions to the current session. This paper proposes a novel approach, called Global Context Enhanced Graph Neural Networks (GCE-GNN) to exploit item transitions over all sessions in a more subtle manner for better inferring the user preference of the current session. Specifically, GCE-GNN learns two levels of item embeddings from session graph and global graph, respectively: (i) Session graph, which is to learn the session-level item embedding by modeling pah wise item-transitions within the current session; and (ii) Global graph, which is to learn the global-level item embedding by modeling pairwise item-transitions over all sessions. In GCE-GNN, we propose a novel global-level item representation learning layer, which employs a session-aware attention mechanism to recursively incorporate the neighbors' embeddings of each node on the global graph. We also design a session-level item representation learning layer, which employs a GNN on the session graph to learn session-level item embeddings within the current session. Moreover, GCE-GNN aggregates the learnt item representations in the two levels with a soft attention mechanism. Experiments on three benchmark datasets demonstrate that GCE-GNN outperforms the state-of-the-art methods consistently.
引用
收藏
页码:169 / 178
页数:10
相关论文
共 26 条
[1]  
Hamilton WL, 2017, ADV NEUR IN, V30
[2]  
Hidasi Balazs, 2016, P INT C LEARN REPR
[3]   When Recurrent Neural Networks meet the Neighborhood for Session-Based Recommendation [J].
Jannach, Dietmar ;
Ludewig, Malte .
PROCEEDINGS OF THE ELEVENTH ACM CONFERENCE ON RECOMMENDER SYSTEMS (RECSYS'17), 2017, :306-310
[4]   Self-Attentive Sequential Recommendation [J].
Kang, Wang-Cheng ;
McAuley, Julian .
2018 IEEE INTERNATIONAL CONFERENCE ON DATA MINING (ICDM), 2018, :197-206
[5]  
Kipf T.N., 2017, INT C LEARN REPR ICL, P1
[6]   Neural Attentive Session-based Recommendation [J].
Li, Jing ;
Ren, Pengjie ;
Chen, Zhumin ;
Ren, Zhaochun ;
Lian, Tao ;
Ma, Jun .
CIKM'17: PROCEEDINGS OF THE 2017 ACM CONFERENCE ON INFORMATION AND KNOWLEDGE MANAGEMENT, 2017, :1419-1428
[7]  
Li Y., 2016, 4 INT C LEARN REPR I
[8]   STAMP: Short-Term Attention/Memory Priority Model for Session-based Recommendation [J].
Liu, Qiao ;
Zeng, Yifu ;
Mokhosi, Refuoe ;
Zhang, Haibin .
KDD'18: PROCEEDINGS OF THE 24TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2018, :1831-1839
[9]   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
[10]  
Rendle S., 2010, P 19 INT C WORLD WID, P811, DOI [DOI 10.1145/1772690.1772773, 10.1145/1772690.1772773]