arXiv:2512.07332cs.LGcs.AI2025-12

让知识图谱嵌入自适应调整几何形状,提升表示能力。

Local-Curvature-Aware Knowledge Graph Embedding: An Extended Ricci Flow Approach

  • 基于扩展里奇流,让嵌入与局部曲率动态耦合演化。
  • 曲率指数衰减至平坦,距离收敛到全局最优解。
  • 适合处理结构异质性强的知识图谱,如复杂关系网络。

知识图谱嵌入(KGE)依赖嵌入空间的几何结构来编码语义与拓扑关系。现有方法将所有实体置于单一均匀流形(欧氏、球面、双曲或其组合)上,以建模线性、对称或层次模式。然而,预设的均匀流形无法适应真实图谱在局部区域呈现的剧烈曲率变化。由于几何先验固定,与图谱局部曲率不匹配会导致实体间距离失真,降低嵌入表达力。为此,我们提出RicciKGE,使KGE损失梯度与局部曲率在扩展里奇流中耦合,令实体嵌入与底层流形几何共同动态演化,实现相互适应。理论上,当耦合系数有界且合理选取时,我们严格证明:i)所有边的曲率指数衰减,即流形被驱动趋向欧氏平坦;ii)KGE距离严格收敛至全局最优,表明几何平坦化与嵌入优化相互促进。在链接预测与节点分类基准上的实验结果验证了RicciKGE在适应异构知识图谱结构方面的有效性。

原文摘要 · Abstract (English)

Knowledge graph embedding (KGE) relies on the geometry of the embedding space to encode semantic and structural relations. Existing methods place all entities on one homogeneous manifold, Euclidean, spherical, hyperbolic, or their product/multi-curvature variants, to model linear, symmetric, or hierarchical patterns. Yet a predefined, homogeneous manifold cannot accommodate the sharply varying curvature that real-world graphs exhibit across local regions. Since this geometry is imposed a priori, any mismatch with the knowledge graph's local curvatures will distort distances between entities and hurt the expressiveness of the resulting KGE. To rectify this, we propose RicciKGE to have the KGE loss gradient coupled with local curvatures in an extended Ricci flow such that entity embeddings co-evolve dynamically with the underlying manifold geometry towards mutual adaptation. Theoretically, when the coupling coefficient is bounded and properly selected, we rigorously prove that i) all the edge-wise curvatures decay exponentially, meaning that the manifold is driven toward the Euclidean flatness; and ii) the KGE distances strictly converge to a global optimum, which indicates that geometric flattening and embedding optimization are promoting each other. Experimental improvements on link prediction and node classification benchmarks demonstrate RicciKGE's effectiveness in adapting to heterogeneous knowledge graph structures.

知识图谱嵌入模型几何学习里奇流

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