arrow
Return

Diffusion Model with Selective Attention for Temporal Knowledge Graph Reasoning

delete2026-01-01
delete0
PRE
AI
R
Rushan Geng
C
Chen Ge
C
Cuicui Luo *
DOI:10.1007/978-3-032-05981-9_9delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Temporal knowledge graph reasoning aims to predict missing entities at future time steps, and as a critical task, it has attracted widespread attention in recent years due to its impressive ability to capture historical correlations and forecast future events. Although existing approaches, such as graph learning and logic rules, have partially addressed this problem, they still face limitations in modeling the uncertainty of future events especially when predicting rare or unseen facts. To address these challenges, we propose a diffusion model based on a selective attention mechanism (DMSA) for temporal knowledge graph reasoning. In our method, the encoder incorporates selective attention to emphasize key information, while the diffusion module introduces noise to enhance the model's capability to predict unseen events. By integrating selective attention with the diffusion module, our model improves both its memory and its ability to predict future, unseen events. Experimental results on five public datasets demonstrate that our proposed model achieves state-of-the-art performance across multiple evaluation metrics.
Keywords:
Temporal knowledge graph
Temporal knowledge graph reasoning
Diffusion model
Selective attention

Journal

M
MACHINE LEARNING AND KNOWLEDGE DISCOVERY IN DATABASES. RESEARCH TRACK, ECML PKDD 2025, PT II
IF:
0
Papers:
28
Citations:
0

Organization

U
university of chinese academy of sciences, cas
Scholars:
4.1W
Papers: 3.8W
Citations: 75
C
chinese academy of sciences
Scholars:
56.7W
Papers: 45.0W
Citations: 704
Cited Papers

Cited Papers

errShare
errSave
KGAT
err2019-07-25
err0
errOAAI
errXiang Wang; Xiangnan He; Yixin Cao; Meng Liu; Tat-Seng Chua
errShare
errSave
errShare
errSave
Modeling Relational Data with Graph Convolutional Networks
err2018-06-03
err0
PREAI
errMichael Schlichtkrull; Thomas N. Kipf; Peter Bloem; Rianne van den Berg; Ivan Titov; Max Welling
errShare
errSave
Convolutional 2D Knowledge Graph Embeddings
err2018-04-25
err0
errOAAI
errTim Dettmers; Pasquale Minervini; Pontus Stenetorp; Sebastian Riedel
errShare
errSave
Mining Long Short-Term Evolution Patterns for Temporal Knowledge Graph Reasoning
err2025-01-01
err0
PREAI
errWang,Fengyi; Zhu,Guanghui; Hou,Haojun; Yuan,Chunfeng; Huang,Yihua
errShare
errSave
errShare
errSave
researcher View more