面向知识超图链接预测的生成对抗负采样方法

郭正山 左 劼 段 磊 李仁昊 何承鑫 肖英劼 王培妍

(四川大学计算机学院 成都 610065)

知识图谱是一种以图结构存储真实世界中事实的知识库，其基本构成单位是一个包含头尾实体与描述两者间关系的三元组.然而，现实世界中的多元关系事实通常无法以三元组形式进行表达，如Freebase[1]知识库中多元关系的占比超过61%[2].另一方面，将多元关系事实拆分为多个三元组会导致部分事实的信息丢失.例如，图1左部分中的事实“刘翔2004年在雅典奥运会的110米栏项目上获得金牌”，“人物-时间-赛事-项目-奖项”是一个涉及“刘翔”“2004年”“雅典奥运会”“110米栏”和“金牌”的五元关系，无法在保持事实完整性的情况下进行拆分.针对此问题，知识超图对知识图谱的事实表达形式进行了推广，以包含多个实体及多元关系的多元组为基本单位，增强了对多元关系事实的表达能力，在信息检索[3]、推荐系统[4-5]、自然语言处理[6]等场景中均发挥重要作用.

Fig. 1 An example of negative sampling in knowledge hypergraph图1 知识超图负采样的简单示例

由于现实世界中新知识不断产生及知识获取困难等问题，现有知识超图不可避免地面临不完整性问题.例如，在Freebase知识库中，99%的“人物”缺少“种族”信息[7]；在DBpedia[8]知识库中，58%的“学者”缺少“研究领域”的信息[9].因此，面向知识超图补全的链接预测任务已成为该领域的研究热点.其通过实体间现有的链接关系预测未知链接，以能自动学习和推理知识超图为目标，是完成诸多下游应用的重要前置任务.

针对知识超图链接预测任务，现有工作主要将实体和关系编码到低维向量空间中以捕获两者之间的联系，通过评分函数分别对正、负样本进行合理性打分，再结合样本真实标签计算预测损失以优化模型参数.在上述过程中，事实与非事实分别作为正、负样本为链接预测模型提供训练梯度.然而，现有知识超图出于对空间有效性的考虑只存储事实[10]，这使得如何构建代表非事实的负样本成为关键.目前，均匀随机采样方法[11]是链接预测任务中广泛使用的一种负采样方法，其通过将正样本中的任一实体以相同概率随机替换为知识超图中其他实体以生成负样本.然而，该方法大概率会采样到不属于同一实体类型或语义不相关的实体作为替换，生成的低质量负样本可能会使模型在训练阶段面临零损失(zero loss)问题[12].

例1.对图1左图中的知识超图多元组P3使用均匀随机负采样，生成图1右图中的6个候选负样本N1，N2，…，N6.其中N1，N2和N3因采样到与原实体语义不相关的替换实体，在映射至嵌入空间中时与原事实P3距离较远，是无法在链接预测训练阶段有效贡献梯度的低质量负样本.反之，N4，N5和N6是有助于链接预测模型训练的高质量负样本.

为解决上述问题，研究者提出了针对知识图谱链接预测的负采样方法，包括基于缓存和基于生成对抗网络的方法.对于前者，NSCaching[13]通过AutoML[14]技术动态更新缓存中的高质量负样本，然而该模型需针对元组中的不同实体位置分别设定缓存，具有较高的时空开销，因此不适合扩展至知识超图.对于后者，一系列工作[10,12,15]将生成对抗网络用于知识库链接预测任务.具体地说，网络中的生成器倾向于生成使判别器难以判断真伪的样本，而判别器在模型迭代过程中不断提高判别能力.在引入到知识库链接预测任务中后，该结构适用于生成可以解决零损失问题的高质量负样本.然而，这些方法均是针对知识图谱场景设计的，迁移至知识超图场景时，会面临2种挑战：

挑战1.现有负采样方法均针对具有固定实体数目的三元组进行采样，而在知识超图场景下，多元组中实体数目会变化，如：四元组、五元组等.因此无法直接替换其中的链接预测方法以适用于知识超图.

挑战2.现有基于生成对抗网络的负采样方法均需加入预训练步骤以提高生成对抗网络的稳定性.然而在知识超图场景下，多元组中实体数量的增加会进一步提高预训练步骤的复杂度.

为应对2种挑战，本文提出一个面向知识超图链接预测的生成对抗负采样方法HyperGAN，通过引入生成对抗网络提高负样本质量，以对链接预测模型提供指导.具体地，针对挑战1，对以往负采样方法中共享的链接预测与判别器模块进行了分离.解耦后的链接预测模块可适用于知识超图的链接预测方法，而不影响判别器模块进行样本质量判别.针对挑战2，简化了生成器与判别器的结构，因此无需像以往负采样方法一样进行预训练，在适用于知识超图链接预测任务的同时提高了对抗训练的稳定性.在多个知识超图数据集上的实验表明，HyperGAN能生成高质量负样本并提升链接预测模型的准确度.

