3 个月前

关系感知的集成学习用于知识图谱嵌入

关系感知的集成学习用于知识图谱嵌入

摘要

知识图谱(Knowledge Graph, KG)嵌入是自然语言处理领域的一项基础任务,已有多种方法被提出,以不同方式探索语义模式。本文提出一种关系感知的集成学习方法,通过协同利用现有方法来构建集成模型。然而,采用关系感知的集成策略探索语义信息时,其搜索空间远大于传统集成方法。为解决这一问题,本文提出一种“分治-搜索-合并”算法——RelEns-DSC,该算法独立地为每种关系学习集成权重。该方法在计算开销上与传统集成方法相当,但性能显著更优。在基准数据集上的实验结果表明,所提方法能够高效搜索关系感知的集成权重,并在知识图谱嵌入任务中达到当前最优性能。相关代码已公开,地址为:https://github.com/LARS-research/RelEns。

代码仓库

lars-research/relens
官方
GitHub 中提及
LARS-research/RelEns
GitHub 中提及

基准测试

基准方法指标
link-property-prediction-on-ogbl-biokgRelEns
Ext. data: No
Number of params: 849427106
Test MRR: 0.9618 ± 0.0002
Validation MRR: 0.9627 ± 0.0004
link-property-prediction-on-ogbl-wikikg2RelEns
Ext. data: No
Number of params: 2176767622
Test MRR: 0.7392 ± 0.0011
Validation MRR: 0.7509 ± 0.0009

用 AI 构建 AI

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

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

Hyper Newsletters

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