arXiv:2608.22981cs.LG2026-08

用语义损失增强知识图谱的层次结构一致性,提升链接预测性能。

Hierarchy-Aware Semantic Losses for Knowledge Graph Link Prediction

论文配图:Hierarchy-Aware Semantic Losses for Knowledge Graph Link Prediction
图 1 · 摘自论文原文
  • 基于盒嵌入设计语义损失,显式约束表示满足类别层级关系。
  • 在AIFB、CoDEx、BioKG上MRR提升7.6%至15.5%。
  • 相比添加子类边,该方法更高效且效果更优,适合结构不完整场景。

知识图谱常包含编码语义信息的本体类别层次结构,但多数链接预测方法忽略或间接通过额外图边引入这些信息。近期工作提出层次感知图神经网络(GNN),利用盒嵌入生成的语义损失,在表示学习中促进子类关系满足。本文评估该方法在三个基准数据集(AIFB、CoDEx、BioKG)上的链接预测表现。将GNN编码器与基于盒嵌入的语义损失结合,使学习表示更符合本体导出的类别层级,并与标准模型及通过添加子类边引入层次信息的模型对比。在所有数据集上,层次感知语义损失显著提升均倒数排名(MRR),并持续优于添加子类边的模型。相较基线GNN模型,各数据集上MRR分别提升7.6%、2.4%和15.5%。结果表明,本体层级提供与图结构互补的信息,通过语义损失强制层级一致性是有效且参数高效的改进方式。

原文摘要 · Abstract (English)

Knowledge graphs are often accompanied by ontological class hierarchies that encode valuable semantic information, yet many link prediction methods either ignore such hierarchies or incorporate them indirectly through additional graph edges. Recent work introduced hierarchy-aware graph neural networks (GNNs), which use semantic losses derived from box embeddings to encourage satisfaction of subclass relationships during GNN-based representation learning. While this approach has shown promise for biological regression tasks, its effectiveness for knowledge graph link prediction has not been investigated. In this paper we evaluate hierarchy-aware semantic losses on link prediction across three benchmark datasets: AIFB, CoDEx, and BioKG. We combine graph neural network encoders with box-embedding-based semantic losses that encourage learned representations to better satisfy ontology-derived class hierarchies, and compare this approach to both standard link prediction models and models incorporating subclass relations as graph edges. Across all datasets, hierarchy-aware semantic losses significantly improve mean reciprocal rank (MRR) and consistently outperform models that incorporate hierarchy information through additional subclass edges. Relative to the baseline GNN models, MRR improved by 7.6%, 2.4%, and 15.5% on AIFB, CoDEx, and BioKG, respectively. Furthermore, semantic losses consistently outperform the alternative of augmenting the graph with subclass edges. These results are consistent with ontology-derived class hierarchies providing complementary information to graph structure, and suggest that encouraging hierarchical consistency through semantic losses is an effective and comparatively parameter-efficient mechanism for improving knowledge graph link prediction.

知识图谱链接预测层次结构盒嵌入

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。