本文工作的主要贡献有3个方面：

1) 提出了一个面向知识超图链接预测的生成对抗负采样方法HyperGAN.其通过提高负样本质量改善链接预测任务的效果.

2) HyperGAN通过解耦预测模型与判别器提升了方法通用性，简化后的生成对抗网络结构无需预训练也可进行稳定的对抗训练，降低了时间开销.

3) 在5个公开的真实知识超图数据集上进行了多组对比实验，相比其他先进方法，HyperGAN方法在性能与效率方面均取得提升.针对负样本的具体案例分析与定量分析进一步验证了HyperGAN方法的有效性.

1 相关工作

本节主要介绍知识超图链接预测方法和针对知识库链接预测领域的负采样方法的相关工作.

1.1 知识超图链接预测方法

当前针对知识超图的链接预测研究工作可分为3类：基于平移的方法、基于神经网络的方法和基于张量分解的方法[16].

1) 基于平移的方法.TransE[11]是首次提出基于平移的知识图谱链接预测方法，其通过将知识图谱三元组中的关系建模为从头实体到尾实体的平移变换，从而学习实体和关系之间的联系.然而，该方法在关系建模上的限制使其不具有完全表达性[17].此外，该方法只适用于知识图谱，不能应用到知识超图.为解决此问题，Wen等人[18]提出的m-TransH方法通过将实体映射到知识超图多元关系超平面，用基于平移的方法解决了知识超图链接预测问题.RAE方法[19]在此基础上考虑到2个实体同时出现在一个多元关系中的重要性，其将该情况下的相关性概率引入到损失函数中并使用一个全连接神经网络对模型进行训练.然而，上述扩展自TransE的m-TransH及RAE方法均不可避免地继承了其不具有完全表达性的缺点.最近，Abboud等人[7]提出的BoxE方法解决了这一问题，该方法将多元关系视为空间中的一组框(boxes)，实体视为空间中的点，通过计算实体点到对应关系框的距离，来对知识超图中的事实进行评分.

2) 基于神经网络的方法.由于神经网络在各类自然语言处理任务上的成功应用，部分研究工作将其引入到知识超图链接预测中，并取得了较好的表现.NaLP[20]方法将知识超图中多元关系事实表示为“角色-实体”对的形式，使用卷积神经网络(con-volutional neural network, CNN)和全连接网络(fully connected network, FCN)来衡量所有角色与实体对之间的相关性，从而对事实成立的合理性进行评分.然而，NaLP并未考虑不同“角色-实体”对间的差异.为改进此不足，HINGE[21]与NeuInfer[22]方法将多元关系事实表示为主三元组与辅助“角色-实体”对的形式，并分别通过CNN与FCN来计算事实的合理性得分.Galkin等人[23]提出的StarE方法将信息传递网络应用到知识超图链接预测问题上，通过迭代更新实体和关系的嵌入表示并将其送入Transformer[24]中，进而对事实合理性进行评估.一般来说，上述基于神经网络的链接预测方法需要经过额外数据处理步骤确定主三元组或“角色-实体”对中的实体和关系，同时也会面临参数过多导致的训练困难和不易推广到大规模知识超图等问题.

3) 基于张量分解的方法.最初，许多基于张量分解的知识超图链接预测方法是对知识图谱链接预测方法的推广，例如n-CP[2],n-TuckER[25]和m-DistMult[26]方法分别是对CP[27],TuckER[28]和DistMult[2]方法的拓展，但是这些方法不但没有改善原始方法的不足，甚至在知识超图链接预测任务中对缺点进行了放大，于是研究者开始针对知识超图场景设计特定的链接预测方法.Fatemi等人[2]提出的HypE模型考虑了实体在多元组中不同位置的信息差异，其基于位置信息学习实体的嵌入表示并进一步计算事实成立的合理性得分.Liu等人[25]提出的GETD模型通过使用TuckER方法[28]将知识超图中事实的高阶张量表示分解为一个核张量加若干个因子矩阵，并使用分解得到的核张量与关系和实体的嵌入表示相乘以获得多元关系事实成立的合理性得分.此外，针对核张量具有较多参数的问题，该模型利用张量环式(tensor ring, TR)分解方法[29]将其分解为多个三阶张量以降低模型复杂度.虽然GETD具有完全表达性，但其不能同时处理具有不同元数的事实.因此，当数据集中同时包含不同元数的事实时，必须将事实以元数进行划分，再分别进行模型训练.最近，Yao等人[30]提出的S2S模型，通过将实体和关系的嵌入表示划分为多个块，允许嵌入表示从具有混合元数的事实中学习.此外，该方法采用神经网络架构搜索技术，稀疏化核张量的参数以降低模型复杂度.

