Saved in:
Bibliographic Details
Main Authors: Mirtaheri, Mehrnoosh, Rossi, Ryan A., Kim, Sungchul, Mahadik, Kanak, Yu, Tong, Chen, Xiang, Rostami, Mohammad
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2507.18977
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909705162457088
author Mirtaheri, Mehrnoosh
Rossi, Ryan A.
Kim, Sungchul
Mahadik, Kanak
Yu, Tong
Chen, Xiang
Rostami, Mohammad
author_facet Mirtaheri, Mehrnoosh
Rossi, Ryan A.
Kim, Sungchul
Mahadik, Kanak
Yu, Tong
Chen, Xiang
Rostami, Mohammad
contents 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
format Preprint
id arxiv_https___arxiv_org_abs_2507_18977
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Towards Improving Long-Tail Entity Predictions in Temporal Knowledge Graphs through Global Similarity and Weighted Sampling
Mirtaheri, Mehrnoosh
Rossi, Ryan A.
Kim, Sungchul
Mahadik, Kanak
Yu, Tong
Chen, Xiang
Rostami, Mohammad
Artificial Intelligence
Machine Learning
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
title Towards Improving Long-Tail Entity Predictions in Temporal Knowledge Graphs through Global Similarity and Weighted Sampling
topic Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2507.18977