LLM-Guided Knowledge Distillation for Temporal Knowledge Graph Reasoning
Authors: Wang Xing, Wei Song, Siyu Lin, Chen Wu, Man Wang
Organizations: School of Computer Science and Technology, Xidian University · School of Computing and Artificial Intelligence, Southwest Jiaotong University · School of Information Science and Engineering, Chongqing Jiaotong University
Temporal knowledge graphs (TKGs) support reasoning over time-evolving facts, yet state-of-the-art models are often computationally heavy and costly to deploy. Existing compression and distillation techniques are largely designed for static graphs; directly applying them to temporal settings may overlook time-dependent interactions and lead to performance degradation. We propose an LLM-assisted distillation framework specifically designed for temporal knowledge graph reasoning. Beyond a conventional high-capacity temporal teacher, we incorporate a large language model as an auxiliary instructor to provide enriched supervision. The LLM supplies broad background knowledge and temporally informed signals, enabling a lightweight student to better model event dynamics without increasing inference-time complexity. Training is conducted by jointly optimizing supervised and distillation objectives, using a staged alignment strategy to progressively integrate guidance from both teachers. Extensive experiments on multiple public TKG benchmarks with diverse backbone architectures demonstrate that the proposed approach consistently improves link prediction performance over strong distillation baselines, while maintaining a compact and efficient student model. The results highlight the potential of large language models as effective teachers for transferring temporal reasoning capability to resource-efficient TKG systems.
Figures & tables
Figure 1: Comparison between different distillation methods.
Figure 2: The training and prediction on quadruple knowledge with the large language model.
Dataset
Entities
Relations
Train
Valid
Test
YAGO
10,623
10
161,540
19,523
20,026
WIKI
12,544
24
539,286
67,538
63,110
Table 1: Statistical data of two datasets
Model
Encoder
Decoder
TTransE
–
∥s+p−o+t∥2
TADistMult
pseq=LSTM(p:t)
(s∘o)pseq⊤
Table 2: Introduction of classic TKG models
YAGO
WIKI
Model
Method
MRR
MR
Hits@1
Hits@3
Hits@10
MRR
MR
Hits@1
Hits@3
Hits@10
TTransE
BKD
7.65
1410.12
3.50
7.83
15.61
7.94
2383.67
4.75
8.22
14.04
FitNet
7.59
1201.69
3.06
7.18
16.48
7.86
2148.86
3.93
7.78
14.67
RKD
7.01
1186.27
3.56
6.95
13.47
7.89
2052.37
4.72
7.49
12.85
Ours
7.69
1193.15
3.61
7.89
16.57
7.92
1985.63
4.86
8.36
14.94
TADistMult
BKD
61.90
973.89
58.51
64.13
67.59
45.89
3150.11
42.46
48.87
51.18
Table 3: Distillation results of different model method combinations on two datasets
Artificial Intelligence Research Center, Chang Gung University · 2Graduate Institute of Oral Biology, National Taiwan University · 3Electrical Engineering, National Taiwan University of Science and Technology