综上所述，基于张量分解的链接预测方法能有效捕获实体和关系间的潜在交互，取得了当前最好的预测准确度，并且大多数有严格数学理论支撑其完全表达性，因而在知识超图链接预测任务上发挥着越来越大的作用.

1.2 负采样方法

1.1节的知识超图链接预测模型一般通过构造事实评分函数，根据输入样本与真实标签的差距计算损失以优化模型参数.其中，高质量负样本应对训练梯度有所贡献以使模型具有更好的泛化能力，因此一些研究人员将不同负采样方法应用于知识库链接预测任务，以获得高质量的负样本.

最初，大多数模型使用TransE[11]在知识图谱上提出的均匀随机采样(uniformly random sampling)方法，其将正样本中的头实体或尾实体以相同概率随机替换为实体集中的其他实体，以生成对应的负样本.该方法因简单有效的特性，被后续应用到知识超图的链接预测任务中.然而，均匀随机采样方法的采样空间过大，这使得随机采样到高质量负样本的概率过小，对提升模型泛化能力的贡献不大[10].另一方面，该方法还可能采样到错误的负样本(即未出现在数据集中的正样本)，不利于模型训练[15].针对该问题，Wang等人[31]在TransH中提出一种基于伯努利采样(Bernoulli sampling)的方法，通过统计知识库中关系的一对多、多对一、多对多映射属性，使模型以不同的概率替换头实体和尾实体，从而减少了生成错误负样本的概率，但该方法也是从固定分布中采样，所以不能随着模型的训练动态改变，造成大多数生成的负样本可信度得分很小，对基于梯度的优化算法几乎没有贡献.因此，上述2种方法都难以生成高质量的负样本，这可能导致模型训练时出现零损失问题[12]，即在训练一段时间后，生成的大部分负样本对模型损失函数为零，导致梯度消失，对模型训练没有贡献.

Zhang等人[13]提出针对知识图谱链接预测的负采样模型NSCaching，其通过设定缓存(cache)来直接存储负样本，并在训练过程中对负样本进行动态更新，以使其中包含更多高质量的负样本.然而，该模型需针对元组中不同实体位置设定相应缓存，且在每次模型训练后，需及时更新所有缓存中的负样本，具有较高的时空开销，因而不适合应用于知识超图.Ahrabian等人[32]考虑到知识图谱的空间结构，提出了SANS模型.该模型假设头、尾实体的k跳邻居实体与其有更高的相关性，于是将这些相关性高的实体作为事实中负采样的候选集，进而从中采样生成高质量的负样本.然而，该方法亦不易于拓展应用至空间结构复杂的知识超图.RotatE[33]采用自对抗负采样方法进行知识图谱链接预测，其使用模型评分函数获得负样本的概率分布，然后将概率作为负样本的权重引入到链接预测模型的损失函数中.该方法通过模型本身衡量负样本的质量，无额外开销且简单有效，因而可灵活应用于知识超图.

近年来受生成对抗网络(generative adversarial network, GAN)的启发，一些工作尝试使用对抗式训练框架生成高质量的负样本，以解决对模型训练的零损失问题.Cai等人[10]提出的KBGAN模型采用2个不同的知识图谱链接预测模型分别作为对抗网络的生成器与判别器，其中生成器负责负样本生成，而判别器负责对输入的样本进行正负判别.随着模型的迭代，负样本生成器与样本判别器通过对抗式训练同时提高性能.测试阶段则采用判别器中的嵌入表示对样本进行评估，以获得良好的链接预测效果.KSGAN[15]通过在KBGAN模型上增加一个选择器，筛选生成器生成的负样本，进一步提升了模型性能.Wang等人[12]提出的IGAN模型通过采用双层全连接神经网络作为生成器，为正样本生成对应的高质量负样本，辅助基于平移的链接预测模型进行训练，缓解模型零损失问题的同时也降低了生成器的复杂性.尽管上述基于GAN的负采样方法取得了较好的表现，但为增强稳定性并缓解退化问题，必须进行预训练，这带来了额外的训练成本[13].

