统一建模多种复杂知识图谱,提升链接预测效果
UniHR: Hierarchical Representation Learning for Unified Knowledge Graph Link Prediction
- 将异构事实统一为三元组表示,支持多类型知识图谱
- 通过事实内与事实间消息传递,显著提升预测准确率
- 适用于真实场景中混合类型知识图谱的统一建模
现实世界知识图谱不仅包含标准三元组事实,还包含超关系事实(带键值对)、时间事实(带时间戳)和嵌套事实(事实间的关系)。这些更丰富的表达形式因能更好建模复杂语义而受到关注。然而,现有研究大多仅针对特定类型事实,难以泛化到包含多种事实类型的现实场景;且在复杂表示下,难以实现通用的层次化建模(事实内与事实间)。为此,我们提出UniHR框架,包含学习优化的层次化数据表示(HiDR)模块与统一的层次结构学习(HiSL)模块。HiDR将超关系、时间、嵌套等知识图谱统一为三元组形式;HiSL引入事实内与事实间消息传递机制,同时增强单个事实的语义信息与事实间的结构信息。在5类知识图谱、9个数据集上的实验验证了UniHR的有效性,并揭示了统一表示在复杂现实场景中的巨大潜力。代码与数据已开源。
原文摘要 · Abstract (English)
Real-world knowledge graphs (KGs) contain not only standard triple-based facts, but also more complex, heterogeneous types of facts, such as hyper-relational facts with auxiliary key-value pairs, temporal facts with additional timestamps, and nested facts that imply relationships between facts. These richer forms of representation have attracted significant attention due to their enhanced expressiveness and capacity to model complex semantics in real-world scenarios. However, most existing studies suffer from two main limitations: (1) they typically focus on modeling only specific types of facts, thus making it difficult to generalize to real-world scenarios with multiple fact types; and (2) they struggle to achieve generalizable hierarchical (inter-fact and intra-fact) modeling due to the complexity of these representations. To overcome these limitations, we propose UniHR, a Unified Hierarchical Representation learning framework, which consists of a learning-optimized Hierarchical Data Representation (HiDR) module and a unified Hierarchical Structure Learning (HiSL) module. The HiDR module unifies hyper-relational KGs, temporal KGs, and nested factual KGs into triple-based representations. Then HiSL incorporates intra-fact and inter-fact message passing, focusing on enhancing both semantic information within individual facts and enriching the structural information between facts. To go beyond the unified method itself, we further explore the potential of unified representation in complex real-world scenarios. Extensive experiments on 9 datasets across 5 types of KGs demonstrate the effectiveness of UniHR and highlight the strong potential of unified representations. Code and data are available at https://github.com/zjukg/UniHR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。