使用Debate Dynamics在知识图谱上进行推理(2020)7.31+8.1+8.2+8.3梳理总结

在这里插入图片描述

摘要

我们提出了一种新的基于 Debate Dynamics 的知识图谱自动推理方法。

其主要思想是将 三元组分类 任务定义为两个强化学习主体之间的辩论游戏,这两个主体提取论点(知识图中的路径),目的是分别促进事实为真(命题)或事实为假(对立面)。基于这些论点,一个二元分类器(称为裁判)决定事实是真是假。这两个代理可以被认为是稀疏的。对抗性特征生成器(Adversarial Feature Generator),为正命题或反命题提供可解释的证据。与其他黑盒方法相比,这些论点允许用户了解法官的决定。

triple classification(三元组分类)
指将知识图谱中的三元组(实体-关系-实体)表示为三元组分类任务,其中实体和关系是输入特征,而实体对之间的关系类型是输出标签或类别。
三元组分类任务旨在训练模型来预测新的三元组类型,以便可以自动构建和扩展知识图谱。
例如,在一个三元组 (巴黎,是首都,法国) 中,巴黎和法国是实体,而“是首都”是关系,三元组分类任务的目标是识别出“是首都”这个关系类型。这种任务对于自动构建大规模知识图谱和语义解析具有重要意义。
输入特征:巴黎, 法国,是首都
输出标签:关系类型(是首都)

由于这项工作的重点是创建一种可解释的方法,以保持有竞争力的预测准确性,我们将我们的方法以三元组分类和链接预测任务为基准。因此,我们发现我们的方法在基准数据集FB15k-237、WN18RR和Hetionet上优于几个基线。我们还进行了一项调查,发现提取的论点对用户来说是有信息的。

Debate Dynamics是一种用于推理的框架。它的主要目的是从知识图谱中自动提取和评估论点,以支持决策制定和信息检索等任务。
Debate Dynamics可以自动从知识图谱中提取论点,并将它们表示为辩论图,该图包括论点和它们之间的关系。然后,它将辩论图输入到论证框架中,该框架使用论证理论来生成反驳和评估论点的强度。最后,它可以根据论证框架的结果做出决策。
Debate Dynamics的优势在于它可以处理复杂的知识图谱,并自动提取和评估论点。它还可以支持多种论证框架,因此可以应用于各种不同的问题领域,例如自然语言处理、信息检索和决策支持系统等。

Adversarial Feature Generator(对抗特征生成器)
是一种机器学习技术,通常用于生成能够欺骗分类器的特征。它的基本思想是通过训练两个对抗的神经网络,一个生成器和一个判别器,从噪声中生成具有特定特征的样本,以欺骗判别器,使它无法准确识别样本的真实标签。
在Adversarial Feature Generator中,生成器试图生成具有与真实样本类似的特征,而判别器则试图区分生成的特征与真实样本的特征。这两个神经网络通过反复训练来提高自己的表现,直到生成器可以生成与真实样本无法区分的特征为止。
例如,生成器可以生成看起来像猫的图像,但判别器却将其误判为狗的图像。这些具有欺骗性的图像可以用于攻击分类器或鉴别器,并且可以帮助我们识别和修复分类器或鉴别器中的漏洞和弱点。

介绍

关于现实世界的各种信息可以用实体及其关系来表达。
知识图谱(KGs)以三元组(s,p,o)的形式存储关于世界的事实,其中s(主体)和o(对象)对应于图中的节点,p(谓词)表示连接两者的边类型。KG中的节点表示整个世界的实体,谓词描述实体对之间的关系。KGs可用于不同领域的各种人工智能任务,如命名实体消除歧义、内部语言处理、视觉相关性检测或协作过滤。
然而,一个主要问题是,大多数真实世界的KGs是不完整的(即,真实的事实缺失)或包含虚假的事实。为解决这一问题而设计的机器学习算法试图根据观察到的连接模式来插入缺失的三元组或检测虚假事实。此外,许多任务,如问答或协作过滤,可以在预测KG中的新链接方面进行模拟。