与1.2节提到的负采样方法相比，本文工作主要致力于解决知识超图链接预测中的负采样问题.由于知识超图结构复杂、采样空间大，目前工作大多采用均匀随机采样的方式生成负样本，还未有简单通用的有效方法.本文将生成对抗网络引入到知识超图负采样中，提出的HyperGAN方法能够应用于不同关系元数及多种关系类型的知识超图，通过采用GAN的输出得分作为衡量负样本质量的标准，过滤生成高质量负样本以提升链接预测模型的性能.此外，HyperGAN的简单结构使得其不需要进行额外预训练，节约训练成本.

2 预备知识

本节给出知识超图链接预测的一些基本概念.

定义1.知识超图.一个知识超图可以表示为B=(E,R,F)，其中E为实体(节点)集，R为多元关系集，F为知识超图中的事实集合.对于每个多元关系r∈R，其元数|r|为关系r涉及到的实体数量.

在知识图谱中，一个事实通常被表示为三元组(eh,r,et)的形式，其中eh,et∈E分别表示头实体与尾实体，r∈R表示头尾实体间的关系.类似地，知识超图中的事实f∈F可以被表示为多元组(r,e1,e2,…,en)的形式，其中r∈R，ei∈E表示关系中第i个实体，该事实f也被称为n元关系事实.

这里用Fall表示世界上全部事实的集合，所有在Fall中的多元组(r,e1,e2,…,en)为正样本，否则为负样本.由于知识库高度不完整，并不能存储所有事实[25]，所以当前知识超图仅包含一部分事实F⊂Fall.

定义2.知识超图链接预测.给定一个候选多元组f∉F，链接预测任务的目的就是判断该多元组f是否属于F中缺少的事实，即f是否属于FallF.

3 HyperGAN方法

本文提出的HyperGAN由3个模块组成：1)知识超图链接预测模块.将负样本置信度引入损失函数以改善模型性能，并将训练好的嵌入表示共享至其余2个模块.2)生成器模块.生成高质量负样本与判别器进行对抗训练.3)判别器模块.在链接预测阶段，为链接预测模块提供负样本置信度；在生成对抗负采样阶段，通过判别正样本与生成器提供的负样本提高自身辨别能力.图2为HyperGAN的整体架构.

Fig. 2 The architecture of HyperGAN图2 HyperGAN架构图

3.1 知识超图链接预测模块

在知识超图链接预测模块中，为了学习到多元关系事实中实体和关系的嵌入表示，本文采用常用的对数损失函数[25,33]优化链接预测模块：

(1)

其中，Fbatch表示训练过程中每次迭代使用的小批量正样本集，f表示其中的一个正样本多元组(r,e1,…,en)，φ表示链接预测模块中的事实合理性评分函数，N(f)表示基于正样本f均匀随机采样生成的负样本集合，f′∈N(f)表示其中的一个负样本.具体地，对于f中的任意一个实体ei，用实体集E中的其他实体替换该实体以生成不属于F的负样本：

(2)

(3)

其中，φD表示判别器模块的评分函数.为平衡不同负样本之间的影响，将上述负样本权重融入式(1)所述链接预测模块，最终损失函数表示为

(4)

3.2 生成器模块

生成对抗网络中生成器的目标是生成能在对抗生成阶段高效贡献梯度的负样本，即高质量负样本.具体地，HyperGAN中生成器使用链接预测模块训练好的实体和关系的嵌入表示，将正样本集Fbatch作为输入，经过实体替换操作，即对于Fbatch中每个正样本f，按式(2)替换其中的实体，最终获得候选负样本集Nbatch：

(5)

之后，生成器使用3.1节链接预测模块共享的实体、关系嵌入表示进行训练.对于任一负样本f′∈Nbatch，将其关系与实体的嵌入表示按顺序拼接后，作为双层全连接神经网络的输入，经过前向传播，输出负样本f′生成的概率：

p(f′|f,θG)=
σ(tanh(⊕(f)·W1+b1)·W2+b2)，

(6)

R(f′|f)=tanh(φD(f′)-φD(f)),

(7)

其中，φD表示判别器模块的评分函数.对于生成的高质量负样本集，生成器的总奖励为其中所有负样本的奖励期望值：

(8)

由于生成器输出的负样本下标是一个离散值，面临不可微分的问题，因此采用基于策略梯度的强化学习方法[34]计算生成器梯度：

∇RG(θG)=R(f′|f)×∇Glogp(f′|f,θG),

(9)

根据式(9)中θG的优化方式，生成器倾向于为高质量负样本输出更高的生成概率，以最大化奖励期望.

3.3 判别器模块

(10)

其中，φD表示判别器中的评分函数.以正样本f为例，经过判别器输出的置信度φD(f|θD)表示为

φD(f|θD)=σ(tanh(⊕(f)·W3+b3)·W4+b4),

(11)

其中，W3和W4分别表示全连接神经网络第一、二层的权重矩阵，b3和b4分别表示偏置向量，θD表示判别器模块的参数.

