论文解析:Graph Convolutional Matrix Completion

论文解析:Graph Convolutional Matrix Completion

  • 作者:Rianne van den Berg, Thomas N. Kipf, Max Welling(也是发表GCN的团队)
  • 会议:KDD 2018

方法

图变分自编码器(GAE)的应用(GAE的博客),算法核心就是下面的图
在这里插入图片描述
将推荐系统的user-item交互建模为二分图,user和item分别是二分图的左半部和右半部节点,user对item的评分是二分图的边权,通过GCN得到user和item的embedding,然后通过一个双线性softmax预测user-item得分的概率分布(每个得分对应一个概率)。loss采用交叉熵损失。
user-item评分矩阵和user-item二分图的等价关系如下图所示:
在这里插入图片描述

实验

实验分别在MovieLens(100K/1M/10M), Flixster, Douban和YahooMusic四组数据集上。由于这些数据集都是有评分的数据集,因此采用RMSE作为评价指标。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章