DynaGen: Unifying Temporal Knowledge Graph Reasoning with Dynamic Subgraphs and Generative Regularization
By: Jiawei Shen , Jia Zhu , Hanghui Guo and more
Temporal Knowledge Graph Reasoning (TKGR) aims to complete missing factual elements along the timeline. Depending on the temporal position of the query, the task is categorized into interpolation and extrapolation. Existing interpolation methods typically embed temporal information into individual facts to complete missing historical knowledge, while extrapolation techniques often leverage sequence models over graph snapshots to identify recurring patterns for future event prediction. These methods face two critical challenges: limited contextual modeling in interpolation and cognitive generalization bias in extrapolation. To address these, we propose a unified method for TKGR, dubbed DynaGen. For interpolation, DynaGen dynamically constructs entity-centric subgraphs and processes them with a synergistic dual-branch GNN encoder to capture evolving structural context. For extrapolation, it applies a conditional diffusion process, which forces the model to learn underlying evolutionary principles rather than just superficial patterns, enhancing its ability to predict unseen future events. Extensive experiments on six benchmark datasets show DynaGen achieves state-of-the-art performance. On average, compared to the second-best models, DynaGen improves the Mean Reciprocal Rank (MRR) score by 2.61 points for interpolation and 1.45 points for extrapolation.
Similar Papers
A Generative Adaptive Replay Continual Learning Model for Temporal Knowledge Graph Reasoning
Information Retrieval
Helps computers remember old facts without forgetting new ones.
Task-Aware Retrieval Augmentation for Dynamic Recommendation
Information Retrieval
Improves online suggestions by learning changing tastes.
Temporal Reasoning with Large Language Models Augmented by Evolving Knowledge Graphs
Machine Learning (CS)
Helps AI understand changing facts over time.