3.4 模型训练

HyperGAN方法的训练过程由知识超图链接预测和生成对抗负采样2阶段构成.

知识超图链接预测阶段的主要目的是优化知识超图节点的嵌入表示与链接预测评分函数的参数，因此，链接预测模块与判别器模块共同参与计算预测损失.其中，固定参数的判别器只进行前向传播以输出权重，并不进行参数更新.这一阶段的训练通过最小化式(4)中的损失函数提升高质量负样本对模型的贡献、降低低质量负样本对模型的干扰，以使模型学得更好的实体和关系的嵌入表示，提升模型的链接预测准确度.

生成对抗负采样阶段的主要目的是对抗训练生成器和判别器模块，共同提高生成器生成高质量负样本的能力与判别器分辨正负样本的能力.在这一阶段中，链接预测模块的参数被固定，生成器模块使用链接预测模块训练好的实体和关系的嵌入表示，对训练正样本集生成高质量负样本集合并送至判别器；判别器对输入的正负样本进行事实合理性判别.在对抗训练过程中，分别采用随机梯度下降法与Adam[35]算法优化式(9)的梯度与式(10)中的二元交叉熵损失，2个模块交替进行参数更新，预期在达到平衡时获得各自最好的性能.

值得注意的是，以往方法将结构复杂的链接预测模型作为生成对抗网络判别器，HyperGAN方法则将预测模型与判别器解耦，采用双层全连接神经网络构建生成器与判别器，因而在保持预测能力的同时简化了生成对抗网络结构.同时，考虑到可能面临的生成对抗网络不稳定性[36]，HyperGAN使用结构相同、参数不一致的生成器与判别器缓解两者间建模能力不平衡的问题.此外，HyperGAN方法采用间断式两阶段训练模式，即在每次迭代时均进行链接预测阶段训练，而间隔C次迭代才进行一次生成对抗训练，减小每次迭代带来的不稳定性.基于以上原因，HyperGAN方法无需像以往结构复杂[15]或共享链接预测模块与判别器[10,12]的工作一样进行预训练，也可保证其训练过程中的稳定性.设计的间断式两阶段训练相比每次迭代均进行训练，亦提升了训练效率.

最终，使用知识超图链接预测模块学习到的嵌入表示作为对实体和关系的最终表示.算法1给出了HyperGAN方法的伪代码.

算法1.HyperGAN方法.

输入：事实集F、链接预测模块损失函数L、判别器损失函数LD、生成器奖励期望RG、总训练次数K、生成器G、判别器D、间隔次数C、对抗网络训练次数M；

输出：事实集F中所有实体嵌入表示e、关系嵌入表示r.

① fori← 1 toKdo

② ifimodC=0 do/*每间隔C次*/

③ forj←1 toMdo

④ for Fbatchof F do

⑤ 通过G生成候选负样本集Nbatch；

⑥ 过滤得到高质量负样本集；

⑦ 计算LD并更新参数θD；

/*式(10)(11)*/

⑧ 计算RG并更新参数θG；

/*式(8)(9)*/

① http://www.freebase.com/

② https://www.wikidata.org/

⑨ end for

⑩ end for

/*式(3)*/

4 实 验

本节为验证HyperGAN方法在知识超图链接预测任务上的有效性，设计了4组实验：1)在5个公开数据集上对多个知识超图链接预测模型应用HyperGAN负采样方法，并将其与均匀随机采样和自对抗负采样方法进行对比，以评估HyperGAN方法的有效性；2)通过效率实验对比不同负采样方法对链接预测模型效率的影响；3)通过参数敏感性实验分析模型参数对实验的鲁棒性影响；4)通过负样本案例分析及定量分析验证HyperGAN方法生成了质量较高的负样本.以上实验均基于PyTorch[37]1.6.0,CUDA 11.0和Python 3.6.6的实验环境，在使用RTX3090显卡的服务器上进行.在详细说明这些实验之前，首先介绍实验中用到的数据集、评价指标、基线模型和实验设置.

4.1 数据集

本文使用5个公开的真实大型知识超图数据集对HyperGAN方法进行评估，其详细统计信息如表1所示.数据集的简要描述：

Table 1 Statistics of Datasets表1 数据集的统计信息

1) JF17K-3[25]是基于Freebase(1)https://github.com/liuyuaa/GETD/过滤得到的三元关系数据集.首先将Freebase中出现次数较少的实体所对应的事实删除，在删除涉及到字符串、数字及枚举类型的事实之后，从每个多元关系中随机选出10 000个事实并进一步删除出现次数少于5的实体所对应的事实；然后，利用文献[18]中的逆向化方法生成多元组；最后过滤元数为3的多元组来构成数据集.

