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
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Recommendation | Gowalla (test) | Recall@200.1395 | 126 | |
| Recommendation | Amazon-Book (test) | Recall@200.1316 | 101 | |
| Recommendation | Yelp 2018 (test) | Recall@206.59 | 90 | |
| Recommendation | MovieLens-100K (test) | RMSE0.905 | 55 | |
| Collaborative Filtering | Yelp 2018 | NDCG@203.79 | 42 | |
| Collaborative Filtering | Gowalla | NDCG@200.1204 | 40 | |
| Collaborative Filtering | Amazon Books | NDCG@202.24 | 39 | |
| Collaborative Filtering | Amazon-Book (test) | Recall@202.88 | 35 | |
| Collaborative Filtering | Yelp 2018 (test) | Recall@204.62 | 35 | |
| Recommendation | MovieLens 1M (test) | -- | 34 |
Showing 10 of 42 rows