arXiv:2501.11911cs.IR2025-01被引 9

将时序图学习融入大模型,提升时序知识图谱预测能力

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

  • 引入时序图学习捕捉事件的时间与关系模式
  • 在三个数据集上优于当前最优方法,显著提升预测精度
  • 适合研究时序推理与多模态对齐的AI从业者

时序知识图谱预测(TKGF)旨在基于历史观测事件预测未来事件。近年来,大语言模型(LLMs)展现出强大能力,引发对其在时序知识图谱(TKGs)上推理应用的研究热潮。现有基于LLM的方法通常将检索到的历史事实或静态图表示融入大模型,但受限于时序模式建模不足以及图与语言间跨模态对齐效果差,难以充分捕捉TKG中的时序与结构信息。为此,本文提出新框架TGL-LLM,将时序图学习集成至基于大模型的时序知识图谱模型中。具体地,引入时序图学习以捕获时序与关系模式,并生成历史图嵌入;设计混合图标记化方法,在大模型内充分建模时序模式;采用两阶段训练范式,在高质量多样数据上微调大模型,实现图与语言更好对齐。在三个真实数据集上的大量实验表明,该方法超越多种当前最优(SOTA)方法。

原文摘要 · Abstract (English)

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.

时序图谱大模型图学习多模态对齐

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。