一种残膜回收机防缠绕挑膜装置的制 一种秧草收获机用电力驱动行走机构

基于多异构图图神经网络的物品推荐方法与流程

2022-02-20 22:27:58 来源:中国专利 TAG:


1.本发明属于数据挖掘信息推荐技术领域,涉及一种物品推荐方法,具体涉及一种基于多异构图图神经网络的物品推荐方法,可应用于短视频推荐、商品推荐等技术领域。


背景技术:

2.随着基于互联网技术的电子商务、社交媒体平台的快速发展,推荐系统已经成为很多企业提升盈利能力的不可或缺的工具。如何从海量的信息中挖掘用户最可能感兴趣的内容,提升物品推荐的准确性成为物品推荐的核心问题。尽管该问题已经存在很多得到广泛应用的解决方法,但这些方法在很大程度上仍依赖于用户-物品交互信息来进行推荐,然而用户-物品交互信息通常是稀疏的(即只有少量用户和物品间存在交互行为),因此如何充分挖掘稀疏的用户-物品交互信息,并利用其它的用户属性信息、物品属性信息等信息辅助推荐是提升物品推荐准确性的关键所在。由于推荐系统中的用户-物品交互信息等许多形式的信息都可以被准确方便地用图的形式进行表示,与图神经网络的特点相契合,因此利用图神经网络进行物品推荐成为了推荐领域一个热门的方向。
3.koren等人在其发表的论文"matrix factorization techniques for recommender systems"(computer,2009,pp.30-37)中提出了一种基于矩阵分解的物品推荐方法mf,该方法是目前应用最广泛的物品推荐方法,该方法首先利用用户-物品交互信息构建用户-物品交互矩阵,然后通过矩阵分解得到用户嵌入向量和物品嵌入向量,从而利用用户嵌入向量和物品嵌入向量的内积进行用户-物品交互概率预测,并最终根据预测结果向用户推荐一组交互概率最高的物品。但是由于该方法仅仅隐式地利用交互信息,获取的用户嵌入向量和物品嵌入向量依然有很大的局限性,得到物品推荐的结果往往并不理想。
4.近几年,由于图神经网络可以通过构造图,利用图结构信息从而显式地使用交互信息,因而被越来越多地应用到推荐领域。he等人在其发表的论文"lightgcn:simplifying and powering graph convolution network for recommendation"(proceedings of the 42nd international acm sigir conference on research and development in information retrieval.2019)中,提出了一种简化和增强图卷积网络的物品推荐方法lightgcn。相比起mf,lightgcn可以对用户-物品交互异构图中的用户、物品节点进行学习,由于包含了图的结构信息及邻居节点的局部相似性信息,最终可以获得更好的用户嵌入向量和物品嵌入向量,得到更准确的物品推荐结果。具体来说,lightgcn首先通过在用户-物品交互异构图上进行多次图卷积操作,不断聚合邻域节点信息来更新用户嵌入向量和物品嵌入向量,然后分别组合所有层学到的用户嵌入向量和物品嵌入向量作为最终的用户嵌入向量和物品嵌入向量,从而进行用户-物品交互概率预测,并最终根据预测结果为用户生成一组交互概率最高的物品。尽管已经获得巨大的成功,但由于仅仅使用稀疏的用户-物品交互信息,忽视用户、物品属性信息以及用户-用户、物品-物品间可能的关系,lightgcn对信息的利用仍不充分,从而导致物品推荐的准确性受到限制。除此之外,lightgcn在更新中心节点时,仅仅使用邻居的聚合作为新的中心节点,完全丢弃了用户或物品节点自身携带的
信息(即放弃了节点自连接),通过使用将各图神经网络层的嵌入向量加权组合(即层组合)来获得等效于节点自连接的效果。但是由于用户-物品交互图进行多次图卷积操作后,节点自身包含了很多之前聚合的信息,实际上已经不局限于节点自身的原始信息,完全放弃节点自连接而改为使用层组合操作并不一定在所有情况下都能获得更好的效果,反而可能会降低物品推荐的准确性。


