Evolving Beyond Snapshots: Harmonizing Structure and Sequence via Entity State Tuning for Temporal Knowledge Graph Forecasting
Siyuan Li, Yunjia Wu, Yiyong Xiao, Pingyang Huang, Peize Li, Ruitong Liu, Yan Wen, Te Sun
Abstract
Temporal knowledge graph (TKG) forecasting requires predicting future facts by jointly modeling structural dependencies within each snapshot and temporal evolution across snapshots. However, most existing methods are stateless: they recompute entity representations at each timestamp from a limited query window, leading to episodic amnesia and rapid decay of long-term dependencies. To address this limitation, we propose Entity State Tuning (EST), an encoder-agnostic framework that endows TKG forecasters with persistent and continuously evolving entity states. EST maintains a global state buffer and progressively aligns structural evidence with sequential signals via a closed-loop design. Specifically, a topology-aware state perceiver first injects entity-state priors into structural encoding. Then, a unified temporal context module aggregates the state-enhanced events with a pluggable sequence backbone. Subsequently, a dual-track evolution mechanism writes the updated context back to the global entity state memory, balancing plasticity against stability. Experiments on multiple benchmarks show that EST consistently improves diverse backbones and achieves state-of-the-art performance, highlighting the importance of state persistence for long-horizon TKG forecasting.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on11
- Recurrent Event Network: Autoregressive Structure Inferenceover Temporal Knowledge GraphsWoojeong Jin, Meng Qu, Xisen Jin, Xiang RenEMNLP 2020 · 353 citations
- Temporal Knowledge Graph Reasoning Based on Evolutional Representation LearningZixuan Li, Xiaolong Jin, Wei Li, Saiping Guan et al.SIGIR 2021 · 345 citations
- Learning from History: Modeling Temporal Knowledge Graphs with Sequential Copy-Generation NetworksCunchao Zhu, Muhao Chen, Changjun Fan, Guangquan Cheng et al.AAAI 2021 · 343 citations
- Knowledge Graph Prompting for Multi-Document Question AnsweringYu Wang, Nedim Lipka, Ryan A. Rossi, Alexa F. Siu et al.AAAI 2024 · 290 citations
- Explainable Subgraph Reasoning for Forecasting on Temporal Knowledge GraphsZhen Han, Peng Chen, Yunpu Ma, Volker TrespICLR 2021 · 213 citations
Related papers
- Graph Hawkes Transformer for Extrapolated Reasoning on Temporal Knowledge GraphsHaohai Sun, Shangyi Geng, Jialun Zhong, Han Hu et al.EMNLP 2022 · 46 citations
- Transformer-based Reasoning for Learning Evolutionary Chain of Events on Temporal Knowledge GraphZhiyu Fang, Shuai-Long Lei, Xiaobin Zhu, Chun Yang et al.SIGIR 2024 · 17 citations
- TimeTraveler: Reinforcement Learning for Temporal Knowledge Graph ForecastingHaohai Sun, Jialun Zhong, Yunpu Ma, Zhen Han et al.EMNLP 2021 · 164 citations
- Dual History Enhancement with Hybrid Hypergraph-Graph Networks for Temporal Knowledge Graph ReasoningKailun Ye, Xiangjie Kong, Yuchao Zhang, Xuan Wang et al.WWW 2026
- TeMP: Temporal Message Passing for Temporal Knowledge Graph CompletionJiapeng Wu, Meng Cao, Jackie Chi Kit Cheung, William L. HamiltonEMNLP 2020 · 137 citations
