Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Global Context Enhanced Graph Neural Networks for Session-based Recommendation

About

Session-based recommendation (SBR) is a challenging task, which aims at recommending items based on anonymous behavior sequences. Almost all the existing solutions for SBR model user preference only based on the current session without exploiting the other sessions, which may contain both relevant and irrelevant item-transitions to the current session. This paper proposes a novel approach, called Global Context Enhanced Graph Neural Networks (GCE-GNN) to exploit item transitions over all sessions in a more subtle manner for better inferring the user preference of the current session. Specifically, GCE-GNN learns two levels of item embeddings from session graph and global graph, respectively: (i) Session graph, which is to learn the session-level item embedding by modeling pairwise item-transitions within the current session; and (ii) Global graph, which is to learn the global-level item embedding by modeling pairwise item-transitions over all sessions. In GCE-GNN, we propose a novel global-level item representation learning layer, which employs a session-aware attention mechanism to recursively incorporate the neighbors' embeddings of each node on the global graph. We also design a session-level item representation learning layer, which employs a GNN on the session graph to learn session-level item embeddings within the current session. Moreover, GCE-GNN aggregates the learnt item representations in the two levels with a soft attention mechanism. Experiments on three benchmark datasets demonstrate that GCE-GNN outperforms the state-of-the-art methods consistently.

Ziyang Wang, Wei Wei, Gao Cong, Xiao-Li Li, Xian-Ling Mao, Minghui Qiu• 2021

Related benchmarks

TaskDatasetResultRank
Next-item predictionTmall (test)
Hits@100.2833
28
Next-item predictionBooks Industrial dataset (test)
Hits@1077.61
22
Next-item predictionDIGINETICA
Hits@1036.32
14
Next-item predictionDiginetica (test)
Hits@1036.32
14
Next-item predictionElectronics
Hits@1072.93
14
Next-item predictionBeauty (test)
Hits@1089.34
14
Attribute estimationBooks
Hits@1089.78
8
Attribute estimationDIGINETICA
Hits@1089.89
8
Session-based recommendationKKBOX
P@1036.55
8
Attribute estimationTMALL
Hits@1055.88
8
Showing 10 of 14 rows

Other info

Follow for update