技术实现要素:

5.本发明的目的在于针对上述现有技术的不足,提出一种基于多异构图图神经网络的物品推荐方法,用于解决现有技术存在的信息利用不充分导致的物品推荐准确性较低的技术问题。
6.为实现上述目的,本发明采取的技术方案包括如下步骤:
7.(1)获取用户信息、物品信息以及用户-物品交互信息:
8.从真实数据集中提取m个用户u={u1,...,um,...,um}的属性信息su={u1,...,um,...,um}、n个物品i={i1,...,in,...,in}的属性信息si={i1,...,in,...,in},以及m个用户与n个物品的d条用户-物品交互信息l={l1,...,ld,...,ld},每个用户um与n个物品或其中的部分物品存在交互关系,其中,m≥500,n≥500,um表示第m个用户,um表示um的包含k1个属性特征的属性信息,k1≥0,in表示第n个物品,in表示in的包含k2个属性特征的属性信息,k2≥0,d<<m
×
n,ld表示第d条用户-物品交互信息;
9.(2)构建属性异构图:
10.(2a)对每个用户um的属性信息um、每个物品in的属性信息in分别进行one-hot编码,得到um的one-hot编码向量和in的one-hot编码向量并为增加m-k1个零元素,同时为增加n-k2个零元素,实现对和的扩展后,得到扩展后的维度为m的one-hot编码向量和的扩展后的维度为n的one-hot编码向量并将m个用户u扩展后的one-hot编码向量作为u的新的属性信息,将n个物品i扩展后的one-hot编码向量作为i的新的属性信息;
11.(2b)构建以m个用户u={u1,...,um,...,um}及其新的属性信息n个物品i={i1,...,in,...,in}及其新的属性信息为节点,以用户u的新的属性信息与物品i的新的属性信息的并集中的每个属性信息为边且形式为矩阵的属性异构图g1,其中,表示大小为m
×
m的,由新的用户属性数据构建的用户属性特征矩阵,表示大小为n
×
n的,由新的物品属性数据
构建的物品属性特征矩阵,表示实数域;
12.(3)构建用户-物品交互异构图:
13.构建以m个用户u={u1,...,um,...,um}与n个物品i={i1,...,in,...,in}分别为节点,以d条用户-物品交互信息l={l1,...,ld,...,ld}为边且形式为矩阵的用户-物品交互异构图g2,其中,表示大小为m
×
n的、第m个用户um与第n个物品in存在交互时阵元r
m,n
=1,不存在交互时阵元r
m,n
=0的用户-物品交互信息矩阵,[
·
]
t
表示转置操作;
[0014]
(4)构建全相邻异构图:
[0015]
构建以m个用户u={u1,...,um,...,um}和n个物品i={i1,...,in,...,in}为节点,以每两个用户的连线以及每两个物品的连线为边的形式为矩阵的全相邻异构图g3,其中,1
(m
×
m)
表示大小为m
×
m的,所有阵元均为1的用户-用户全相邻矩阵,1
(n
×
n)
表示大小为n
×
n的,所有阵元均为1的物品-物品全相邻矩阵;
[0016]
(5)构建基于多异构图图神经网络的物品推荐模型h:
[0017]
构建包含顺次连接的属性辅助层gcn
a1
、图卷积神经网络gcn、全相邻辅助层gcn
a2
和预测层的物品推荐模型h,图卷积神经网络gcn包含顺次连接的k个图卷积层gcn1,...,gcnk,...,gcnk,且属性辅助层gcn
a1
和每个图卷积层gcnk与预测层相连接,其中,k≥2,gcnk表示第k个图卷积层;
[0018]
(6)对基于多异构图图神经网络的物品推荐模型进行迭代训练:
[0019]
(6a)初始化迭代次数为t,最大迭代次数为t,t≥1000,当前物品推荐模型为h
t
,初始化得到第m个用户的用户嵌入向量为第n个物品的物品嵌入向量为并令t=0,h
t
=h,
[0020]
(6b)将属性异构图g1、用户-物品交互异构图g2、全相邻异构图g3、第t次迭代中的用户嵌入向量和物品嵌入向量作为当前物品推荐模型为h
t
的输入进行交互概率预测:
[0021]
(6b1)属性辅助层gcn
a1
对属性异构图g1与第t次迭代中的用户嵌入向量和物品嵌入向量进行不包含节点自连接的图卷积操作,得到用户嵌入向量和物品嵌入向量
[0022][0023]
[0024]
其中,表示在属性异构图g1中用户um拥有的所有用户属性的集合,表示在属性异构图g1中拥有用户属性的所有用户的集合,表示在属性异构图g1中物品in拥有的所有物品属性的集合,表示在属性异构图g1中拥有物品属性的所有物品的集合;
[0025]
(6b2)图卷积神经网络gcn的前k-1个图卷积层gcn1,...,gcnk,...,gcn
k-1
对用户-物品交互异构图g2,以及第t次迭代中的用户嵌入向量和物品嵌入向量进行不包含节点自连接的图卷积操作,得到用户嵌入向量和物品嵌入向量最后一个图卷积层gcnk对用户-物品交互异构图g2,以及第t次迭代中的用户嵌入向量和物品嵌入向量进行包含节点自连接的图卷积操作,得到用户嵌入向量和物品嵌入向量其中第k个图卷积层gcnk得到的用户嵌入向量和物品嵌入向量的表达式分别为:
[0026][0027][0028]
其中,分别表示第t次迭代中第k层图卷积神经网络层得到的用户嵌入向量和物品嵌入向量,入向量和物品嵌入向量,是在用户-物品交互异构图g2中和用户um有交互的所有物品的集合,n
in
是在用户-物品交互异构图g2中和物品in有交互的所有用户的集合;
[0029]
(6b3)全相邻辅助层gcn
a2
对全相邻异构图g3,以及第t次迭代中的用户嵌入向量和物品嵌入向量进行包含节点自连接的图卷积操作,得到用户嵌入向量和物品嵌入向量
[0030][0031][0032]
其中,代表除用户um外的任意用户,代表除物品in外的任意物品,是全相
邻异构图g3中除用户um外的所有用户的集合,是全相邻异构图g3中除用户的所有用户的集合,是全相邻异构图g3中除物品in外的所有物品的集合,是全相邻异构图g3中除物品外的所有物品的集合;
[0033]
(6b4)预测层对第t次迭代中的属性辅助层gcn
a1
输出的用户嵌入向量和物品嵌入向量每一层图卷积神经网络gcnk输出的用户嵌入向量和物品嵌入向量以及全相邻辅助层gcn
a2
输出的用户嵌入向量和物品嵌入向量进行层组合,得到最终用户嵌入向量和最终物品嵌入向量并通过最终嵌入向量的点积得到用户um和物品in的预测交互概率
[0034]
(6c)采用贝叶斯个性化排名损失函数bpr,根据用户-物品预测交互概率计算贝叶斯计算个性化排名损失值l
bpr
,再采用反向传播,通过损失值l
bpr
计算基于多异构图图神经网络的物品推荐模型的参数梯度并使用梯度下降法,通过更新基于多异构图图神经网络的物品推荐模型h的参数θ;
[0035]
(6d)判断t≥t是否成立,若是,得到训练好的物品推荐模型h*,否则,令t=t 1,并执行步骤(6b);
[0036]
(7)获取向用户推荐的物品:
[0037]
选取训练好的物品推荐模型h*最终输出的m
×
n条预测交互概率中每个用户um与n个物品中不存在交互关系的物品的预测交互概率,并按照由大到小的顺序进行排序,然后选取前s个预测交互概率对应的物品作为向用户um推荐的物品,s≥1。
[0038]
本发明与现有技术相比,具有以下优点:
[0039]
第一,本发明所构建的物品推荐模型中包含有属性辅助层、图卷积神经网络、全相邻辅助层和预测层,在对该模型进行训练以及获取向用户推荐的物品的过程中,属性辅助层、图卷积神经网络和全相邻辅助层分别在属性异构图、用户-物品交互异构图和全相邻异构图上进行了图卷积操作,除考虑了用户-物品交互信息外,还考虑了用户、物品属性信息以及用户-用户、物品-物品间可能的关系,避免了现有技术中因为信息利用不充分对推荐精度的影响,有效提高了物品推荐的准确性,能够更好地预测用户对物品的偏好。
[0040]
第二,本发明所构建的物品推荐模型在进行训练时,除了在预测层进行层组合操作外,还在第k个图卷积神经网络层和全相邻辅助层中的图卷积操作中保留了节点自连接操作以利用更多有意义的信息,实际上,在大多数情况下,随着模型层数的增加,除了没有语义特征的用户、物品信息以外,节点自身还包含了很多之前聚合的邻域信息,因此在除了属性辅助层和图卷积神经网络的第一层外的其他层上,保留节点自连接都能一定程度上提升物品推荐的准确性。
附图说明
[0041]
图1是本发明的实现流程图。
具体实施方式:
[0042]
下面结合附图和具体实施例对本发明作进一步的详细描述,应当说明的是,具体实施例仅仅用以解释本发明,并不用于限定本发明。
[0043]
参照图1,本发明包括如下步骤:
[0044]
步骤1)获取用户信息、物品信息以及用户-物品交互信息:
[0045]
从真实数据集flixster中提取m个用户u={u1,...,um,...,um}的属性信息su={u1,...,um,...,um}、n个物品i={i1,...,in,...,in}的属性信息si={i1,...,in,...,in},以及m个用户与n个物品的d条用户-物品交互信息l={l1,...,ld,...,ld},每个用户um与n个物品或其中的部分物品存在交互关系,其中,m≥500,n≥500,um表示第m个用户,um表示um的包含k1个属性特征的属性信息,k1≥0,in表示第n个物品,in表示in的包含k2个属性特征的属性信息,k2≥0,d<<m
×
n,ld表示第d条用户-物品交互信息。在不同数据集中用户的属性特征一般包括年龄、性别、籍贯、职业等;在不同数据集中物品可以是电影、音乐、商品等物品,它们的属性特征一般包括名称、类型、作者等;本实例中,通过挖掘用户对电影的交互信息和属性信息,本发明向用户推荐一组电影,其中m=3000,n=3000,d=26173,用户的属性特征个数k1=3,为年龄、性别、职业,物品的属性特征个数k2=3,为名称、类型、导演。
[0046]
步骤2)构建属性异构图:
[0047]
步骤2a)为了将用户、属性信息构建为图结构从而显式地使用属性信息,对每个用户um的属性信息um、每个物品in的属性信息in分别进行one-hot编码,得到um的one-hot编码向量和in的one-hot编码向量由于k1<<m,k2<<n,为了使用一组模型参数训练本发明中的各个层,为增加m-k1个零元素,同时为增加n-k2个零元素,实现对和的扩展后,得到扩展后的维度为m的one-hot编码向量和的扩展后的维度为n的one-hot编码向量并将m个用户u扩展后的one-hot编码向量作为u的新的属性信息,将n个物品i扩展后的one-hot编码向量作为i的新的属性信息。
[0048]
步骤2b)构建以m个用户u={u1,...,um,...,um}及其新的属性信息n个物品i={i1,...,in,...,in}及其新的属性信息为节点,以用户u的新的属性信息与物品i的新的属性信息的并集中的每个属性信息为边且形式为矩阵的属性异构图g1,其中,表示大小为m
×
m的,由新的用户属性数据构建的用户属性特征矩阵,表示大小为n
×
n的,由新的物品属性数据
构建的物品属性特征矩阵,表示实数域。
[0049]
步骤3)构建用户-物品交互异构图:
[0050]
构建以m个用户u={u1,...,um,...,um}与n个物品i={i1,...,in,...,in}分别为节点,以d条用户-物品交互信息l={l1,...,ld,...,ld}为边且形式为矩阵的用户-物品交互异构图g2,其中,表示大小为m
×
n的、第m个用户um与第n个物品in存在交互时阵元r
m,n
=1,不存在交互时阵元r
m,n
=0的用户-物品交互信息矩阵,[
·
]
t
表示转置操作。
[0051]
步骤4)构建全相邻异构图:
[0052]
构建以m个用户u={u1,...,um,...,um}和n个物品i={i1,...,in,...,in}为节点,以每两个用户的连线以及每两个物品的连线为边的形式为矩阵的全相邻异构图g3,其中,1
(m
×
m)
表示大小为m
×
m的,所有阵元均为1的用户-用户全相邻矩阵,1
(n
×
n)
表示大小为n
×
n的,所有阵元均为1的物品-物品全相邻矩阵。
[0053]
步骤5)构建基于多异构图图神经网络的物品推荐模型h:
[0054]
构建包含顺次连接的属性辅助层gcn
a1
、图卷积神经网络gcn、全相邻辅助层gcn
a2
和预测层的物品推荐模型h,图卷积神经网络gcn包含顺次连接的k个图卷积层gcn1,...,gcnk,...,gcnk,且属性辅助层gcn
a1
和每个图卷积层gcnk与预测层相连接,其中,k≥2,gcnk表示第k个图卷积层;本实例中,k=3。
[0055]
步骤6)对基于多异构图图神经网络的物品推荐模型进行迭代训练:
[0056]
步骤6a)初始化迭代次数为t,最大迭代次数为t,t≥1000,当前物品推荐模型为h
t
,h
t
的第m个用户的用户嵌入向量为第n个物品的物品嵌入向量为并令t=0,h
t
=h,本实例中,t=2000,用户嵌入向量物品嵌入向量维度大小均固定为64;
[0057]
步骤6b)将属性异构图g1、用户-物品交互异构图g2、全相邻异构图g3、第t次迭代中的用户嵌入向量和物品嵌入向量作为当前物品推荐模型为h
t
的输入进行交互概率预测:
[0058]
步骤6b1)属性辅助层gcn
a1
对属性异构图g1与第t次迭代中的用户嵌入向量和物品嵌入向量进行不包含节点自连接的图卷积操作,得到用户属性嵌入向量和物品属性嵌入向量
[0059]
[0060][0061]
其中,表示在属性异构图g1中用户um拥有的所有用户属性的集合,表示在属性异构图g1中拥有用户属性的所有用户的集合,表示在属性异构图g1中物品in拥有的所有物品属性的集合,表示在属性异构图g1中拥有物品属性的所有物品的集合;
[0062]
为了更加充分的挖掘利用有限的数据信息,所述属性辅助层gcn
a1
将利用用户、物品属性信息作为辅助信息构造的属性异构图加入模型的训练中,进行不包含节点自连接的图卷积操作更新用户嵌入向量和物品嵌入向量。不同于传统属性嵌入向量获取方法仅仅利用全连接网络隐式地使用属性信息获得用户嵌入向量和物品嵌入向量,本发明通过将属性信息构造成图,利用图结构信息显式地使用属性信息,不仅可以获得节点本身的属性,还可以在属性异构图上通过邻域聚合获得邻居节点的属性信息,能够有效提升物品推荐的准确性。
[0063]
步骤6b2)图卷积神经网络gcn的前k-1个图卷积层gcn1,...,gcnk,...,gcn
k-1
对用户-物品交互异构图g2,以及第t次迭代中的用户属性嵌入向量和物品属性嵌入向量进行不包含节点自连接的图卷积操作,得到用户属性嵌入向量和物品属性嵌入向量最后一个图卷积层gcnk对用户-物品交互异构图g2,以及第t次迭代中的用户属性嵌入向量和物品属性嵌入向量进行包含节点自连接的图卷积操作,得到用户属性嵌入向量和物品属性嵌入向量其中第k个图卷积层gcnk得到的用户属性嵌入向量和物品属性嵌入向量的表达式分别为:
[0064][0065][0066]
其中,分别表示第t次迭代中第k层图卷积神经网络层得到的用户嵌入向量和物品嵌入向量,入向量和物品嵌入向量,是在用户-物品交互异构图g2中和用户um有交互的所有物品的集合,是在用户-物品交互异构图g2中和物品in有交互的所有用户的集合;
[0067]
所述图卷积神经网络gcn为了充分挖掘用户-物品交互信息,在最后一个图卷积层gcnk中对利用用户-物品交互信息构造的用户-物品交互异构图g2进行保留节点自连接的图卷积操作。一般认为,在用户-物品交互异构图中,用户节点和物品节点仅仅包含没有语义特征的信息,然而,随着图卷积层数的增加,节点自身还包含了很多之前聚合的邻域信息,因此在最后一层保留自连接图卷积操作,可以通过提高邻域信息的权重来获得更好的用户嵌入向量和物品嵌入向量,从而提升物品推荐的准确性。实际上,根据每个真实数据集的具体情况,除了最后一层,在其它部分层保留节点自连接也可以更进一步提升物品推荐的准确性。
[0068]
步骤6b3)全相邻辅助层gcn
a2
对全相邻异构图g3,以及第t次迭代中的用户属性嵌入向量和物品属性嵌入向量进行包含节点自连接的图卷积操作,得到用户嵌入向量和物品嵌入向量
[0069][0070][0071]
其中,代表除用户um外的任意用户,代表除物品in外的任意物品,是全相邻异构图g3中除用户um外的所有用户的集合,是全相邻异构图g3中除用户的所有用户的集合,是全相邻异构图g3中除物品in外的所有物品的集合,是全相邻异构图g3中除物品外的所有物品的集合;
[0072]
所述全相邻辅助层gcn
a2
为了进一步挖掘推荐数据信息,对利用用户关系和物品关系作为辅助信息构造的全相邻异构图g3进行包含节点自连接的图卷积操作。该方法能充分利用了用户-用户、物品-物品间可能的关系,扩宽了节点间信息传递的通道,客观上使得远方的节点对目标节点拥有一定的影响力,使得节点能够获得更多远程信息,从而提高物品推荐的准确性。实际上,除了本发明所采取的仅在最后一个图卷积数据网络层后增加全相邻辅助层gcn
a2
外,在每一层图卷积神经网络后都增加全相邻辅助层gcn
a2
在多数情况下也能够进一步提升物品推荐的准确性,但这会增加算法的时间复杂度。
[0073]
步骤6b4)预测层对第t次迭代中的属性辅助层gcn
a1
输出的用户属性嵌入向量和物品属性嵌入向量每一层图卷积神经网络gcnk输出的用户嵌入向量和物品嵌入向量以及全相邻辅助层gcn
a2
输出的用户嵌入向量和物品嵌入向量进行层组合,得到最终用户嵌入向量和最终物品嵌入向量并通过最终嵌入向量的点积得到用户um和物品in的预测交互概率
[0074]
所述的组合节点属性辅助层gcn
a1
、图卷积神经网络gcn中每一层和全相邻辅助层gcn
a2
输出的用户嵌入向量和物品嵌入向量获得最终用户嵌入向量和物品嵌入向量
的具体过程可表达为:
[0075][0076][0077]
其中,a
a1
、a
a2
和ak代表辅助层gcn
a1
、辅助层gcn
a2
和第k层对于构成最终嵌入向量的重要性,它既可以作为一个被手动调整超参数,又可以作为一个模型参数被优化;
[0078]
所述的通过最终嵌入向量的点积得到用户um和物品in的预测交互概率的具体过程可表达为:
[0079][0080]
其中,(
·
)
t
表示向量的转置;
[0081]
步骤6c)采用贝叶斯个性化排名损失函数bpr,根据用户-物品预测交互概率计算贝叶斯计算个性化排名损失值l
bpr
,再采用反向传播,通过损失值l
bpr
计算基于多异构图图神经网络的物品推荐模型的参数梯度并使用梯度下降法,通过更新基于多异构图图神经网络的物品推荐模型h的参数θ;其中,贝叶斯个性化排名损失函数bpr的公式以及可训练参数的更新公式分别为:
[0082][0083][0084]
其中,表示与同一用户um有交互信息的物品和没有交互信息的物品构成的对比训练样本,r

