用自适应空间融合提升知识图谱嵌入的多样性与精度
HyperComplEx: Adaptive Multi-Space Knowledge Graph Embeddings
- 根据关系类型动态选择超球、复数或欧式空间进行建模
- 在百万级论文数据集上达到0.612 MRR,比最优基线高4.8%
- 适合需要处理复杂关系结构的大规模知识图谱任务
知识图谱已成为科学与企业领域表示复杂关系数据的基础结构。现有嵌入方法在大规模建模多样关系类型时存在局限:欧氏模型难以捕捉层次结构,向量空间模型无法表达关系不对称性,而双曲模型在对称关系上表现不佳。我们提出 HyperComplEx,一种通过可学习注意力机制自适应融合双曲、复数与欧氏空间的混合嵌入框架。基于关系特性的空间加权策略动态为每类关系选择最优几何结构,多空间一致性损失确保跨空间预测的一致性。我们在涵盖1K至1000万篇论文(对应约25K至4500万三元组)的计算机科学知识图谱上进行了评估,结果表明其持续优于TransE、RotatE、DistMult、ComplEx、SEPA和UltraE等前沿基线。标准基准测试也显示显著更优性能。在1000万论文数据集上,HyperComplEx实现0.612 MRR,相较最佳基线提升4.8%,同时保持高效训练,每三元组推理仅需85毫秒。模型通过自适应维度分配实现近线性扩展。我们开源了代码与数据集,以促进可复现的大规模知识图谱嵌入研究。
原文摘要 · Abstract (English)
Knowledge graphs have emerged as fundamental structures for representing complex relational data across scientific and enterprise domains. However, existing embedding methods face critical limitations when modeling diverse relationship types at scale: Euclidean models struggle with hierarchies, vector space models cannot capture asymmetry, and hyperbolic models fail on symmetric relations. We propose HyperComplEx, a hybrid embedding framework that adaptively combines hyperbolic, complex, and Euclidean spaces via learned attention mechanisms. A relation-specific space weighting strategy dynamically selects optimal geometries for each relation type, while a multi-space consistency loss ensures coherent predictions across spaces. We evaluate HyperComplEx on computer science research knowledge graphs ranging from 1K papers (~25K triples) to 10M papers (~45M triples), demonstrating consistent improvements over state-of-the-art baselines including TransE, RotatE, DistMult, ComplEx, SEPA, and UltraE. Additional tests on standard benchmarks confirm significantly higher results than all baselines. On the 10M-paper dataset, HyperComplEx achieves 0.612 MRR, a 4.8% relative gain over the best baseline, while maintaining efficient training, achieving 85 ms inference per triple. The model scales near-linearly with graph size through adaptive dimension allocation. We release our implementation and dataset family to facilitate reproducible research in scalable knowledge graph embeddings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。