在KGs上进行推理的大多数机器学习方法都将实体和谓词嵌入到低维向量空间中,然后可以基于这些嵌入来计算三元组的合理性得分。大多数基于嵌入的方法的共同点是它们的黑盒性质,因为它隐藏了用户对这个分数的贡献。当涉及到在现实世界中部署KGs时,这种缺乏透明度的情况构成了一个潜在的限制。在机器学习领域,可解释性最近引起了关注。与 一次性黑盒模型(one-way black-box) 相比,可理解的机器学习方法 能够构建机器和用户交互并相互影响的系统。

如果一个模型是一种一次性黑盒模型(one-way black-box),一次性黑盒模型指的是在知识图谱中,对于每个查询(输入),都需要重新训练模型,而不是将之前的知识用于预测。
以下是一个可能的一次性黑盒模型的例子:
假设我们有一个电影推荐系统,其中知识图谱包括电影、演员、导演等实体以及他们之间的关系。我们想要使用一次性黑盒模型来为用户推荐电影。
对于每个用户查询,我们需要重新训练模型以预测该用户可能喜欢的电影。例如,假设用户A查询推荐电影,我们需要重新训练模型以从知识图谱中提取与用户A相关的实体和关系,并预测用户A可能喜欢的电影。然后,对于用户B的查询,我们需要再次重新训练模型以预测用户B可能喜欢的电影。这个过程需要不断重复,因此被称为一次性黑盒模型。
一次性黑盒模型的缺点是需要大量的计算资源和时间来训练模型,并且无法利用之前的知识进行预测。但是,它的优点是可以根据每个用户的特定偏好进行个性化推荐,从而提高推荐的准确性和效果。

可理解的机器学习方法是指可以解释和理解其决策过程的机器学习方法。这些方法不仅可以生成高质量的预测结果,而且可以解释和传达其预测结果的原因和依据,使用户可以更好地理解和信任模型的决策。
在可理解的机器学习方法中,模型的决策过程通常是基于可解释的特征和规则,而不是基于黑箱模型。这意味着用户可以通过查看模型的特征和规则来理解模型的决策过程,从而使模型的决策更为透明和可靠。

大多数可解释的机器学习方法可以大致分为两组:后解释性(post-hoc interpretability)集成透明度(integrated transparency)
虽然后解释性旨在解释已经训练的黑盒模型的结果,但基于集成透明度的方法要么采用 内部解释机制 ,要么由于 模型复杂性低 自然地可以被解释 (例如,线性模型)。由于低复杂性和预测准确性往往是相互冲突的目标,因此通常需要在性能和可解释性之间进行权衡。
这项工作的目标是设计一种具有集成透明度的KG推理方法,该方法既不牺牲性能,又 允许人类参与

后解释性(post-hoc interpretability)
指在模型已经训练好并且预测能力已经被评估之后,对模型进行解释的过程。与之相对的是先解释性(ante-hoc interpretability),它是指在训练模型时就考虑到模型可解释性的过程。
后解释性的目的是为了让用户理解模型的决策过程和结果,以便更好地理解模型的行为、检测模型的偏差和错误,并为决策提供支持和解释。

Integrated Transparency(集成透明度)
指在机器学习中,将可解释性与模型的其他性能指标集成在一起的方法。旨在实现模型的高性能和可解释性之间的平衡,从而提高模型在实际应用中的可靠性和可信度。
集成透明度方法通常包括以下几个步骤:
1.确定可解释性指标:确定与模型可解释性相关的指标,例如局部解释性、全局解释性和可视化等。
2.确定其他性能指标:确定与模型其他性能指标相关的指标,例如准确性、召回率和精确度等。
3.集成指标:将可解释性指标和其他性能指标集成在一起,以综合评估模型的性能。这可以通过加权平均、决策树或神经网络等方法实现。
4.解释性反馈:根据集成指标的结果,为用户提供解释性反馈,以帮助他们理解模型的决策过程和结果。

Internal Explanation Mechanism(内部解释机制)(集成透明度中)
指在机器学习模型中添加一种用于解释模型决策过程的机制,这种机制通常是由一系列规则、约束或者其他形式的知识表示来实现的。它可以帮助用户理解模型是如何做出决策的,以及模型内部的特征和权重是如何影响模型决策的。
内部解释机制通常包括以下几个方面:
特征重要性:确定模型中每个特征对决策的影响程度。
规则和约束:定义模型内部的规则和约束,以支持对模型决策的解释。
可视化:通过可视化方式展示模型内部的决策过程和特征影响。
解释性反馈:为用户提供有关模型决策过程的解释性反馈。

