arXiv:2403.10231v2 通知类型: replace-cross 摘要：为了从知识图谱（KG）中推导出新的事实，链接预测器通过对图结构的学习并收集局部证据来回答给定的查询。然而，现有的方法由于利用整个KG进行预测而面临着严重的可扩展性问题，这阻碍了它们在大规模KG上的应用，并且无法通过简单的抽样方法直接解决。在这项工作中，我们提出了单次子图链接预测，以实现高效和自适应的预测。设计原则是，而不是直接作用于整个KG，预测过程被分解为两个步骤，即(i) 根据查询提取仅一个子图，以及(ii) 在这个单一的、查询依赖的子图上进行预测。我们揭示了无参且计算高效的启发式方法Personalized PageRank（PPR）能够有效识别潜在的答案和支持证据。通过基于子图的高效预测，我们进一步介绍了在数据空间和模型空间中自动搜索最佳配置的方法。实证上，我们在五个大规模基准测试中实现了提升的效率和领先的性能。该代码已公开发布在：https://github.com/tmlr-group/one-shot-subgraph。