arXiv:2507.13001cs.LGcs.AI2025-07

为知识图谱关系设计自适应几何变换,提升嵌入表示效果

SMART: Relation-Aware Learning of Geometric Representations for Knowledge Graphs

  • 基于注意力机制为每类关系选择最优几何变换
  • 在三个基准数据集和金融知识图谱上达到领先性能
  • 适合需要精准关系建模的场景,如金融、医疗推理

知识图谱嵌入(KGE)模型常将关系建模为几何变换,如平移、缩放、旋转等。现有主流方法多采用单一或组合变换统一表示所有关系,忽视了关系特异性。本文提出SMART框架,通过评估每种关系与不同几何变换的匹配度,利用注意力机制为每个关系选择最佳变换类型,或通过多数投票确定全局变换策略。该方法在低维空间学习关系特定的几何变换,并将其相关性映射到高维关系嵌入中。在三个基准知识图谱及一个真实金融知识图谱上的实验表明,模型性能接近顶尖水平。

原文摘要 · Abstract (English)

Knowledge graph representation learning approaches provide a mapping between symbolic knowledge in the form of triples in a knowledge graph (KG) and their feature vectors. Knowledge graph embedding (KGE) models often represent relations in a KG as geometric transformations. Most state-of-the-art (SOTA) KGE models are derived from elementary geometric transformations (EGTs), such as translation, scaling, rotation, and reflection, or their combinations. These geometric transformations enable the models to effectively preserve specific structural and relational patterns of the KG. However, the current use of EGTs by KGEs remains insufficient without considering relation-specific transformations. Although recent models attempted to address this problem by ensembling SOTA baseline models in different ways, only a single or composite version of geometric transformations are used by such baselines to represent all the relations. In this paper, we propose a framework that evaluates how well each relation fits with different geometric transformations. Based on this ranking, the model can: (1) assign the best-matching transformation to each relation, or (2) use majority voting to choose one transformation type to apply across all relations. That is, the model learns a single relation-specific EGT in low dimensional vector space through an attention mechanism. Furthermore, we use the correlation between relations and EGTs, which are learned in a low dimension, for relation embeddings in a high dimensional vector space. The effectiveness of our models is demonstrated through comprehensive evaluations on three benchmark KGs as well as a real-world financial KG, witnessing a performance comparable to leading models

知识图谱几何变换嵌入学习

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