TGIE4REC: enhancing session-based recommendation with transition and global information

Shiwei Gao,Jingyu Wang,Yufeng Zeng,Xiaohui Dong
DOI: https://doi.org/10.1007/s11227-024-05897-1
IF: 3.3
2024-01-29
The Journal of Supercomputing
Abstract:Predicting the next most likely interactive item based on the current session is the goal of session-based recommendation (SBR). In order to model the adjacent item transition information from previous session sequences and the current session sequences, the most advanced techniques in SBR use graph neural networks and attention mechanisms. Position-aware attention is used to incorporate the reversed position information in an item in order to learn the importance of each item in the session when generating the session representation. However, these methods have certain drawbacks. First, using data from previous sessions always introduces uncorrelated items (noise). Second, learning the sequence transition relations between items in the session sequence is challenging due to reverse position coding. This study presents a novel SBR technique called TGIE4Rec. Specifically, TGIE4Rec learns two levels of session embedding, global information enhanced session embedding and transition information enhanced session embedding. The global information enhanced session representation learning layer employs the information of other sessions and the current session to learn global-level session embedding, and the transition information enhanced session representation learning layer employs the items of the current session to learn new session embedding and integrates the time information into the item representation in the session sequence for neighbor embedding learning, so as to further enhance the sequential transition relations in the session sequence. Experiments on three benchmark datasets have demonstrated that TGIE4Rec is superior to other advanced methods.
computer science, theory & methods,engineering, electrical & electronic, hardware & architecture
What problem does this paper attempt to address?