当前位置:   article > 正文

Reasoning With Neural Tensor Networks for Knowledge Base Completion

reasoning with neural tensor networks for knowledge base completio

[ 论文阅读地址 ]

1. 问题描述

输入

  一个三元组:(e1,R,e2),例如(Bengal tiger, has part, tail)。

输出

  三元组(e1,R,e2) 中,e1e2 有关系R 的置信度。

2. 模型

输出


论文中关于打分函数的形象图示

打分函数:

g(e1,R,e2)=uRTf(e1TWR[1:k]e2+VR[e1e2]+bR)

其中:

  • uRT 是属于类别R的权值。
  • f 是tanh激活函数,对输入的每个元素单独做tanh激活,即f:RnRn
  • e1e2 是输入的实体的向量表示,最初是直接赋随机值,训练时会更新。e1Rd×1 文章后文中提到了两点改进:
    • 不是每个输入实体对应一个向量,而是每个词语对应一个向量,实体向量由词向量求均值得到(作者提到,他们也采用了RNN进行实体向量的学习,但是由于数据中90%的实体都是由不超过两个词语构成,因而RNN并没有比简单的取平均更好)。同时在网络训练时候,也会对词向量进行更新。
    • 对于词向量,采用已经训练好的词向量去作为初始化值会比随机初始化效果更好。

      原论文中的图片
  • WR[1:k] 为和类别R 有关的张量,WR[1:k]Rd×d×ke1TWR[1:k]e2 可以看成
声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop】
推荐阅读
相关标签
  

闽ICP备14008679号