arXiv:2507.18977cs.AIcs.LG2025-07

提升时序知识图谱中长尾实体预测性能,通过全局相似度与加权采样。

Towards Improving Long-Tail Entity Predictions in Temporal Knowledge Graphs through Global Similarity and Weighted Sampling

  • 引入全局实体相似度与加权采样策略,增强模型对新实体的适应能力。
  • 在两个基准数据集上实现10%的总链接预测提升,长尾实体MRR提升15%。
  • 适用于增量训练场景,尤其适合处理稀疏连接的新实体预测。

时序知识图谱(TKG)补全模型通常假设训练时可访问完整图结构,忽略了其动态演化带来的挑战:(i) 模型需泛化并吸收新知识,(ii) 处理新出现或未见实体的稀疏连接问题。本文提出一种专为TKG设计的增量训练框架,针对训练中未见或连接稀疏的实体进行优化。该方法结合模型无关的增强层与加权采样策略,可嵌入任意现有TKG补全模型。增强层基于更广泛的全局实体相似度,突破图神经网络仅依赖局部邻域的局限。训练时采用加权采样,强化罕见实体相关边的贡献。在两个基准数据集上的实验表明,该框架在整体链接预测、归纳式链接预测及长尾实体预测上均优于现有方法,显著提升10%的总体性能,并使长尾实体的MRR提升15%。结果验证了该方法在缓解灾难性遗忘和增强模型鲁棒性方面的潜力,尤其适用于增量学习环境。

原文摘要 · Abstract (English)

Temporal Knowledge Graph (TKG) completion models traditionally assume access to the entire graph during training. This overlooks challenges stemming from the evolving nature of TKGs, such as: (i) the model's requirement to generalize and assimilate new knowledge, and (ii) the task of managing new or unseen entities that often have sparse connections. In this paper, we present an incremental training framework specifically designed for TKGs, aiming to address entities that are either not observed during training or have sparse connections. Our approach combines a model-agnostic enhancement layer with a weighted sampling strategy, that can be augmented to and improve any existing TKG completion method. The enhancement layer leverages a broader, global definition of entity similarity, which moves beyond mere local neighborhood proximity of GNN-based methods. The weighted sampling strategy employed in training accentuates edges linked to infrequently occurring entities. We evaluate our method on two benchmark datasets, and demonstrate that our framework outperforms existing methods in total link prediction, inductive link prediction, and in addressing long-tail entities. Notably, our method achieves a 10\% improvement and a 15\% boost in MRR for these datasets. The results underscore the potential of our approach in mitigating catastrophic forgetting and enhancing the robustness of TKG completion methods, especially in an incremental training context

时序知识图谱长尾预测增量学习实体相似度

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