Incorporating Link Prediction into Multi-Relational Item Graph Modeling for Session-Based Recommendation

被引:20
作者
Wang, Wen [1 ,2 ]
Zhang, Wei [1 ,3 ]
Liu, Shukai [4 ]
Liu, Qi [4 ]
Zhang, Bo [4 ]
Lin, Leyu [4 ]
Zha, Hongyuan [5 ]
机构
[1] East China Normal Univ, Sch Comp Sci & Technol, Shanghai 200062, Peoples R China
[2] Tencent, WeChat Search Applicat Dept, Beijing 518057, Peoples R China
[3] Shanghai Inst AI Educ, Shanghai 200062, Peoples R China
[4] Tencent, WeChat Search Applicat Dept, Beijing 100080, Peoples R China
[5] Chinese Univ Hong Kong, Sch Data Sci, Shenzhen 518172, Guangdong, Peoples R China
基金
中国国家自然科学基金;
关键词
Task analysis; Predictive models; Training; Adaptation models; Social networking (online); Message service; Recurrent neural networks; Session-based recommendation; graph neural networks; user behavior modeling;
D O I
10.1109/TKDE.2021.3111436
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Session-based recommendation aims at predicting the next item that a user is more likely to interact with by a target behavior type. Most of the existing session-based recommendation methods focus on developing powerful representation learning approaches to model items' sequential correlations, whereas they usually encounter the following limitations. First, they only utilize sessions that belong to the target behavior type, neglecting the potential of leveraging other behavior types as auxiliary information for modeling user preference. Second, they separately model item-to-item relations for each session, overlooking to globally characterize the relations across different sessions for better item representations. To overcome these limitations, we first build a Multi-Relational Item Graph (MRIG) involving target and auxiliary behavior types over all sessions. Consequently, a novel Graph Neural Network (GNN) based model is devised to encode MRIG's item-to-item relations into target and auxiliary session-based representations, and adaptively fuse them to represent user interests. To facilitate model training, we further incorporate link prediction into multi-relational item graph modeling, acting as a simple but relevant task to session-based recommendation. The extensive experiments on real-world datasets demonstrate the superiority of the model over diverse and competitive baselines, validating its main components' significant contributions.
引用
收藏
页码:2683 / 2696
页数:14
相关论文
共 56 条
[1]   Dynamic Graph Attention-Aware Networks for Session-Based Recommendation [J].
Abugabah, Ahed ;
Cheng, Xiaochun ;
Wang, Jianfeng .
2020 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2020,
[2]  
Bahdanau D, 2016, Arxiv, DOI [arXiv:1409.0473, DOI 10.48550/ARXIV.1409.0473]
[3]  
Ben-Shimon D, 2015, P 9 ACM C REC SYST, P357, DOI DOI 10.1145/2792838.2798723
[4]  
Bengio Y., 2009, P 26 ANN INT C MACH, P41, DOI DOI 10.1145/1553374.1553380
[5]  
Bruna Joan., 2014, 2 INT C LEARN REPR I
[6]   Social-Enhanced Attentive Group Recommendation [J].
Cao, Da ;
He, Xiangnan ;
Miao, Lianhai ;
Xiao, Guangyi ;
Chen, Hao ;
Xu, Jiao .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2021, 33 (03) :1195-1209
[7]  
Defferrard M, 2016, ADV NEUR IN, V29
[8]  
Ding JT, 2018, PROCEEDINGS OF THE TWENTY-SEVENTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P3343
[9]  
Duvenaudt D, 2015, ADV NEUR IN, V28
[10]   Neural Multi-Task Recommendation from Multi-Behavior Data [J].
Gao, Chen ;
He, Xiangnan ;
Gan, Dahua ;
Chen, Xiangning ;
Feng, Fuli ;
Li, Yong ;
Chua, Tat-Seng ;
Jin, Depeng .
2019 IEEE 35TH INTERNATIONAL CONFERENCE ON DATA ENGINEERING (ICDE 2019), 2019, :1554-1557