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

Graph Convolutional Matrix Completion

About

We consider matrix completion for recommender systems from the point of view of link prediction on graphs. Interaction data such as movie ratings can be represented by a bipartite user-item graph with labeled edges denoting observed ratings. Building on recent progress in deep learning on graph-structured data, we propose a graph auto-encoder framework based on differentiable message passing on the bipartite interaction graph. Our model shows competitive performance on standard collaborative filtering benchmarks. In settings where complimentary feature information or structured data such as a social network is available, our framework outperforms recent state-of-the-art methods.

Rianne van den Berg, Thomas N. Kipf, Max Welling• 2017

Related benchmarks

TaskDatasetResultRank
RecommendationGowalla (test)
Recall@200.1395
266
RecommendationGowalla
Recall@200.1395
153
RecommendationAmazon-Book (test)
Recall@200.1316
152
RecommendationMovieLens 1M (test)--
116
RecommendationYelp 2018 (test)
Recall@206.59
110
RecommendationAmazon-Book
Recall@202.88
103
RecommendationYelp (test)
NDCG@202.8
82
RecommendationYelp 2018
Recall@204.62
73
RecommendationMovieLens-100K (test)
RMSE0.905
55
Collaborative FilteringYelp 2018 (test)
Recall@204.62
45
Showing 10 of 47 rows

Other info

Follow for update