4 个月前

增强和调优知识图谱嵌入

增强和调优知识图谱嵌入

摘要

知识图谱嵌入是知识图谱中链接预测任务最成功的方法之一,即完成一组不完整的关联事实的任务。这些模型的一个缺点是它们对模型超参数(特别是正则化项)具有很强的敏感性,需要进行广泛的调参才能达到良好的性能[Kadlec等人, 2017]。我们提出了一种高效的大规模超参数调优方法,通过在概率框架下解释这些模型来实现。在引入每个实体的超参数后,我们采用变分期望最大化方法以最小的额外成本调整数千个这样的超参数。我们的方法对模型的具体细节具有鲁棒性,并在标准基准数据上实现了链接预测的新水平。

基准测试

基准方法指标
link-prediction-on-fb15kDistMult (after variational EM)
Hits@10: 0.914
MRR: 0.841
link-prediction-on-fb15k-237DistMult (after variational EM)
Hits@10: 0.548
MRR: 0.357
link-prediction-on-wn18DistMult (after variational EM)
MRR: 0.911
link-prediction-on-wn18rrDistMult (after variational EM)
MRR: 0.455

用 AI 构建 AI

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

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

Hyper Newsletters

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