2) JF17K-4[25]是基于Freebase依据类似1)中流程构建的四元关系数据集.

3) WikiPeople-3[30]是基于Wikidata(2)https://github.com/AutoML-Research/过滤得到的三元关系数据集.首先抽取出实体类型为“人物”的全部事实，移除其中涉及到图像类型及包含未知实体的事实，以及删除出现次数少于30的实体所对应的事实，最终过滤得到元数为3的事实来构成数据集.

4) WikiPeople-4[30]是基于Wikidata依据类似3)中流程构建的四元关系数据集.

5) M-FB15K-3是基于M-FB15K[2]过滤得到的三元关系数据集.

4.2 评价指标

本文使用2个广泛应用于知识超图链接预测模型的度量指标对模型性能进行评估：平均倒数排名MRR和击中率Hits@k，k∈{1,3,10}.给定知识超图事实集F，用f=(r,e1,e2,…,en)表示其中任一正样本.对于f中位置m上的实体em，使用知识超图实体集E中的全部实体将其替换，从而得到|E|个样本，删除其中出现在F中的样本，这样就构造出对应于正样本f位置m的负样本集Nm(f)，令Hm(f)={f}∪Nm(f).通过使用链接预测模型中评分函数对Hm(f)中所有样本打分，将得到的评价分数进行降序排序，得到正样本f在Hm(f)中的排名rankm(f).MRR是事实集中所有正样本排名的倒数平均值，Hits@k是正样本排序在前k个的比例.MRR和Hits@k的具体计算：

(12)

(13)

其中，r是正样本f中的关系，cond(·)是条件函数，当条件成立时值为1，否则为0.MRR和Hits@k值越大表明模型性能越好.

4.3 基线模型和实验设置

本文选择6个知识超图链接预测模型作为实验基线模型：

1) m-DistMult[26].对DistMult算法[2]进行了扩展，通过对多元关系和实体的嵌入表示计算哈达玛积，将其加和得到多元关系事实成立的得分.

2) RAE[19].对m-TransH算法[18]进行了扩展，考虑到2个实体同时出现在一个多元关系中的相关性，将其引入模型损失函数进行训练.

3) n-CP[2].对CP算法[27]进行了扩展，通过赋予一个实体多个嵌入表示，将实体在关系中的位置信息引入模型以进行知识超图链接预测.

4) n-TuckER[25].对TuckER算法[28]进行了扩展，基于TuckER分解将知识超图中事实的高阶张量表示分解为一个核张量加若干个因子矩阵的形式，并通过将关系和实体的嵌入表示与核张量相乘，获得多元关系事实成立的得分.

5) GETD[25].对n-TuckER算法进行了扩展，通过引入TR分解，将核张量进一步分解为多个三阶张量，降低了模型的复杂度.

6) S2S[30].对n-TuckER算法进行了扩展，通过将实体和关系的嵌入表示划分为多个块，允许模型从具有混合元数的事实中学习嵌入表示.

为验证HyperGAN方法对链接预测模型的性能提升效果，本文分别在基线模型中链接预测性能良好的GETD与S2S模型上应用本文提出的HyperGAN方法，并在相同的参数设置下，与模型本身采用的均匀随机采样方法和拓展到知识超图的自对抗负采样(Self-Adv)方法[33]进行对比实验.

HyperGAN中生成器和判别器模块分别通过SGD和Adam[35]优化器进行训练，实体和关系的嵌入表示维度、批处理大小、训练次数等参数与框架中采用的知识超图链接预测模型相关，与其在原文中的参数保持一致.其他实验默认参数设置如下，生成对抗网络的间隔迭代次数C=20，SGD和Adam优化器的学习率为0.01，自对抗负采样方法的采样率α为0.5或1，生成对抗网络训练次数M与全连接神经网络隐藏层的维度d在不同数据集中的设置如表2所示.HyperGAN的实验代码和所用数据集将被公开在https://github.com/GuoZhengshan/HyperGAN.

Table 2 Hyper-Parameter Settings of HyperGAN表2 HyperGAN的超参数设置

4.4 有效性分析

表3和表4分别给出了基于HyperGAN方法的多个知识超图链接预测模型在数据集JF17K-3/4和WikiPeople-3/4上的实验结果.其中涉及GETD模型的结果由取自其论文文献[25]中的公开源代码①实验完成.对于S2S模型，由于其未在论文中公开源代码，为进行对比分析，本文使用作者公开GitHub②仓库中的源代码进行实验，并对缺失部分进行了复现补全(该代码未给出元数为4情况下的代码).此外，针对S2S模型代码中缺失超参数的部分，本文参考GETD模型中的超参数进行设置与微调，因此本文报告的部分实验结果与S2S原论文报告的结果有一定差距.尽管如此，由于HyperGAN方法是面向知识超图链接预测模型的通用负采样优化方法，对比加入HyperGAN方法前后的性能实验结果依旧验证了该方法的有效性.除上述基线模型，其余方法的实验结果均采用对应文章中报告的最好结果.

