中文

球面知识图谱嵌入:基于几何正则化的模型

机器学习 2025-11-05 v1 人工智能

摘要

知识图谱嵌入(KGE)已成为多关系数据表示学习的基本技术。许多经典模型,如 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.

关键词

引用

@article{arxiv.2511.02460,
  title  = {SKGE: Spherical Knowledge Graph Embedding with Geometric Regularization},
  author = {Xuan-Truong Quan and Xuan-Son Quan and Duc Do Minh and Vinh Nguyen Van},
  journal= {arXiv preprint arXiv:2511.02460},
  year   = {2025}
}