SOTAVerified

Graph Convolutional Matrix Completion

2017-06-07Code Available1· sign in to hype

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

Code Available — Be the first to reproduce this paper.

Reproduce

Code

Abstract

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.

Tasks

Benchmark Results

DatasetModelMetricClaimedVerifiedStatus
Douban MontiGC-MCRMSE0.73Unverified
Flixster MontiGC-MCRMSE0.92Unverified
MovieLens 100KGC-MCRMSE (u1 Splits)0.91Unverified
MovieLens 100KGC-MCRMSE (u1 Splits)0.91Unverified
MovieLens 10MGC-MCRMSE0.78Unverified
MovieLens 1MGC-MCRMSE0.83Unverified
YahooMusic MontiGC-MCRMSE20.5Unverified

Reproductions