arXiv:2507.05110cs.AI2025-07

提出新框架让知识图谱推理在分布变化时仍稳定有效

Rule Learning for Knowledge Graph Reasoning under Agnostic Distribution Shift

  • 用特征去相关提升规则学习对分布偏移的鲁棒性
  • 在7个基准数据集上验证了跨环境稳定性与优越性能
  • 适合需要真实场景可靠推理的应用,如医疗、金融

逻辑规则学习是知识图谱推理的重要方法,旨在从已知事实中提取显式规则以推断缺失知识。然而,该方法依赖独立同分布(I.I.D.)假设,易受训练阶段选择偏差或测试阶段分布偏移(如查询偏移)影响,导致性能下降。本文首次将此问题形式化为分布外(OOD)知识图谱推理,并提出稳定规则学习(StableRule)框架。该框架结合特征去相关与规则学习网络,缓解分布外场景下的协变量偏移,增强模型泛化能力。在七个基准知识图谱上的大量实验表明,StableRule在多种异构环境下均表现出更强的鲁棒性与有效性,具备实际应用价值。

原文摘要 · Abstract (English)

Logical rule learning, a prominent category of knowledge graph (KG) reasoning methods, constitutes a critical research area aimed at learning explicit rules from observed facts to infer missing knowledge. However, like all KG reasoning methods, rule learning suffers from a critical weakness-its dependence on the I.I.D. assumption. This assumption can easily be violated due to selection bias during training or agnostic distribution shifts during testing (e.g., as in query shift scenarios), ultimately undermining model performance and reliability. To enable robust KG reasoning in wild environments, this study investigates logical rule learning in the presence of agnostic test-time distribution shifts. We formally define this challenge as out-of-distribution (OOD) KG reasoning-a previously underexplored problem, and propose the Stable Rule Learning (StableRule) framework as a solution. StableRule is an end-to-end framework that combines feature decorrelation with rule learning network, to enhance OOD generalization in KG reasoning. By leveraging feature decorrelation, StableRule mitigates the adverse effects of covariate shifts arising in OOD scenarios, improving the robustness of the rule learning network. Extensive experiments on seven benchmark KGs demonstrate the framework's superior effectiveness and stability across diverse heterogeneous environments, highlighting its practical significance for real-world applications.

知识图谱规则学习分布外推理鲁棒性

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