May, 2019

STAR-GCN:堆叠和重构的图卷积网络用于推荐系统

TL;DR本文介绍了一种名为 STAR-GCN 的新型图卷积神经网络模型,用于推荐系统中节点表示的学习,尤其是在冷启动场景中。STAR-GCN 采用一堆 GCN 编码器 - 解码器和中间监督来提高最终预测性能。与具有一次性编码节点输入的图卷积矩阵完成模型不同,STAR-GCN 学习低维度的用户和项目潜在因素作为输入,以抑制模型空间复杂性。此外,STAR-GCN 可以通过重构蒙版输入节点嵌入来为新节点产生节点嵌入,从而本质上解决了冷启动问题。此外,我们还发现了训练 GCN 模型进行链接预测任务时的标签泄漏问题,并提出了一种训练策略以避免这个问题。我们在多个评分预测数据集上进行了实证研究,结果表明我们的模型在五个真实世界数据集中的四个中实现了最先进的性能,并且在预测冷启动场景中的评分方面显示出了显著的改进。