4 个月前

超网络知识图谱嵌入

超网络知识图谱嵌入

摘要

知识图谱是大规模事实数据库的图形表示,通常存在不完整性问题。推断实体(节点)之间缺失的关系(链接)是链接预测的任务。尽管ConvE是一种最新的链接预测方法,通过实现卷积神经网络从连接的主题和关系向量中提取特征,其结果令人印象深刻,但该方法缺乏直观性和理解度。我们提出了一种超网络架构,该架构生成简化的特定关系卷积滤波器,这些滤波器(i)在标准数据集上超越了ConvE和所有先前的方法;并且(ii)可以被表述为张量分解,从而纳入到已建立的链接预测因子分解模型家族中。因此,我们证明了卷积仅仅提供了一种方便的计算手段来引入稀疏性和参数绑定,以找到非线性表达能力和需要学习的参数数量之间的有效平衡。

代码仓库

ibalazevic/HypER
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
link-prediction-on-fb15kHypER
Hits@1: 0.734
Hits@10: 0.885
Hits@3: 0.829
MRR: 0.790
link-prediction-on-fb15k-237HypER
Hits@1: 0.252
Hits@10: 0.520
Hits@3: 0.376
MRR: 0.341
link-prediction-on-wn18HypER
Hits@1: 0.947
Hits@10: 0.958
Hits@3: 0.955
MRR: 0.951
link-prediction-on-wn18rrHypER
Hits@1: 0.436
Hits@10: 0.522
Hits@3: 0.477
MR: 5796
MRR: 0.465

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
超网络知识图谱嵌入 | 论文 | HyperAI超神经