arXiv:2605.30747cs.AI2026-05KDD被引 1

用扩散模型生成带环和分支的复杂逻辑规则,提升知识图谱推理能力。

Generating Graph-Like Logical Rules for Knowledge Graph Reasoning via Diffusion Models

论文配图:Generating Graph-Like Logical Rules for Knowledge Graph Reasoning via Diffusion Models
图 1 · 摘自论文原文
  • 将图结构规则生成视为条件生成任务,分两阶段训练:先监督预训练,再强化学习优化
  • 在6个基准数据集上表现优于传统方法,图结构规则能有效补充链式规则
  • 首次将扩散模型用于规则挖掘,适合需要高可解释性的知识推理场景

逻辑规则是知识图谱推理的核心,因其可解释性及对关系模式的建模能力备受重视。然而,现有规则挖掘方法多聚焦于简单的链式规则,忽视了图结构(如环、分支)所蕴含的丰富关系信息。这一局限因搜索空间的组合爆炸而加剧,尤其对图结构规则尤为显著。同时,生成模型如扩散模型虽在其他领域成功应用,却难以直接用于规则挖掘——其训练目标与高质量规则学习不匹配,且知识图谱规则质量指标不可微,无法直接指导优化。为此,我们提出GRiD框架,将图结构规则发现重构为基于目标关系的离散生成过程。GRiD采用两阶段训练策略:首先通过监督预训练,从知识图谱元图中采样的子图学习结构先验;随后利用强化学习,通过策略梯度优化,直接以不可微的规则质量指标引导模型调优。在六个基准数据集上的实验表明,GRiD在知识图谱补全任务中表现具有竞争力。消融实验验证了其高效性与鲁棒性,并进一步显示图结构规则可与链式规则互补。代码与数据集已开源于https://github.com/Haoxiang-Cheng/GRiD。

原文摘要 · Abstract (English)

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, cannot 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 code and datasets are available in https://github.com/Haoxiang-Cheng/GRiD.

知识图谱逻辑规则扩散模型推理增强

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