Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Temporal Knowledge Graph Forecasting Without Knowledge Using In-Context Learning

About

Temporal knowledge graph (TKG) forecasting benchmarks challenge models to predict future facts using knowledge of past facts. In this paper, we apply large language models (LLMs) to these benchmarks using in-context learning (ICL). We investigate whether and to what extent LLMs can be used for TKG forecasting, especially without any fine-tuning or explicit modules for capturing structural and temporal information. For our experiments, we present a framework that converts relevant historical facts into prompts and generates ranked predictions using token probabilities. Surprisingly, we observe that LLMs, out-of-the-box, perform on par with state-of-the-art TKG models carefully designed and trained for TKG forecasting. Our extensive evaluation presents performances across several models and datasets with different characteristics, compares alternative heuristics for preparing contextual information, and contrasts to prominent TKG methods and simple frequency and recency baselines. We also discover that using numerical indices instead of entity/relation names, i.e., hiding semantic information, does not significantly affect the performance ($\pm$0.4\% Hit@1). This shows that prior semantic knowledge is unnecessary; instead, LLMs can leverage the existing patterns in the context to achieve such performance. Our analysis also reveals that ICL enables LLMs to learn irregular patterns from the historical context, going beyond simple predictions based on common or recent information.

Dong-Ho Lee, Kian Ahrabian, Woojeong Jin, Fred Morstatter, Jay Pujara• 2023

Related benchmarks

TaskDatasetResultRank
Temporal Knowledge Graph reasoningICEWS18 (test)
Hits@121.07
79
Temporal Knowledge Graph reasoningICEWS14 (test)
Hits@129.5
59
Temporal Knowledge Graph reasoningICEWS05-15 (test)
Hits@136
41
Temporal Knowledge Graph CompletionICEWS14 v1 (test)
MRR0.4327
29
Temporal Knowledge Graph reasoningYAGO (test)
Hits@10.766
27
Temporal Knowledge Graph PredictionICEWS05-15 (test)
Hits@137.25
20
Temporal Knowledge Graph ForecastingICEWS18 (test)
Hits@10.177
16
Temporal Knowledge Graph ForecastingICEWS 14 (test)
Hits@130.7
9
Showing 8 of 8 rows

Other info

Follow for update