SRM-TGA: A session-based recommendation model supported by temporal graph attention

Dunlu Peng,Li Ji
DOI: https://doi.org/10.2139/ssrn.4560897
IF: 8.139
2024-04-08
Knowledge-Based Systems
Abstract:The objective of session-based recommendation (SBR) is to use the current session to predict the next item an anonymous user will most likely click on. Currently, almost all SBR methods model the transformation relationships between items, while ignoring other item-related information such as item categories and click time, which can more accurately describe the characteristics of items. In addition, sessions are usually modeled as graphs, where information on all neighbors is used to enrich item features, which may lead to prediction bias due to future interest transfer. To address these issues, this paper proposes a session-based recommendation model supported by temporal graph attention (SRM-TGA). The model does not use future items, thereby avoiding interference from future information that is irrelevant to the user's current interests. SRM-TGA uses category and temporal information to construct a static graph with category transformation relationships and a dynamic graph with time spans between items. Besides, graph attention networks and temporal graph attention are applied to learn static embeddings and dynamic embeddings of items, respectively. By fusing the two kinds of embeddings to generate enhanced item embeddings and combining them with positional information, the final session embedding is generated for the prediction. Experimental results on three real datasets, Diginetica, Yoochoose1/4, and Yoochoose1/64, demonstrate that the proposed SRM-TGA model outperforms comparative baseline models in predictive performance.
computer science, artificial intelligence
What problem does this paper attempt to address?