模型复杂性低是指机器学习模型的结构和参数相对简单,模型的计算和决策过程也相对容易理解和解释。这种模型通常具有较少的参数和较少的层数,可以更快地训练和预测,并且更容易被理解和解释。

允许人类参与(Allowing a human-in-the-loop)可以包括以下几个方面:
数据标注:人类可以对数据进行标注,以帮助模型更好地理解和学习数据的特征和关系。
模型训练:人类可以在模型训练的过程中提供反馈,例如调整模型参数、选择特征等,以改善模型的性能。
模型推断:人类可以对模型的推断结果进行审核和纠错,以提高模型的准确性和可靠性。
模型决策:人类可以参与到模型的决策过程中,例如提供决策依据、制定决策规则等,以保证决策的合理性和可靠性。

本文介绍了一种基于强化学习的三元组分类新方法——R2D2(Reveal Relations using Debate Dynamics)。受通过辩论增加AI的安全性的启发,我们将三元组分类任务建模为两个代理之间的辩论,每个代理都分别提出支持论点(三元组为真)或反对论点(三元组错误)的论据。基于这些论据,一个称为裁判的二元分类器决定事实是真是假。与大多数基于 表示学习 的方法不同,论据可以显示给用户,这样他们就可以追溯裁判的判断,并可能推翻判决或请求额外的论据。因此,R2D2的集成透明机制不是基于 低复杂性组件,而是基于可解释特征的自动提取。
虽然深度学习使手动特征工程在很大程度上成为冗余,但这一优势是以产生难以解释的结果为代价的。我们的工作是试图通过使用深度学习技术自动选择稀疏的、可解释的特征来打破这一循环。这项工作的主要贡献如下。

  1. 据我们所知,R2D2构成了第一个基于Debate Dynamics的KGs推理模型。
  2. 我们在数据集FB15k-237和WN18RR上对R2D2的三重分类进行了基准测试。我们的发现表明,R2D2在准确性、PR AUC和ROC AUC方面优于所有基线方法,同时更具可解释性。
  3. 为了证明R2D2原则上可以用于KG完成,我们还评估了它在FB15k-237子集上的链路预测性能。为了包括现实世界的任务,我们在Hetionet上使用R2D2来寻找药物的基因-疾病关联和新的靶向疾病。R2D2在两个数据集上的标准测量

Representation learning(表示学习)
指一类机器学习方法,旨在自动学习数据的特征表示,以便更好地理解和处理数据。它通过学习数据的高层次特征表示,可以实现更好的数据压缩、分类、聚类、生成等任务,并且可以提高模型的泛化能力和鲁棒性。
表示学习可以自动从原始数据中学习特征表示,避免了手工设计特征表示的过程,并且可以适应不同的数据分布和任务需求。
表示学习通常分为两种类型:有监督和无监督学习。有监督表示学习是指在有标签数据上学习特征表示,例如卷积神经网络(CNN)在图像分类任务中学习特征表示;无监督表示学习是指在无标签数据上学习特征表示,例如自编码器(Autoencoder)在数据压缩和降维任务中学习特征表示。

“低复杂度组件”则是指简单的组件或子模型,它们可以快速地训练和组合,但可能会导致模型的准确性和稳定性降低。
因此,这句话的意思是R2D2算法的集成透明机制并不是基于简单的组件或子模型。相反,它可能使用更复杂和高级的技术来将多个训练模型组合成一个整体,以提高模型的准确性和稳定性。

背景与相关工作

为了指明三元组(s,p,o)是真还是假,我们考虑二元的特征函数在这里插入图片描述
对于所有(s,p,o)∈KG,我们假设在这里插入图片描述(即KG是一个为真的事实的集合)。然而,在XG中不包含三元组的情况下,它并不意味着相应的事实是假的,而是未知的(开放世界假设)。由于目前使用的大多数KG都是不完整的,因为它们不包含所有的真三元组,或者实际上包含虚假的事实,因此许多经典的机器学习任务都与KG推理有关。

KG推理可以大致分为以下两个任务:

  1. 缺失三元组的推理(KG完成或链接预测);
  2. 预测三元组的真值(三元组分类);

