Attribute Enhanced Graph Attention Networks for Session-based Recommendation

Session-based recommendation (SBR) aims to predict the next item that the anonymous users will interact with based on their short behavior sequence. However, most of the current approaches towards SBR only focus on modeling users’ interest preferences based on a series of items, ignoring extra item attributes, i.e., the brand attribute. This paper proposes a novel approach, called the Attribute Enhanced Graph Attention Networks for Session-based Recommendation (AGSR), using the brand attribute to better model user preferences. Specifically, (i) AGSR constructs the session graph of the current session to capture the user’s session-level item embedding; (ii) AGSR constructs the brand global graph to capture the brand-level item embedding and brand embedding; (iii) Based on the learned node embeddings, AGSR apply attention layers to extract users’ interest and brand preferences. Finally, AGSR makes the recommendation based on item features and user interest and brand preferences, so as to provide better recommendations. Extensive experiments on three real-world datasets demonstrate the effectiveness of our approach.