Flow-Modulated Scoring for Semantic-Aware Knowledge Graph Completion
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 61 of its 95 references that are in the map. First posted on arXiv 29 Jun 2025.
Builds on
61 of 95 references.
References · 61
- 2017Attention Is All You Needcited by 60,361
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2017Inductive Representation Learning on Large Graphscited by 6,320
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2018Neural Ordinary Differential Equationscited by 3,779
- 2017Modeling Relational Data with Graph Convolutional Networkscited by 1,694
- 2020DiffWave: A Versatile Diffusion Model for Audio Synthesiscited by 1,354
- 2022Diffusion-LM Improves Controllable Text Generationcited by 1,027
- 2022Building Normalizing Flows with Stochastic Interpolantscited by 995
- 2014Embedding Entities and Relations for Learning and Inference in Knowledge Basescited by 932
- 2016Complex Embeddings for Simple Link Predictioncited by 902
38 more
- 2019RotatE: Knowledge Graph Embedding by Relational Rotation in Complex Spacecited by 898
- 2023Improving and generalizing flow-based generative models with minibatch optimal transportcited by 879
- 2024SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformerscited by 737
- 2017Convolutional 2D Knowledge Graph Embeddingscited by 730
- 2023Unifying Large Language Models and Knowledge Graphs: A Roadmapcited by 668
- 2020A Survey on Knowledge Graphs: Representation, Acquisition and Applicationscited by 643
- 2020WaveGrad: Estimating Gradients for Waveform Generationcited by 622
- 2020Knowledge Graphscited by 621
- 2021CSDI: Conditional Score-based Diffusion Models for Probabilistic Time Series Imputationcited by 582
- 2025Mean Flows for One-step Generative Modelingcited by 561
- 2021SRDiff: Single Image Super-Resolution with Diffusion Probabilistic Modelscited by 415
- 2019KG-BERT: BERT for Knowledge Graph Completioncited by 324
- 2024Discrete Flow Matchingcited by 317
- 2022Rectified Flow: A Marginal Preserving Approach to Optimal Transportcited by 283
- 2023Multisample Flow Matching: Straightening Flows with Minibatch Couplingscited by 278
- 2021Neural Bellman-Ford Networks: A General Graph Neural Network Framework for Link Predictioncited by 275
- 2017DeepPath: A Reinforcement Learning Method for Knowledge Graph Reasoningcited by 266
- 2023Flow Matching on General Geometriescited by 265
- 2019Inductive Relation Prediction by Subgraph Reasoningcited by 239
- 2018SimplE Embedding for Link Prediction in Knowledge Graphscited by 195
- 2019Quaternion Knowledge Graph Embeddingscited by 175
- 2019DRUM: End-To-End Differentiable Rule Mining On Knowledge Graphscited by 175
- 2019Learning Hierarchy-Aware Knowledge Graph Embeddings for Link Predictioncited by 133
- 2020RNNLogic: Learning Logic Rules for Reasoning on Knowledge Graphscited by 118
- 2022A Survey of Knowledge Graph Reasoning on Graph Types: Static, Dynamic, and Multimodalcited by 106
- 2020BoxE: A Box Embedding Model for Knowledge Base Completioncited by 105
- 2023LLMs for Knowledge Graph Construction and Reasoning: Recent Capabilities and Future Opportunitiescited by 97
- 2021Knowledge Graph Reasoning with Relational Digraphcited by 92
- 2022SimKGC: Simple Contrastive Knowledge Graph Completion with Pre-trained Language Modelscited by 91
- 2022Multi-Modal Knowledge Graph Construction and Application: A Surveycited by 74
- 2020Relational Message Passing for Knowledge Graph Completioncited by 61
- 2023Making Large Language Models Perform Better in Knowledge Graph Completioncited by 48
- 2024Paths-over-Graph: Knowledge Graph Empowered Large Language Model Reasoningcited by 44
- 2022Knowledge Graph Embedding: A Survey from the Perspective of Representation Spacescited by 41
- 2022AdaProp: Learning Adaptive Propagation for Graph Neural Network based Knowledge Graph Reasoningcited by 28
- 2021Scientific Language Models for Biomedical Knowledge Base Completion: An Empirical Studycited by 23
- 2023A Universal Question-Answering Platform for Knowledge Graphscited by 17
- 2023Data Scarcity in Recommendation Systems: A Surveycited by 17
50 most cited of 61 All 61 (JSON)
Built on by
Citing papers per year: 2025: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2024MUSE: Integrating Multi-Knowledge for Knowledge Graph Completion
- 2024MUSE: Multi-Knowledge Passing on the Edges, Boosting Knowledge Graph Completion
- 2026RADD: Retrieval-Augmented Discrete Diffusion for Multi-Modal Knowledge Graph Completion
- 2026PEARL: Path-Entity Aligned Relational Learning with Contextual Subgraphs for Inductive Knowledge Graph Completion
- 2024Retrieval, Reasoning, Re-ranking: A Context-Enriched Framework for Knowledge Graph Completion
- 2025Enhancing Knowledge Graph Completion with Entity Neighborhood and Relation Context
- 2024Context-aware Inductive Knowledge Graph Completion with Latent Type Constraints and Subgraph Reasoning
- 2025RECIPE-TKG: From Sparse History to Structured Reasoning for LLM-based Temporal Knowledge Graph Completion