虽然这些任务的不同表述通常在文献中找到(例如,完成任务可能涉及预测主体或对象实体以及一对实体之间的关系),但在整个工作中使用以下定义。
在这里插入图片描述

KG completion(Knowledge Graph completion)指示图谱补全
是指通过自动化方法来填补知识图谱中的缺失信息。在文中其任务是:根据实体与给定的主语-谓语对能否形成正确的三元组的可能性大小来对它们进行排序或筛选。
rank object entities指对对象实体进行排序。

许多对KG的机器学习的方法可以被训练在这两种设置中操作。例如,一个三元组分类器 的形式为:在这里插入图片描述
导出了一种由在这里插入图片描述给出的完成方法,其中可以使用不同对象实体的函数值来产生排序。虽然R2D2的体系结构是为三元组分类设计的,但我们证明了它原则上也可以在KG完成设置中工作。

Triple Classifier(三元组分类器)
指一种用于对知识图谱中三元组进行分类的算法或模型。
Triple Classifier的目标是对给定的三元组进行分类,即判断该三元组是否正确或者是否存在错误。这种分类通常包括以下几个类别:
正确的三元组:表示该三元组在知识图谱中存在,并且与实际世界的事实相符合。
错误的三元组:表示该三元组在知识图谱中不存在,或者与实际世界的事实不符合。
不确定的三元组:表示该三元组的正确性无法确定,需要进一步的验证和确认。

表示学习是一种有效且流行的技术,是许多KG精化方法的基础。其基本思想是将实体和关系投影到低维向量空间中,然后将三元组的可能性建模为嵌入空间上的函数。

最近提出了多跳推理方法 MINERVA,这与我们的工作有很大关系,其基本思想是向代理显示查询主题和谓词,并让他们执行策略引导的遍历以找到正确的对象实体。MINERVA产生的路径也导致了某种程度的可解释性。然而,我们发现,只有积极挖掘论文和对立论点,从而暴露辩论的双方,才能让用户做出明智的决定。为这两个位置挖掘证据也可以被视为对抗性特征生成,使分类器(判断)对矛盾证据或损坏的数据具有鲁棒性。

我们的方法

我们根据两个对立主体之间的距离来制定三元组分类的任务。
因此,一个查询三元组对应于辩论的中心陈述。代理通过挖掘KG上的路径来进行,这些路径为论文或对立面提供了证据。
更准确地说,他们顺序遍历图,并根据考虑 过去转换(Past Transition) 和查询三元组的策略选择下一跳。这个转换将添加到当前路径,从而扩展参数。所有路径都由一个名为裁判的二进制分类器处理,该分类器试图根据代理提供的参数来区分真三元组和假三元组。
辩论的主要步骤可以概括如下:

  1. 辩论围绕着一个三元组问题向两个代理提出。
  2. 两位代理轮流从KGs中提取路径,作为各自论点的论据。
  3. 裁判将自变量与查询三元组一起处理,并估计查询三元组的真值。

当裁判的参数通过监督式学习来进行拟合的时候,两个代理(Agent)都是通过强化学习(Reinforcement Learning)算法训练来实现在图中导航的。通过下面列出的固定范围决策过程对代理的学习任务进行建模。

Past Transition(过去转化)
指在逻辑推理中,将一个谓词的过去时态转化为现在时态,以便进行推理和推导。过去时态和现在时态在语法形式上有所不同,但是它们在语义上是等价的,可以互相转化。
在知识推理中,经常需要使用过去时态描述过去的事件或状态,但是在逻辑推理中,过去时态无法直接参与推理。因此,需要使用Past Transition将过去时态转化为现在时态,以便进行推理和推导。例如,将“John was a student”(John曾经是学生)转化为“John is a student”(John是学生),以便在推理过程中使用。
Past Transition通常包括以下几个步骤:
识别谓词的过去时态:首先需要识别谓词的过去时态形式,例如“was”、“had”等。
转化为现在时态:将过去时态转化为现在时态,例如将“was”转化为“is”。
修改主语:根据需要,可能需要修改主语的人称和数,以便与现在时态一致。

