RelBall用四元数旋转建模知识图谱关系,支持全类型关系与语义层次。
RelBall: Relation Ball with Quaternion Rotation for Knowledge Graph Completion

- 用模长变换和尾部中心关系球建模语义层次与多对多关系
- 在多个数据集上达到领先链接预测性能,覆盖对称/反称/非交换组合等模式
- 适合需要理解关系层级与复杂关联的场景,如医疗、金融知识图谱
现实世界的知识图谱常缺少大量有效事实。知识图谱补全(KGC)旨在利用已知三元组预测缺失链接,从而提升图谱覆盖率。核心挑战在于建模多样化的关系模式,如对称性、反对称性、逆关系、复合关系及语义层次。现有模型如RotatE可捕捉对称、反对称、逆关系和可交换复合关系,但难以处理非可交换复合关系。Rotate3D通过三维旋转引入非可交换性,但仍无法有效建模知识图谱中普遍存在的语义层次,且不能充分表示一对多关系。为此,本文提出RelBall,基于Rotate3D进行两项改进:首先引入模长变换以建模层次结构,使抽象概念对应较小模长,具体实例对应较大模长;其次引入尾部中心关系球,可统一建模一对一、一对多、多对一及多对多关系。RelBall具备三大优势:(1) 全面覆盖前述各类关系模式;(2) 提供可解释的层次化表示,模长直接反映语义层级;(3) 支持各类关系类型。在多个数据集上的实验表明,RelBall在链接预测任务中表现优于多种基线模型。
原文摘要 · Abstract (English)
Real-world knowledge graphs are often incomplete, lacking many valid facts. Knowledge Graph Completion (KGC) aims to predict missing links using known triples, thereby enhancing graph coverage. A key challenge is modeling diverse relational patterns such as symmetry, antisymmetry, inversion, composition and semantic hierarchy. Existing models such as RotatE can capture symmetric, antisymmetric, inverse, and commutative composition patterns, yet struggle with non-commutative composition. Rotate3D addresses this by introducing non-commutativity via three-dimensional rotations, but still fails to capture the semantic hierarchies prevalent in knowledge graphs. Moreover, both models cannot effectively model one-to-many relations. To overcome these limitations, we propose RelBall, which extends Rotate3D with two innovations. First, our model introduces modulus transformation to model hierarchies, driving abstract concepts toward smaller moduli and concrete instances toward larger ones. Second, it introduces a tail-centric relation ball to model one-to-one, one-to-many, many-to-one, and many-to-many relations. RelBall offers the following advantages: (1) coverage of all relational patterns, including the ones mentioned above; (2) an interpretable hierarchical representation where the modulus directly reflect semantic levels; (3) support for one-to-one, one-to-many, many-to-one, and many-to-many relations. Experiments on multiple datasets demonstrate RelBall's competitive link prediction performance against various baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。