arXiv:2501.00252cs.LGcs.DB2025-01被引 2

提出首个面向时序知识图谱的模式感知数据增强方法,提升长尾实体预测效果。

Towards Pattern-aware Data Augmentation for Temporal Knowledge Graph Completion

  • 基于三元闭包设计分层评分算法,融合全局语义与局部时间结构生成新样本。
  • 发现模型对特定实体存在偏好,该偏好会加剧数据分布不均问题。
  • 通过两阶段训练识别难学样本,避免虚假负样本干扰,适配现有模型使用。

时序知识图谱补全(TKGC)旨在预测缺失事实,其核心挑战是数据分布不均,导致长尾实体和时间戳表现差、训练不稳定。本文首次揭示现有方法存在模型偏好现象:某些属性(如近期活跃)的实体更受模型青睐,引发误差累积并恶化分布不均问题。为此,提出 Booster——首个面向 TKGs 的数据增强策略。该方法基于三元闭包构建分层评分算法,结合全局语义模式与局部时间结构,实现模式感知的新样本验证;同时采用两阶段训练识别偏离模型偏好模式的样本,并通过频率过滤策略避免虚假负样本误导。实验表明,Booster 可无缝集成至现有 TKGC 模型,在多个数据集上最高提升 8.7% 性能。

原文摘要 · Abstract (English)

Predicting missing facts for temporal knowledge graphs (TKGs) is a fundamental task, called temporal knowledge graph completion (TKGC). One key challenge in this task is the imbalance in data distribution, where facts are unevenly spread across entities and timestamps. This imbalance can lead to poor completion performance or long-tail entities and timestamps, and unstable training due to the introduction of false negative samples. Unfortunately, few previous studies have investigated how to mitigate these effects. Moreover, for the first time, we found that existing methods suffer from model preferences, revealing that entities with specific properties (e.g., recently active) are favored by different models. Such preferences will lead to error accumulation and further exacerbate the effects of imbalanced data distribution, but are overlooked by previous studies. To alleviate the impacts of imbalanced data and model preferences, we introduce Booster, the first data augmentation strategy for TKGs. The unique requirements here lie in generating new samples that fit the complex semantic and temporal patterns within TKGs, and identifying hard-learning samples specific to models. Therefore, we propose a hierarchical scoring algorithm based on triadic closures within TKGs. By incorporating both global semantic patterns and local time-aware structures, the algorithm enables pattern-aware validation for new samples. Meanwhile, we propose a two-stage training approach to identify samples that deviate from the model's preferred patterns. With a well-designed frequency-based filtering strategy, this approach also helps to avoid the misleading of false negatives. Experiments justify that Booster can seamlessly adapt to existing TKGC models and achieve up to an 8.7% performance improvement.

知识图谱数据增强时序建模模式感知

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