Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

DGRec: Graph Neural Network for Recommendation with Diversified Embedding Generation

About

Graph Neural Network (GNN) based recommender systems have been attracting more and more attention in recent years due to their excellent performance in accuracy. Representing user-item interactions as a bipartite graph, a GNN model generates user and item representations by aggregating embeddings of their neighbors. However, such an aggregation procedure often accumulates information purely based on the graph structure, overlooking the redundancy of the aggregated neighbors and resulting in poor diversity of the recommended list. In this paper, we propose diversifying GNN-based recommender systems by directly improving the embedding generation procedure. Particularly, we utilize the following three modules: submodular neighbor selection to find a subset of diverse neighbors to aggregate for each GNN node, layer attention to assign attention weights for each layer, and loss reweighting to focus on the learning of items belonging to long-tail categories. Blending the three modules into GNN, we present DGRec(Diversified GNN-based Recommender System) for diversified recommendation. Experiments on real-world datasets demonstrate that the proposed method can achieve the best diversity while keeping the accuracy comparable to state-of-the-art GNN-based recommender systems.

Liangwei Yang, Shengjie Wang, Yunzhe Tao, Jiankai Sun, Xiaolong Liu, Philip S. Yu, Taiqing Wang• 2022

Related benchmarks

TaskDatasetResultRank
Sequential RecommendationML 1M
NDCG@100.1976
140
Sequential RecommendationSports
HR@106.43
61
Sequential RecommendationLastFM
NDCG@104.95
30
Sequential RecommendationBeauty
HR@1010.85
29
Multi-objective Re-rankingBeauty
Hit Rate @ 536.78
13
Multi-objective Re-rankingGrocery
HR@537.47
13
Multi-objective Re-rankingML 1M
HR@548.44
13
Showing 7 of 7 rows

Other info

Follow for update