arXiv:2607.03154cs.CLcs.AI2026-07

用扩散模型生成通用实体表示,提升多领域知识图谱补全效果。

Conditional Diffusion Guided Knowledge Transfer for Multi-Domain Knowledge Graph Completion

  • 基于扩散模型生成跨域通用实体嵌入,以支持知识迁移。
  • 在14个知识图谱上实现尾部实体预测平均提升4.3%的MRR。
  • 特别适合低资源场景,兼顾通用性与领域特异性信息保留。

多领域知识图谱补全(MKGC)旨在通过从其他支持知识图谱中迁移知识来提升目标图谱的缺失三元组预测性能。现有方法通常对跨图谱等价实体施加一致性约束以实现知识转移,但这可能抑制实体的领域特异性上下文信息,并损害所有领域中的实体表示能力,尤其在低资源数据场景下影响显著。为此,我们首次提出基于生成的MKGC范式,设计了DMKGC——一种条件扩散引导的知识迁移框架。核心思想是将每个知识图谱视为实体完整信息的部分视图,通过以支持图谱为条件的扩散模型生成具有信息量的通用实体嵌入。具体而言,首先初始化无域依赖的先验实体嵌入,再在各知识图谱中编码;随后融合支持图谱中的等价实体作为扩散生成的条件引导。利用先验嵌入作为代理生成目标,确保生成过程不偏向任一特定图谱;同时训练生成嵌入在跨图谱上的预测能力,从而保留领域特异性信息。在3个基准上的14个知识图谱上进行大量实验表明,相较于现有最优方法,尾部实体预测的平均MRR提升4.3%,且在低资源设置下仍保持持续增益。

原文摘要 · Abstract (English)

Multi-domain knowledge graph completion (MKGC) aims to improve missing triple prediction in a target KG by transferring knowledge from other support KGs. Existing methods typically enforce consistency constraints on equivalent entities across KGs to transfer knowledge, which risks suppressing domain-specific contextual information of entities. This design can also compromise entity representation information from all KG domains, impeding performance improvements, especially in low-resource data scenarios. To address this, we pioneer a generation-based paradigm for MKGC and propose DMKGC, a conditional diffusion-guided knowledge transfer framework. Our key insight is to treat each KG as a partial view of the entity entire information, and generate informative domain-general entity embeddings through diffusion models conditioned on support KGs. Particularly, we first initialize domain-agnostic entity embeddings as prior entity embeddings, and then encode them within individual KGs. Afterward, we fuse equivalent entities from support KGs as the conditional diffusion generation guidance. We leverage the prior entity embeddings as the proxy generation objective, which ensures this conditional generation to be unbiased towards any conditioned KGs. Simultaneously, we also train the generated embeddings to be predictive across KGs, thus preserving domain-specific information. Extensive experiments on 14 KGs in 3 benchmarks demonstrate a 4.3\% average MRR improvement in tail entity prediction over state-of-the-art methods, with sustained gains in low-resource data settings.

知识图谱扩散模型多领域学习生成方法

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