English

GRATE: Temporal Extensions for Inductive KG Foundation Models via Gated Rotary Attention

Artificial Intelligence 2026-07-11 v1

Abstract

Knowledge graph foundation models such as Ultra and Trix achieve strong inductive transfer by learning relation-graph representations that generalise to unseen entities and relations. Extending this transferability to temporal knowledge graphs (TKGs) remains challenging: existing temporal models tie their parameters to dataset-specific entities, relations, or timestamps and are not designed to transfer to TKGs with disjoint vocabularies. We propose GRATE (Gated Rotary Attention for Temporal Encoding), an entity-side message function that adds no learnable parameters and encodes time through relative time differences by rotating each edge message according to its time gap to the query and applying a query-conditioned gate to select temporally relevant signals. GRATE integrates into NBFNet-style KG foundation models while preserving structural transferability. Existing TKG benchmarks evaluate within shared train/test vocabularies and cannot directly test cross-dataset temporal transfer; we therefore construct GDELTIndT and WIKIIndT, inductive transfer benchmark suites with disjoint entities, relations, and timestamps spanning both interpolation and extrapolation. Across these benchmarks and held-out forecasting datasets, a single jointly pretrained GRATE checkpoint improves over the static base model in most settings.

Keywords

Cite

@article{arxiv.2607.10197,
  title  = {GRATE: Temporal Extensions for Inductive KG Foundation Models via Gated Rotary Attention},
  author = {Jiaxin Pan and Osama Mohammed and Daniel Hernández and Steffen Staab},
  journal= {arXiv preprint arXiv:2607.10197},
  year   = {2026}
}

Comments

Accepted at the ICML 2026 Workshop on Graph Foundation Models: A New Era for Graph Machine Learning. 17 pages, 4 figures