表示有交互信息的用户-物品对集合,r-表示没有交互信息的用户-物品对集合,θ={θu;θi}表示模型可训练的用户嵌入向量权值参数和物品嵌入权值向量参数;β为l2正则化系数,θ'为更新后模型的权值参数,θ为更新前模型的权值参数,λ为学习率,为模型的参数梯度;
[0085]
步骤6d)判断t≥t是否成立,若是,得到训练好的物品推荐模型h*,否则,令t=t 1,并执行步骤(6b);
[0086]
步骤7)获取向用户推荐的物品:
[0087]
选取训练好的物品推荐模型h*最终输出的m
×
n条预测交互概率中每个用户um与n个物品中不存在交互关系的物品的预测交互概率,并按照由大到小的顺序进行排序,然后选取前s个预测交互概率对应的物品作为向用户um推荐的物品,s≥1;本实例中,s=20。
[0088]
下面结合仿真实验对本发明的技术效果作进一步说明:
[0089]
1.仿真条件与内容:
[0090]
仿真实验采用的处理器为intel(r)core(tm)i7-10875h cpu,主频为2.30ghz,内
存为16g,显卡为geforce gtx 2060,操作系统为windows 10企业版64位操作系统的平台上使用结合torch 1.7.1的python 3.8进行。
[0091]
仿真实验使用如表1所示的四个来自真实世界的推荐系统领域数据集:从flixster网站采集的记录用户对电影交互信息的flixster数据集;从豆瓣网站采集的记录用户对电影交互信息的douban数据集;从雅虎网站采集的记录用户对音乐交互信息的yahoomusic数据集以及从last.fm采集的记录用户对音乐交互信息的last.fm数据集。
[0092]
表1
[0093][0094]
对本发明和现有的基于矩阵分解的物品推荐方法mf和简化和增强图卷积网络的物品推荐方法lightgcn的召回率(recall)和归一化折损累计增益(ndcg)进行对比仿真,其结果如表2所示。
[0095]
2.仿真结果分析:
[0096]
为了更好地评价本发明的技术效果,分别计算本发明和两个现有技术在四个数据集上得到的推荐物品与其对应的真实推荐物品的召回率(recall)和归一化折损累计增益(ndcg),设置模型为用户生成可能交互物品列表的长度为20,召回率(recall)可以衡量用户的命中率,描述有多少比例的用户-物品交互记录包含在最终为用户生成的可能交互物品列表中,命中率越高,其值越靠近1,反之越靠近0,该指标越靠近1,说明物品推荐的准确性越高。归一化折损累计增益(ndcg)可以衡量用户的测试物品在为用户生成的可能交互物品列表中的位置,位置越靠前,其值越接近于1,反之越靠近0,该指标越靠近1,说明物品推荐的准确性越高。
[0097]
表2
[0098][0099]
从表2可以看出,本发明相比于两个现有技术在四个数据集上的召回率(recall)和归一化折损累计增益(ndcg)均有不同程度的提升,证明本发明具有较高的准确性。
[0100]
以上仿真实验表明:本发明提出的基于多异构图图神经网络的物品推荐方法首先通过在多个异构图上构造两个辅助层和多个图卷积神经网络层以获得更好的嵌入向量,从而更高效地挖掘用户、物品属性信息和用户-物品交互信息;其次通过保留部分层的节点自连接,充分利用有限的数据,保留了更多聚合到节点上的有意义的信息,提高了物品推荐的准确性。
再多了解一些

本文用于企业家、创业者技术爱好者查询,结果仅供参考。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表

相关文献