arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2010.03496cs.CLcs.AIcs.LG

通过链接预测进行文本中的归纳实体表示

Inductive Entity Representations from Text via Link Prediction

Vrije Universiteit Amsterdam(阿姆斯特丹自由大学);University of Amsterdam(阿姆斯特丹大学);Discovery Lab, Elsevier(Elsevier发现实验室)

Daniel Daza, Michael Cochez, Paul Groth

AI总结:

本文提出通过链接预测目标评估实体表示,展示其在链接预测、实体分类和信息检索中的良好泛化能力,优于现有方法。

摘要:

知识图谱(KG)在多个网络应用中至关重要,包括信息检索、推荐系统和元数据注释。无论是否由领域专家手动构建还是通过自动流程构建,KG通常不完整。最近的研究开始探索如何利用知识图谱中的文本描述来学习实体向量表示以进行链接预测。然而,这些为链接预测学习的表示在其他任务中的泛化程度尚不明确。鉴于学习这些表示的成本,我们理想地希望在转移到不同任务时无需重新训练,同时保持合理的性能。在本文中,我们提出了一种全面的评估协议,用于通过链接预测目标学习的实体表示。我们考虑了归纳链接预测和实体分类任务,这些任务涉及训练期间未见过的实体。我们还考虑了面向实体的检索任务。我们评估了一种基于预训练语言模型的架构,该架构在训练期间未见过的实体上表现出强大的泛化能力,并在链接预测任务上平均提高了22%的MRR。我们进一步提供了证据,表明所学的表示在没有微调的情况下能很好地转移到其他任务。在实体分类任务中,我们相比使用预训练模型的基线方法平均提高了16%的准确性。在信息检索任务中,我们对自然语言查询在NDCG@10上取得了高达8.8%的显著改进。因此,我们展示了所学的表示不仅限于KG特定任务,其泛化能力比之前的工作评估得更广。

英文摘要:

Knowledge Graphs (KG) are of vital importance for multiple applications on the web, including information retrieval, recommender systems, and metadata annotation. Regardless of whether they are built manually by domain experts or with automatic pipelines, KGs are often incomplete. Recent work has begun to explore the use of textual descriptions available in knowledge graphs to learn vector representations of entities in order to preform link prediction. However, the extent to which these representations learned for link prediction generalize to other tasks is unclear. This is important given the cost of learning such representations. Ideally, we would prefer representations that do not need to be trained again when transferring to a different task, while retaining reasonable performance. In this work, we propose a holistic evaluation protocol for entity representations learned via a link prediction objective. We consider the inductive link prediction and entity classification tasks, which involve entities not seen during training. We also consider an information retrieval task for entity-oriented search. We evaluate an architecture based on a pretrained language model, that exhibits strong generalization to entities not observed during training, and outperforms related state-of-the-art methods (22% MRR improvement in link prediction on average). We further provide evidence that the learned representations transfer well to other tasks without fine-tuning. In the entity classification task we obtain an average improvement of 16% in accuracy compared with baselines that also employ pre-trained models. In the information retrieval task, we obtain significant improvements of up to 8.8% in NDCG@10 for natural language queries. We thus show that the learned representations are not limited KG-specific tasks, and have greater generalization properties than evaluated in previous work.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