4 个月前

DRUM:知识图谱上的端到端可微规则挖掘

DRUM:知识图谱上的端到端可微规则挖掘

摘要

本文研究了用于归纳性和可解释链接预测的概率逻辑规则学习问题。尽管归纳链接预测具有重要意义,但大多数先前的研究主要集中在演绎链接预测上,无法处理之前未见过的实体。此外,这些模型大多是黑箱模型,对人类来说不易解释。我们提出了DRUM(Differentiable Rule Mining),这是一种可扩展且可微分的方法,可以从知识图谱中挖掘一阶逻辑规则,从而解决了这些问题。我们的方法通过建立每个规则的学习置信度评分与低秩张量近似之间的联系来加以论证。DRUM利用双向循环神经网络(RNN)在不同关系的规则学习任务之间共享有用信息。我们还通过实验展示了DRUM在多种基准数据集上的归纳链接预测任务中相对于现有规则挖掘方法的高效性。

代码仓库

alisadeghian/DRUM
tf
GitHub 中提及

基准测试

基准方法指标
link-prediction-on-fb15k-237DRUM (T=3)
Hits@1: 0.255
Hits@10: 0.516
Hits@3: 0.378
MRR: 0.343
link-prediction-on-wn18rrDRUM (T=3)
Hits@1: 0.425
Hits@10: 0.586
Hits@3: 0.513
MRR: 0.486

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
DRUM:知识图谱上的端到端可微规则挖掘 | 论文 | HyperAI超神经