arXiv:2506.23137cs.CLcs.AI2025-06被引 2

用动态流机制增强知识图谱关系预测,精度接近完美且参数极少。

Flow-Modulated Scoring for Semantic-Aware Knowledge Graph Completion

  • 将关系建模为动态演化过程,结合上下文与条件流匹配。
  • 在FB15k-237上达99.8% MRR和99.7% Hits@1,仅用0.35M参数。
  • 适合需要高精度、小模型的知识图谱任务,如低资源场景。

知识图谱补全需有效建模实体间的多维语义关系。现有方法依赖静态打分函数,难以兼顾丰富语义上下文与关系动态性。为此,本文提出流调制打分(FMS)框架,将关系视为受静态语义环境支配的动态演化过程。FMS分两阶段:首先通过语义上下文学习模块获取上下文感知的实体嵌入,再利用条件流匹配模块建模两者间的动态流,该流动态调制基础静态得分。通过统一丰富的静态表征与条件化动态流,FMS实现对关系语义更全面的理解。大量实验表明,FMS在标准知识图谱补全任务中达到新纪录:在关系预测基准FB15k-237上,仅用0.35M参数即达99.8% MRR和99.7% Hits@1;WN18RR上获99.9% MRR。其优势延伸至实体预测,在归纳设置下相对MRR提升25.2%,显著优于所有基线。通过融合动态流与丰富上下文,FMS提供一种高效、轻量的新范式。代码已公开于https://github.com/yuanwuyuan9/FMS。

原文摘要 · Abstract (English)

Knowledge graph completion demands effective modeling of multifaceted semantic relationships between entities. Yet, prevailing methods, which rely on static scoring functions over learned embeddings, struggling to simultaneously capture rich semantic context and the dynamic nature of relations. To overcome this limitation, we propose the Flow-Modulated Scoring (FMS) framework, conceptualizing a relation as a dynamic evolutionary process governed by its static semantic environment. FMS operates in two stages: it first learns context-aware entity embeddings via a Semantic Context Learning module, and then models a dynamic flow between them using a Conditional Flow-Matching module. This learned flow dynamically modulates a base static score for the entity pair. By unifying context-rich static representations with a conditioned dynamic flow, FMS achieves a more comprehensive understanding of relational semantics. Extensive experiments demonstrate that FMS establishes a new state of the art across both canonical knowledge graph completion tasks: relation prediction and entity prediction. On the standard relation prediction benchmark FB15k-237, FMS achieves a near-perfect MRR of 99.8\% and Hits@1 of 99.7\% using a mere 0.35M parameters, while also attaining a 99.9\% MRR on WN18RR. Its dominance extends to entity prediction, where it secures a 25.2\% relative MRR gain in the transductive setting and substantially outperforms all baselines in challenging inductive settings. By unifying a dynamic flow mechanism with rich static contexts, FMS offers a highly effective and parameter-efficient new paradigm for knowledge graph completion. Code published at: https://github.com/yuanwuyuan9/FMS.

知识图谱动态建模小模型关系预测

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