arXiv:2602.08815cs.AI2026-02Conference of the …被引 3

让扩散模型学会关注反例,提升时序知识图谱预测准确率

Negative-Aware Diffusion Process for Temporal Knowledge Graph Extrapolation

  • 引入负例感知机制,利用负样本原型强化模型判别能力
  • 在四个公开数据集上达到当前最佳性能,显著优于基线方法
  • 适合对时序知识推理、生成模型优化感兴趣的研究人员

时序知识图谱推理旨在从历史证据中预测未来缺失事实。尽管扩散模型因其捕捉复杂预测分布的能力受到关注,但仍存在两个缺陷:(i) 生成路径仅依赖正向证据,忽略了具有信息量的负向上下文;(ii) 训练目标受交叉熵排序主导,虽改善候选排序,但对去噪嵌入的校准监督不足。为此,我们提出负例感知扩散模型用于时序知识图谱外推(NADEx)。NADEx 将实体、关系和时间间隔的主体中心历史编码为序列嵌入,通过前向过程扰动查询对象,并在反向过程中以变压器去噪器重建,条件于时序-关系上下文。我们进一步设计基于批次负样本原型的余弦对齐正则项,收紧对不合理候选的决策边界。在四个公开时序知识图谱基准上的实验证明,NADEx 实现了最先进的性能。

原文摘要 · Abstract (English)

Temporal Knowledge Graph (TKG) reasoning seeks to predict future missing facts from historical evidence. While diffusion models (DM) have recently gained attention for their ability to capture complex predictive distributions, two gaps remain: (i) the generative path is conditioned only on positive evidence, overlooking informative negative context, and (ii) training objectives are dominated by cross-entropy ranking, which improves candidate ordering but provides little supervision over the calibration of the denoised embedding. To bridge this gap, we introduce Negative-Aware Diffusion model for TKG Extrapolation (NADEx). Specifically, NADEx encodes subject-centric histories of entities, relations and temporal intervals into sequential embeddings. NADEx perturbs the query object in the forward process and reconstructs it in reverse with a Transformer denoiser conditioned on the temporal-relational context. We further derive a cosine-alignment regularizer derived from batch-wise negative prototypes, which tightens the decision boundary against implausible candidates. Comprehensive experiments on four public TKG benchmarks demonstrate that NADEx delivers state-of-the-art performance.

时序知识图谱扩散模型负例学习

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