cs.AI
Generating Graph-like Rules for Knowledge Graph Reasoning via Diffusion Models
May 29, 2026
The paper proposes GRiD, a novel framework that uses a two-phase training strategy (supervised pre-training and RL fine-tuning) to discover complex, graph-like rules for knowledge graph reasoning, overcoming limitations of existing methods.
Logical rules constitute a cornerstone of knowledge graph (KG) reasoning, valued for their interpretability and ability to model relational patterns. However, existing rule mining methods predominantly focus on simple chain-like rules and therefore neglect the richer relational information encoded in graph-like structures, such as cycles and branches. This limitation is further exacerbated by computational bottlenecks caused by the combinatorial explosion of the search space, which is especially challenging for graph-like rules. Meanwhile, generative approaches such as diffusion models, despite their success in other domains, can not be directly applied to rule mining because their training objectives are not aligned with the goal of learning high-quality rules, and non-differentiable KG rule quality metrics cannot directly guide model optimization. To address these limitations, we propose GRiD, a framework that reformulates graph-like rule discovery as a discrete generative process conditioned on the target relation. GRiD employs a two-phase training strategy. First, supervised pre-training enables GRiD to capture structural priors from subgraphs sampled from the KG meta-graph. Subsequently, reinforcement learning is applied to fine-tune GRiD through policy gradient optimization guided directly by non-differentiable rule-quality metrics. Experiments on six benchmark datasets show that GRiD achieves competitive performance on KG completion tasks. Ablation studies confirm the efficiency and robustness of GRiD and further show that graph-like rules complement chain-like rules in KG completion. Our codes and datasets are available in https://github.com/Haoxiang-Cheng/GRiD
LegalGraphRAG: Multi-Agent Graph Retrieval-Augmented Generation for Reliable Legal Reasoning
LegalGraphRAG introduces a multi-agent, hierarchical graph retrieval-augmented g…
Better Later Than Sooner: Neuro-Symbolic Knowledge Graph Construction via Ontology-grounded Post-ext…
The paper proposes a neuro-symbolic framework to construct highly consistent kno…
Evolutionary Refinement of Generative Graph Topologies: A Hybrid WGAN-GA Approach
The paper introduces a hybrid WGAN-GA framework that uses a Genetic Algorithm (G…
Knowledge Graph-Enhanced Zero-Shot Topic Classification: A Multi-Strategy Comparative Study
The paper proposes a zero-shot multi-label topic classification framework and fi…
Graph-Conditioned Mixture of Graph Neural Network Experts for Traffic Forecasting
The paper proposes GC-MoE, a graph-conditioned Mixture of Experts framework, to…
Beyond Trajectory Rewards: Step-level Credit Assignment for Agentic Search via Graph Modeling
The paper introduces Graph-Distance Contribution Reward (GDCR) and Step Advantag…
BIRDNet: Mining and Encoding Boolean Implication Knowledge Graphs as Interpretable Deep Neural Netwo…
BIRDNet is a novel, sparse, and interpretable deep neural network that encodes B…
Diagnosing Live Within-Policy Instruction Conflicts in LLM Agents with Witnessed Resolution Profiles
The paper introduces WIRE, a pipeline for diagnosing live intra-policy rule conf…