Incorporating Link Prediction into Multi-Relational Item Graph Modeling for Session-based Recommendation
Wen Wang,Wei Zhang,Shukai Liu,Qi Liu,Bo Zhang,Leyu Lin,Hongyuan Zha
DOI: https://doi.org/10.1109/tkde.2021.3111436
IF: 9.235
2021-01-01
IEEE Transactions on Knowledge and Data Engineering
Abstract: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.
computer science, information systems, artificial intelligence,engineering, electrical & electronic