
摘要
知识图谱是大规模事实数据库的图形表示,通常存在不完整性问题。推断实体(节点)之间缺失的关系(链接)是链接预测的任务。尽管ConvE是一种最新的链接预测方法,通过实现卷积神经网络从连接的主题和关系向量中提取特征,其结果令人印象深刻,但该方法缺乏直观性和理解度。我们提出了一种超网络架构,该架构生成简化的特定关系卷积滤波器,这些滤波器(i)在标准数据集上超越了ConvE和所有先前的方法;并且(ii)可以被表述为张量分解,从而纳入到已建立的链接预测因子分解模型家族中。因此,我们证明了卷积仅仅提供了一种方便的计算手段来引入稀疏性和参数绑定,以找到非线性表达能力和需要学习的参数数量之间的有效平衡。
代码仓库
ibalazevic/HypER
官方
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| link-prediction-on-fb15k | HypER | Hits@1: 0.734 Hits@10: 0.885 Hits@3: 0.829 MRR: 0.790 |
| link-prediction-on-fb15k-237 | HypER | Hits@1: 0.252 Hits@10: 0.520 Hits@3: 0.376 MRR: 0.341 |
| link-prediction-on-wn18 | HypER | Hits@1: 0.947 Hits@10: 0.958 Hits@3: 0.955 MRR: 0.951 |
| link-prediction-on-wn18rr | HypER | Hits@1: 0.436 Hits@10: 0.522 Hits@3: 0.477 MR: 5796 MRR: 0.465 |