科研之友
微信
新浪微博
Facebook
分享链接
摘要
图神经网络已经成为推荐系统领域的一种主要方法。很多研究把元路径融入到异构图神经网络中，但绝大多数元路径的定义方式只考虑节点之间是否存在连接。而在异构图中同一个节点可能被多条不同类型的边所连接，如用户对物品的浏览、加入购物车、购买等不同交互行为，按照传统的元路径定义方式进行实例化会因为忽略了边的类型而导致学习的节点embedding不准确。针对上述问题，本文提出一种在异构图上把边类型融入到元路径的方法，使节点在每个场景下得到单独训练。然后，使用图注意力机制将不同场景下的同一节点的embedding进行聚合，最终得到该节点的embedding。并用来预测用户与未交互的物品之间的行为关系，从而达到向用户推荐物品的目的。实验表明，本文提出的算法在三个公开数据集上都取得了性能提升，其中，在阿里天池赛数据集上F1、ROC-AUC和PR-RUC指标分别提高了8.75%，6.03%和4.86%。
- 单位
