Integrate Temporal Graph Learning into LLM-based Temporal Knowledge Graph Model

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chang, He, Wu, Jie, Tao, Zhulin, Ma, Yunshan, Huang, Xianglin, Chua, Tat-Seng
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913659096137728
author Chang, He
Wu, Jie
Tao, Zhulin
Ma, Yunshan
Huang, Xianglin
Chua, Tat-Seng
author_facet Chang, He
Wu, Jie
Tao, Zhulin
Ma, Yunshan
Huang, Xianglin
Chua, Tat-Seng
contents Temporal Knowledge Graph Forecasting (TKGF) aims to predict future events based on the observed events in history. Recently, Large Language Models (LLMs) have exhibited remarkable capabilities, generating significant research interest in their application for reasoning over temporal knowledge graphs (TKGs). Existing LLM-based methods have integrated retrieved historical facts or static graph representations into LLMs. Despite the notable performance of LLM-based methods, they are limited by the insufficient modeling of temporal patterns and ineffective cross-modal alignment between graph and language, hindering the ability of LLMs to fully grasp the temporal and structural information in TKGs. To tackle these issues, we propose a novel framework TGL-LLM to integrate temporal graph learning into LLM-based temporal knowledge graph model. Specifically, we introduce temporal graph learning to capture the temporal and relational patterns and obtain the historical graph embedding. Furthermore, we design a hybrid graph tokenization to sufficiently model the temporal patterns within LLMs. To achieve better alignment between graph and language, we employ a two-stage training paradigm to finetune LLMs on high-quality and diverse data, thereby resulting in better performance. Extensive experiments on three real-world datasets show that our approach outperforms a range of state-of-the-art (SOTA) methods.
format Preprint
id arxiv_https___arxiv_org_abs_2501_11911
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Integrate Temporal Graph Learning into LLM-based Temporal Knowledge Graph Model
Chang, He
Wu, Jie
Tao, Zhulin
Ma, Yunshan
Huang, Xianglin
Chua, Tat-Seng
Information Retrieval
Temporal Knowledge Graph Forecasting (TKGF) aims to predict future events based on the observed events in history. Recently, Large Language Models (LLMs) have exhibited remarkable capabilities, generating significant research interest in their application for reasoning over temporal knowledge graphs (TKGs). Existing LLM-based methods have integrated retrieved historical facts or static graph representations into LLMs. Despite the notable performance of LLM-based methods, they are limited by the insufficient modeling of temporal patterns and ineffective cross-modal alignment between graph and language, hindering the ability of LLMs to fully grasp the temporal and structural information in TKGs. To tackle these issues, we propose a novel framework TGL-LLM to integrate temporal graph learning into LLM-based temporal knowledge graph model. Specifically, we introduce temporal graph learning to capture the temporal and relational patterns and obtain the historical graph embedding. Furthermore, we design a hybrid graph tokenization to sufficiently model the temporal patterns within LLMs. To achieve better alignment between graph and language, we employ a two-stage training paradigm to finetune LLMs on high-quality and diverse data, thereby resulting in better performance. Extensive experiments on three real-world datasets show that our approach outperforms a range of state-of-the-art (SOTA) methods.
title Integrate Temporal Graph Learning into LLM-based Temporal Knowledge Graph Model
topic Information Retrieval
url https://arxiv.org/abs/2501.11911