在机器学习中,监督式学习是一种通过给定输入和输出数据,训练模型来预测新数据的方法。在监督式学习中,通常需要定义一个模型和一组参数,然后通过训练数据来拟合这些参数,使得模型能够准确地预测输出结果。
在评判者的场景中,也需要使用模型和参数来评估和判断某个对象或行为的好坏。这些参数可以通过监督式学习的方法来进行拟合,以实现更准确的评判和判断。

状态

对于每一个代理的 完全可观察状态空间S 都被给定为在这里插入图片描述。我们就想用这种状态对代理i(i∈{1,2})在时间t和查询三元组q(q=sq,pq,oq)的 探索位置et(i) 进行编码。因此,一个状态St(i) ∈S在时间t属于N上被表示为:在这里插入图片描述

“fully observable state space”(完全可观察状态空间)
指的是一个系统或环境中,所有的状态都是可以被观察的,也就是说,我们可以得到系统或环境中所有对象的完整信息

“location of exploration”(探索位置)
通常指的是在搜索或推理过程中,需要进一步探索或评估的特定位置或节点。
在推理算法中,探索位置通常是指需要进一步评估或推理的知识点或假设。例如,在基于规则的推理系统中,探索位置通常是指需要应用规则来推导出更多结论的规则或前提条件。

action

对于代理 i i i在一个状态在这里插入图片描述的一组可能的行为被表示成在这里插入图片描述。它由来自节点et(i)出边和对应的目标节点组成。更正式地说,在这里插入图片描述=在这里插入图片描述。此外,我们用来表示代理 i i i在时间t上的操作。我们为每个节点加上 自环(self-loop) 以便代理可以停留在当前节点。

环境

环境通过根据代理的操作更新状态(即通过更改代理的位置)来确定地发展,从而使查询事实保持不变。从形式上将,代理 i i i在时间 t t t上的转换函数为在这里插入图片描述在这里插入图片描述

“outgoing edges”(出边)通常指的是图或有向图中一个节点指向其他节点的边。它是从一个节点指向其他节点的边,表示了节点与其他节点之间的关系或连接。
在知识表示和推理中,出边通常用于表示概念之间的关系。例如,一个基于概念网络的知识表示系统中,概念通常被表示为节点,而概念之间的关系则通过出边来表示。例如,一个表示"动物"概念的节点可能有一些出边,分别指向"狗"、“猫”、"马"等其他动物的节点。

policies

我们将代理 i i i到时间 t t t的历史用元组表示:在这里插入图片描述当t>=1且在这里插入图片描述。代理通过LSTMs编码它们的历史:在这里插入图片描述。其中在这里插入图片描述对应前一个动作的向量空间嵌入(t=0时为零向量),其中rt-1(i)和et(i)分别表示关系和目标实体嵌入R中。此外,在这里插入图片描述为代理 i i i编码查询三元组。实体和关系嵌入都是针对每个主体的,并且是在训练的辩论过程中学习的。

每个代理的历史相关动作分布由下式给出:
在这里插入图片描述
其中在这里插入图片描述

Debate Dynamics

在第一步,具有真值在这里插入图片描述的查询三元组q=(sq,pq,oq)被表示成两个代理。代理1认为这个事实为真,代理2认为这个事实为假。与大多数正式辩论类似,我们考虑固定数量的回合N,在每一个回合n=1,2,…,N中,代理程序使用固定长度T从查询主体节点sq开始,横向扫描图。裁判监督代理的路径并预测三元组的真值。代理1开始行动,根据等式(1-3)生成由状态和动作组成的长度为T的序列。然后,代理2从Sq开始生成一个相似的序列。算法1在推理时包含R2D2的后代码。

为了简化注释,我们枚举了所有操作,并删除了指示哪个代理执行该操作的上标。然后对应第n个序列给出代理i的论据:
在这里插入图片描述

裁判

裁判在R2D2中的作用有两个:

  1. 裁判是一个试图区分真实和虚假事实的二元分类器。
  2. 裁判还评估代理人提取的论点的质量,并为他们分配奖励。

