arXiv:2511.02460cs.LGcs.AI2025-11

将实体嵌入球面以提升知识图谱表示学习效果

SKGE: Spherical Knowledge Graph Embedding with Geometric Regularization

  • 用可学习的非线性映射将实体约束在超球面上
  • 在三个基准数据集上显著优于经典欧氏模型TransE
  • 球面几何自带难负样本机制,提升模型鲁棒性

知识图谱嵌入(KGE)已成为多关系数据表示学习的核心技术。许多经典模型如TransE在无界欧氏空间中运行,存在建模复杂关系能力不足和训练效率低的问题。本文提出球面知识图谱嵌入(SKGE),突破这一范式,将实体表示约束在紧凑流形——超球面上。SKGE采用可学习的非线性球化层将实体映射至球面,并将关系解释为先平移后投影的混合变换。在FB15k-237、CoDEx-S和CoDEx-M三个基准数据集上的大量实验表明,SKGE在包括大规模数据集在内的各项任务中持续且显著优于其强欧氏对手TransE,证明了球面几何先验的有效性。深入分析揭示,该几何约束起到了强大正则化作用,使所有类型关系均获性能提升。更根本的是,我们证明球面几何创造了‘固有的难负样本’环境,自然消除平凡负样本,迫使模型学习更鲁棒、语义一致的表示。研究有力表明,流形选择不仅是实现细节,更是核心设计原则,主张几何先验应成为下一代强大稳定KGE模型的设计基石。

原文摘要 · Abstract (English)

Knowledge graph embedding (KGE) has become a fundamental technique for representation learning on multi-relational data. Many seminal models, such as TransE, operate in an unbounded Euclidean space, which presents inherent limitations in modeling complex relations and can lead to inefficient training. In this paper, we propose Spherical Knowledge Graph Embedding (SKGE), a model that challenges this paradigm by constraining entity representations to a compact manifold: a hypersphere. SKGE employs a learnable, non-linear Spherization Layer to map entities onto the sphere and interprets relations as a hybrid translate-then-project transformation. Through extensive experiments on three benchmark datasets, FB15k-237, CoDEx-S, and CoDEx-M, we demonstrate that SKGE consistently and significantly outperforms its strong Euclidean counterpart, TransE, particularly on large-scale benchmarks such as FB15k-237 and CoDEx-M, demonstrating the efficacy of the spherical geometric prior. We provide an in-depth analysis to reveal the sources of this advantage, showing that this geometric constraint acts as a powerful regularizer, leading to comprehensive performance gains across all relation types. More fundamentally, we prove that the spherical geometry creates an "inherently hard negative sampling" environment, naturally eliminating trivial negatives and forcing the model to learn more robust and semantically coherent representations. Our findings compellingly demonstrate that the choice of manifold is not merely an implementation detail but a fundamental design principle, advocating for geometric priors as a cornerstone for designing the next generation of powerful and stable KGE models.

知识图谱嵌入模型几何先验球面表示

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