前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >图神经网络——【KDD 2019】KGAT

图神经网络——【KDD 2019】KGAT

作者头像
小爷毛毛_卓寿杰
发布2021-09-10 11:34:57
5150
发布2021-09-10 11:34:57
举报
文章被收录于专栏:Soul Joy HubSoul Joy Hub

深度推荐算法(如DeepFM等)模型有个缺点:将训练数据里(用户交互数据)的特征进行独立建模,没有考虑到交互数据之间的关系。这使得这些模型不足以从用户的行为中提取出基于属性的协同信息。

在这里插入图片描述
在这里插入图片描述

如上图:用户u1 看了电影i1 ,这个电影是 ei 导演的,传统的CF方法会着重去找那些也看了电影i1 的用户,比如u4、u5 。而监督学习方法会重点关注那些有相同属性ei 的电影,比如 i2。很显然这两类信息都可以作为推荐信息的补充,但是现有的模型不能做到上面两者信息的融合,而且这里的高阶关系也可以作为推荐信息的补充的。比如图中黄色框图里的用户看了同样由 ei 导演的电影 i2 , 还有灰色框图里电影有e2的参演。

为了解决上面提到的问题,本文提出KGAT方法,将图谱关系信息及用户user点击商品item的交互图融合到一个图空间里,可以融合CF信息及KG信息,同时发现高阶的关系信息:

在这里插入图片描述
在这里插入图片描述

CKG Embedding Layer

KGAT采用了TransR方法,对于三元组( h , r , t ) 的似然得分如下:

在这里插入图片描述
在这里插入图片描述

训练成对损失函数如下:

在这里插入图片描述
在这里插入图片描述

Attentive Embedding Propagation Layers

作者认为每个邻居节点应该贡献不同的权值,因此本文采用的了attention的方式来求得每一个邻居节点的重要性:

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

作者实验选择效果最好的聚合方式:Bi-Interaction(相加与点乘)相加得到最终表示:

在这里插入图片描述
在这里插入图片描述

多层传播:

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

Prediction Layer

预测层将Attentive Embedding Propagation Layers层得到的多层用户表示 及多层item表示进行相乘得到相关性得分:

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

Optimization

观察到的交互表示出用户的偏好,那么相应就应该有更高的预测评分:

在这里插入图片描述
在这里插入图片描述

最终与CKG Embedding Layer的损失函数结合,得到最终的损失函数:

在这里插入图片描述
在这里插入图片描述
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2021-05-17 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • CKG Embedding Layer
  • Attentive Embedding Propagation Layers
  • Prediction Layer
  • Optimization
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档