因此,裁判也扮演着批评者的角色,教导代理人提出有意义的论点。裁判通过前馈神经网络在这里插入图片描述单独处理每个自变量和查询,对每个自变量的输出求和,并通过二元分类器处理结果求和。更具体地说,在单独处理每个参数后,裁判根据在这里插入图片描述做出表述,
其中在这里插入图片描述
在这里插入图片描述
编码查询谓词和查询对象。注意,查询主体不会透露给裁判,因为我们希望裁判仅根据代理的行为而非查询主体的嵌入来做决定。处理完 τ τ τ中的所有自变量后,辩论终止,裁判根据
在这里插入图片描述对查询三元组进行打分。其中,在这里插入图片描述代表分类器可以训练的范式,函数是功能激活函数。

通过交叉熵损失给出了单个查询q的判断的目标函数
在这里插入图片描述

Cross-entropy loss(交叉熵损失)通常是指在机器学习中使用交叉熵作为损失函数的情况。交叉熵损失常用于分类问题,特别是多类别分类问题。
在分类问题中,假设有 n n n 个不同的类别,每个样本 i i i 的真实标签为 y i y_i yi,模型对于该样本的预测概率分布为 y ^ i \hat{y}_i y^i。则交叉熵损失可以表示为:
L C E = − 1 N ∑ i = 1 N ∑ j = 1 n y i , j log ⁡ ( y ^ i , j ) L_{CE} = -\frac{1}{N}\sum_{i=1}^{N}\sum_{j=1}^{n}y_{i,j}\log(\hat{y}_{i,j}) LCE=N1i=1Nj=1nyi,jlog(y^i,j)
其中, N N N 表示样本数量, y i , j y_{i,j} yi,j 表示样本 i i i 是否属于类别 j j j y ^ i , j \hat{y}_{i,j} y^i,j 表示模型对于样本 i i i 属于类别 j j j 的预测概率。
交叉熵损失可以看作是模型预测概率分布和真实标签分布之间的距离的度量。当模型对于样本的预测和真实标签一致时,交叉熵损失为0,表示模型的预测结果完全正确。当模型对于样本的预测和真实标签不一致时,交叉熵损失会增大,表示模型的预测结果与真实标签之间的差异越大。。

因此,通过测试,我们目标是通过在这里插入图片描述来最小化损失。其中 τ τ τ代表训练三元组的集合。为了防止过拟合,在上述式子中添加了一个强度为λ∈R>0的额外惩罚项。

奖励

为了给代理人提供反馈,裁判还单独处理每个论点 τ n τ_n τn,并根据在这里插入图片描述产生一个分数。
其中神经网络 f f f和线性权重向量 w w w都对应于前一段中给出的定义。因此,t(i)n对应于仅基于代理i的第n个论点q的分类分数。由于代理1对论点进行了论证,代理人2对论点进行了反驳,因此奖励由在这里插入图片描述给出。

直观地说,这意味着,每当代理人提取出一个被裁判认为是有力证据的论点时,他们就会获得更高的报酬。

报酬最大化和培训计划

我们利用REINFORCE来最大化代理预期的累积报酬:在这里插入图片描述
因此,代理的最大化问题由在这里插入图片描述给出。

实验

数据集

我们在基准数据集FB15k-237和WN18RR上测量了R2D2在三元组分类和KG完成任务方面的性能。为了在真实世界任务中测试R2D2,我们还考虑了Hetionet,这是一个大规模的 Heterogeneous graph(异构图) ,包含了有关化合物、疾病、基因和分子功能的信息,我们使用R2D2来检测基因-疾病相关性和为药物寻找新的靶向疾病,这两项任务在生物医学领域具有高度的实际相关性。

Heterogeneous graph(异构图)是指由多种类型的节点和边构成的图。在异构图中,不同类型的节点和边之间可以有不同的语义和关系,例如一个社交网络中,用户节点和商品节点、用户节点和关注节点之间的关系是不同的,这些不同类型的节点和边构成了一个异构图。
与传统的同构图(homogeneous graph)相比,异构图具有以下特点:
1 不同类型的节点和边之间有不同的语义和关系。
2 不同类型的节点和边之间可能存在不同的属性和特征。
3 不同类型的节点和边可能具有不同的重要性和影响力。
4 异构图中的节点和边数量可能非常庞大,需要设计有效的算法和模型进行处理。

度量和评估方案

