arXiv:2505.20948cs.AI2025-05中稿 · ICLR被引 13

让知识图谱推理生成更可控的假说,避免冗余结果。

Controllable Logical Hypothesis Generation for Abductive Reasoning in Knowledge Graphs

  • 分两阶段训练,结合监督与强化学习提升可控性。
  • 在三个数据集上优于基线,控制条件遵守率更高。
  • 适合需要精准推理的医疗、科研场景使用。

知识图谱中的溯因推理旨在从观察到的实体生成合理的逻辑假说,广泛应用于临床诊断和科学发现等领域。然而,由于缺乏可控性,单一观察可能在大规模知识图谱中产生大量冗余或无关的假说。为此,我们提出可控假说生成任务,以提升溯因推理的实际应用价值。该任务面临两大挑战:假说空间坍缩和假说过度敏感。为此,我们提出 CtrlHGen 框架,采用两阶段训练范式(监督学习+强化学习)。为缓解假说空间坍缩,设计基于子逻辑分解的数据增强策略,利用简单组件的语义模式学习复杂逻辑结构;为应对假说过度敏感,引入平滑语义奖励(包括 Dice 与重叠分数)及条件遵循奖励,引导生成满足用户指定控制约束。在三个基准数据集上的实验表明,该模型不仅更严格遵守控制条件,且在语义相似度上优于基线方法。代码已开源:https://github.com/HKUST-KnowComp/CtrlHGen。

原文摘要 · Abstract (English)

Abductive reasoning in knowledge graphs aims to generate plausible logical hypotheses from observed entities, with broad applications in areas such as clinical diagnosis and scientific discovery. However, due to a lack of controllability, a single observation may yield numerous plausible but redundant or irrelevant hypotheses on large-scale knowledge graphs. To address this limitation, we introduce the task of controllable hypothesis generation to improve the practical utility of abductive reasoning. This task faces two key challenges when controlling for generating long and complex logical hypotheses: hypothesis space collapse and hypothesis oversensitivity. To address these challenges, we propose CtrlHGen, a Controllable logcial Hypothesis Generation framework for abductive reasoning over knowledge graphs, trained in a two-stage paradigm including supervised learning and subsequent reinforcement learning. To mitigate hypothesis space collapse, we design a dataset augmentation strategy based on sub-logical decomposition, enabling the model to learn complex logical structures by leveraging semantic patterns in simpler components. To address hypothesis oversensitivity, we incorporate smoothed semantic rewards including Dice and Overlap scores, and introduce a condition-adherence reward to guide the generation toward user-specified control constraints. Extensive experiments on three benchmark datasets demonstrate that our model not only better adheres to control conditions but also achieves superior semantic similarity performance compared to baselines. Our code is available at https://github.com/HKUST-KnowComp/CtrlHGen.

知识图谱逻辑推理可控生成

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