Table 3 The Link Prediction Results on the JF17K-3/4 Datasets表3 JF17K-3/4数据集链接预测结果

Table 4 The Link Prediction Results on the WikiPeople-3/4 Datasets表4 WikiPeople-3/4数据集链接预测结果

实验结果表明：本文提出的HyperGAN方法在上述数据集上都获得了良好的效果.具体地说，就Hits@1指标而言，基于GETD模型的HyperGAN方法相较于其原本均匀随机采样方法在JF17K-3/4和WikiPeople-3/4数据集上分别提升了3.29%,1.59%,4.58%和3.02%.扩展到超图的自对抗负采样(Self-Adv)方法也提升了GETD模型的链接预测性能，就WikiPeople-3数据集而言，其在Hits@1和Hits@3指标上分别提升了2.47%和2.24%，但HyperGAN方法与自对抗负采样方法相比，在各数据集上均表现更优，特别是在WikiPeople-4数据集上，Hits@1和Hits@3指标分别提升了2.25%和2.63%.基于S2S模型的HyperGAN方法对链接预测性能提升实验结果可以发现，HyperGAN方法同样在各数据集上都取得了良好的表现，特别是在WikiPeople-4数据集上的MRR和Hits@1指标分别提升了1.35%和3.75%.然而，自对抗负采样方法在S2S模型上性能不稳定，在JF17K-3/4数据集上的指标稍加提升，而在WikiPeople-3/4数据集上反而下降.导致这一点的原因可能是，更复杂的链接预测模型结构会使自对抗负采样方法在计算权重时具有偏差.相比之下，HyperGAN方法为优化负样本质量而设计的生成对抗网络在结合复杂结构的链接预测模型时依旧能够具有稳定的性能.

为进一步研究HyperGAN在更大规模数据集上的效果，本文基于GETD链接预测模型在数据集M-FB15K-3上进行了实验，表5给出了相应的实验结果.具体地说，Self-Adv方法相较于其原本采用的均匀随机采样方法在各性能指标上均表现不佳，而HyperGAN方法取得了良好的表现，特别是在Hits@10指标上提升了1.21%，从而进一步表明HyperGAN在大规模数据集上同样具有稳定的性能.

Table 5 The Link Prediction Results on the M-FB15K-3表5 M-FB15K-3链接预测结果

4.5 效率分析

本节进一步对比了HyperGAN方法和自对抗负采样(Self-Adv)方法[33]对知识超图链接预测模型的效率影响.具体地，在JF17K-3/4数据集的验证集上分别训练加入了HyperGAN与自对抗负采样方法的S2S链接预测模型，并记录性能评价指标随训练迭代次数增加的变化.

如图3所示，HyperGAN方法与自对抗负采样方法在三元关系数据集JF17K-3上对链接预测模型的收敛速度影响相差不大，但在四元关系数据集JF17K-4上，HyperGAN方法相对自对抗负采样方法具有更快的收敛速度.这一现象表明HyperGAN方法相比以往负采样方法不仅对链接预测模型有更好的性能提升，在元数更大的知识超图数据集上也有更高的效率.

Fig. 3 Performance comparison of HyperGAN and Self-Adv on multiple metrics of S2S method图3 HyperGAN方法和Self-Adv方法在S2S模型各评价指标上的对比

4.6 参数敏感性分析

为测试HyperGAN方法的鲁棒性，本文基于知识超图链接预测模型GETD在JF17K-3数据集上分析了部分超参数对模型性能的影响，包括生成对抗网络与链接预测模型的训练迭代间隔次数C、生成对抗网络训练次数M以及生成器与判别器模块的全连接神经网络隐藏层维度d.令间隔次数C∈{10,20,30,40,50,60}，生成对抗网络训练次数M∈{2,6,12,14,16,18,24}，隐藏层的维度d∈{4,5,6,8,10,16,32}，其余超参数均与4.3节实验设置中的参数保持一致，图4给出了相应的实验结果.

Fig. 4 Parameter sensitivity analysis图4 参数敏感性分析

