arXiv:2411.01477cs.LGcs.CL2024-11被引 4

用扩散模型生成新事件,提升稀疏历史数据下的时序知识图推理能力

DPCL-Diff: The Temporal Knowledge Graph Reasoning Based on Graph Node Diffusion Model with Dual-Domain Periodic Contrastive Learning

  • 通过图节点扩散生成符合分布的新事件,增强对稀疏数据的推理能力
  • 在4个公开数据集上超越现有最优模型,显著提升预测准确率
  • 结合双域周期对比学习,有效区分周期性与非周期性事件,适合复杂时序预测

时序知识图谱(TKG)推理旨在预测未来缺失的事实,是一项重要且具有挑战性的任务。传统方法依赖于密切相关的历史事实,对重复或周期性事件预测效果较好,但在历史交互稀疏的情况下性能下降。近期扩散模型在图像生成中的成功为TKG推理提供了新思路。为此,本文提出基于图节点扩散模型与双域周期对比学习的DPCL-Diff框架。图节点扩散模型(GNDiff)向稀疏关联事件引入噪声,模拟新事件生成高质量数据,使其更符合实际分布,显著提升对新事件的推理能力。同时,双域周期对比学习(DPCL)将周期性与非周期性事件分别映射到Poincaré空间和欧氏空间,利用其几何特性有效区分相似周期事件。在四个公开数据集上的实验结果表明,DPCL-Diff显著优于当前最先进的TKG模型,验证了该方法的有效性。本研究还深入分析了GNDiff与DPCL在TKG任务中的协同作用。

原文摘要 · Abstract (English)

Temporal knowledge graph (TKG) reasoning that infers future missing facts is an essential and challenging task. Predicting future events typically relies on closely related historical facts, yielding more accurate results for repetitive or periodic events. However, for future events with sparse historical interactions, the effectiveness of this method, which focuses on leveraging high-frequency historical information, diminishes. Recently, the capabilities of diffusion models in image generation have opened new opportunities for TKG reasoning. Therefore, we propose a graph node diffusion model with dual-domain periodic contrastive learning (DPCL-Diff). Graph node diffusion model (GNDiff) introduces noise into sparsely related events to simulate new events, generating high-quality data that better conforms to the actual distribution. This generative mechanism significantly enhances the model's ability to reason about new events. Additionally, the dual-domain periodic contrastive learning (DPCL) maps periodic and non-periodic event entities to Poincaré and Euclidean spaces, leveraging their characteristics to distinguish similar periodic events effectively. Experimental results on four public datasets demonstrate that DPCL-Diff significantly outperforms state-of-the-art TKG models in event prediction, demonstrating our approach's effectiveness. This study also investigates the combined effectiveness of GNDiff and DPCL in TKG tasks.

时序知识图谱扩散模型周期性推理图神经网络

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