如第二节所述,三元组分类旨在决定查询三元组(sq,pq,oq)是真是假。因此,它是一个二元分类任务。对于每种方法,我们都设置了一个阈值 σ σ σ,通过最大化验证集的准确性来获得。这意味着,对于给定的查询三元组(sq,pq,oq),如果其得分(例如,R2D2的等式(8)给定)大于 σ σ σ,则三元组将被分类为真,否则为假。由于大多数KGs不包含标记为假的事实,因此需要对 负三元组(Negative Triples) 进行采样。我们生成了一组负三元组:对于验证和测试集中观察到的每个三元组,我们制造虚假但看似合理的事实。我们记录PR AUC,ROC AUC在所有的方法上的真实性。

Negative Triples(负三元组)是指不符合事实的三元组。

尽管R2D2的目的是对三元组分类,但可以将其转化为KG完成方法,如下所示:
我们考虑一系列对象实体,每个实体都产生由等式(8)给出的不同分类分数 t τ t_τ tτ。由于 t τ t_τ tτ 可以被解释为对样本合理性的衡量,我们使用分类分数来产生秩。

为了为所有考虑的方法找到最合适的超参数集,我们进行了 交叉验证(Cross-validation) 。因此,使用了将数据集规范拆分为训练集、验证集和测试集的方法。我们确保在训练期间,分配给验证或测试集的三元组(及其各自的逆关系)不包括在KG中。所有方法的测试集结果都是基于验证集上显示最佳性能的超参数(基于最高精度的多重分类和最高MRR的链接预测)报告的。
我们考虑了下列R2D2的超参数范围:
嵌入的潜在维度d的数量选自范围{32,64128}。
代理的LSTM层数选自{1,2,3}。
裁判MLP中的层数在{1,2,4,5}范围内调整。
β β β从{0.02,0.05,0.1}中选择。
每个观点的长度 T T T在{1,2,3}范围内进行调整,辩论轮数N设置为3。
此外,L2正则化强度 λ λ λ被设置为0.02。
训练期间的卷积次数由20和50(三元组分类)或测试时的100(完成KG)给出。
使用学习率为10-4的Adam优化了法官的损失和代理人的奖励。

Cross-validation(交叉验证)
是一种常用的模型评估方法,用于评估机器学习模型的性能和泛化能力。在交叉验证中,将数据集分为训练集和测试集,利用训练集训练模型,然后用测试集评估模型的性能和泛化能力。由于数据集的划分方式可能会影响模型评估结果的稳定性和可靠性,因此交叉验证采用了多次重复划分数据集的方式,从而得到更加准确和稳定的模型评估结果。
具体来说,交叉验证分为以下几个步骤:
1.将数据集分为k个大小相等的子集。
2.选择一个子集作为测试集,其余k-1个子集作为训练集。
3.在训练集上训练模型,并在测试集上评估模型性能。
4.重复步骤2-3,直到每个子集都被用作一次测试集。
5.计算所有评估结果的平均值作为模型的性能指标。
这种交叉验证方法称为k折交叉验证(k-fold cross-validation)。

结果

在这里插入图片描述

总结

知识推理大多数的方法都是将实体和位于嵌入到低维向量空间中,基于这些嵌入来计算三元组的合理性得分。这是黑盒模式,可解释性差,所以可解释性的机器学习方法被想到了。

可解释性的机器学习方法可以分为后解释性和集成透明度,我们本文主要考虑的是集成透明度的方法。

集成透明度又有内部解释机制和模型复杂度本身就低两种情况,而模型复杂度低往往预测的准确性也低。

本文提出一种新的方法——R2D2。旨在将三元组分类任务建模为两个主体之间的辩论,每个主体提出支持三元组为真或三元组为假的两个论点的论据,裁判(二元分类器)判断事实是真是假。而且代理们提出的论据可以给用户,用户可以追溯论据,选择推翻裁判的决定或者请求额外论据。

步骤:

  1. 围绕一个三元组,向两个代理提出这个三元组
  2. 两个代理轮流从KG中提取路径作为其各自论据
  3. 裁判将自变量与查询三元组一起处理,并估计查询三元组的真值。

文中给出了代理i的状态的公式、其一组可能的操作的公式、代理i在时间t上的转换的公式、代理i到时间t的历史的表示、裁判的打分公式、奖励的公式(分别包括代理1和代理2的)。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值