Rianne van den Berg, Thomas N. Kipf, Max Welling
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.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Recommendation Systems | MovieLens 100K | RMSE (u1 Splits) | 0.905 | GC-MC |
| Recommendation Systems | MovieLens 100K | RMSE (u1 Splits) | 0.91 | GC-MC |
| Recommendation Systems | MovieLens 1M | RMSE | 0.832 | GC-MC |
| Recommendation Systems | MovieLens 10M | RMSE | 0.777 | GC-MC |
| Recommendation Systems | YahooMusic Monti | RMSE | 20.5 | GC-MC |
| Recommendation Systems | Douban Monti | RMSE | 0.734 | GC-MC |
| Recommendation Systems | Flixster Monti | RMSE | 0.917 | GC-MC |