1) 间隔次数.如图4(a)所示，当训练间隔次数小于20时，随着次数的增加，模型的各性能指标均呈上升趋势，表明增大链接预测模型与生成对抗网络的训练间隔，有助于各模块自身训练，从而使整体的性能得以提升.然而，当间隔次数到达20时，上述指标逐渐平稳不再增加，表明HyperGAN方法性能逐渐稳定，而当次数到达50时，性能出现下降，表明间隔次数过多会导致HyperGAN各模块间信息交互不及时，对模型训练不利.

2) 生成对抗网络训练次数.如图4(b)所示，随着训练次数的增加，各项性能指标均随之变好，表明充分训练生成对抗网络有助于提升模型性能.但当训练次数达到16时，性能指标开始呈下降趋势，表明过多的训练可能导致过拟合问题.因此，生成对抗网络的训练次数对HyperGAN方法而言是一个较为敏感的超参数.

3) 隐藏层维度.如图4(c)所示，当维度开始增加时，各项性能指标均有所提升，表明加大隐藏层的维度有助于链接预测模型性能的提升，太小的维度可能导致学习过程中信息的丢失.然而，当维度到达10时，各项性能指标逐渐平稳不再增加，表明维度到达一定程度时已能够表达所有信息.

为研究链接预测性能的提升与HyperGAN方法的相关性，本文将HyperGAN应用于知识超图场景下基础链接预测模型m-DistMult[26]并记录性能变化，结果如图5所示.多个数据集上的4种性能指标显示，HyperGAN方法即使面对最基础的链接预测模块，也能通过提升负样本质量辅助提升预测精度.

Fig. 5 Performance comparison of m-DistMult and m-DistMult+HyperGAN on multiple datasets图5 m-DistMult和m-DistMult+HyperGAN方法在各数据集上的对比

4.7 负样本质量分析

为了验证HyperGAN方法能生成高质量的负样本，本文基于知识超图链接预测模型GETD在WikiPeople-3/4和JF17K-3/4数据集上进行了案例分析.如表6所示，列2正样本以多元组的形式表示，其中黑体单词表示被替换的实体，列3为经过HyperGAN方法生成的替换后实体.在WikiPeople-3/4数据集中，HyperGAN方法更倾向于选择同一类型的实体进行替换，例如时间实体会被替换为其他时间，影视作品“西城故事”会替换为“指环王：王者归来”，奖项名称“图灵奖”被替换为“诺贝尔物理学奖”.在JF17K-3/4数据集上，HyperGAN方法更倾向于选择有相似语义关系的实体进行替换.例如对于描述网球比赛关系的事实，“塞雷娜·威廉姆斯”会被替换为同为网球运动员的“贾斯汀·海宁”；对于描述大不列颠贵族等级的事实，“公爵”被替换为“子爵”.在以上4个数据集上的案例分析表明，本文提出的HyperGAN方法会选择同类型或是在语义上相似、具有内在联系的实体作为替换，以生成能为链接预测模型训练贡献更大梯度的高质量负样本，而传统的均匀随机采样方法大概率会采样到不属于同一类型或几乎完全不相关的实体作为替换，从而可能会使模型在训练阶段面临零损失问题.

Table 6 The Positive Samples and Replaced Entities with HyperGAN on the JF17K-3/4 and WikiPeople-3/4表6 JF17K-3/4和WikiPeople-3/4中的正样本与HyperGAN替换后的实体

此外，为进一步定量分析负样本质量对链接预测模型的影响，本文在JF17K,WikiPeople和M-FB15K数据集上基于链接预测模型GETD给出了模型损失随训练次数的变化情况.如图6所示，在应用HyperGAN方法后，模型损失相比原方法下降更快，在正样本一致的情况下，这一现象表明通过HyperGAN方法生成的高质量负样本能为链接预测模型带来更大损失，贡献更多梯度，通过解决零损失问题提升了知识超图链接预测的准确度.

Fig. 6 Training loss in one epoch versus number of epochs for GETD and GETD+HyperGAN on datasets图6 GETD和GETD+HyperGAN方法在各数据集上模型损失随训练次数的变化情况

5 总 结

本文提出了面向知识超图链接预测的生成对抗负采样方法HyperGAN，其通过对抗训练生成高质量负样本以提升链接预测模型的性能.此外，HyperGAN方法无需预训练的特性使其在辅助链接预测模型进行训练时相比以往负采样方法具有更高的效率.在5个大型公开知识超图数据集上的对比实验验证了HyperGAN方法在性能与效率方面的有效性与先进性.

下一步，计划探究知识超图不同类型信息间的隐式关联，以提高链接预测中的负样本质量，并分析负样本质量在不同衡量准则下对链接预测性能的影响.其次，考虑将变分自编码器用于链接预测负采样，并且将并行计算融入链接预测模型以进一步提升其在大规模数据集上的训练效率.此外，知识超图的其余下游应用也是值得探索的研究方向.