摘要
arXiv:2501.19095v1 宣告类型: 新
摘要:知识图谱(Knowledge Graphs,KGs)以实体(节点)和关系的形式存储人类知识,并在各种应用中广泛使用。知识图嵌入是一种有效的方法,可以解决知识发现、链接预测和推理等任务。这通常通过为所有或部分实体分配和学习嵌入表来实现。由于这一方法需要按实体数量线性扩展计算资源,因此在具有数百万节点的真实世界KG中学习嵌入模型可能是计算上不可行的。为了解决这个问题,我们的模型PathE仅为关系(通常比实体少几个数量级)分配嵌入表,并且所需参数少于之前参数高效方法的25%。我们不存储实体嵌入,而是通过利用多个实体-关系路径来为三元组中的 individual 实体提供上下文,从而学习计算它们。在四个基准测试上评估,PathE在关系预测上实现了最先进的性能,并在路径丰富的知识图谱上保持了在链接预测方面的竞争力,同时使用消费级硬件进行训练。我们进行了消融实验来测试我们的设计选择,并分析了模型对关键超参数的敏感性。对于常见于实际应用中的关系多样且连接良好的知识图谱,PathE具有高效和成本效益。