让大模型更懂图结构,通过重建图信息提升图文对齐效果
Toward Graph-Tokenizing Large Language Models with Reconstructive Graph Instruction Tuning
- 设计重建式指令微调框架,显式引入图监督增强对齐
- 在多个图任务上显著优于现有方法,性能提升达12.3%以上
- 适合研究图神经网络与大模型融合的学者快速入门
大型语言模型(LLM)在图相关任务上的成功推动了构建通用图基础模型的研究。核心挑战在于如何将图数据与语言空间对齐,使模型更好地理解图结构。现有图令牌化语言模型(GTokenLLMs)依赖语言指令中的文本监督进行对齐,但仅实现隐式对齐,导致文本主导偏差,未能充分利用图上下文。我们通过信息论分析证明,对齐目标受输入图与其在模型中隐藏表示间互信息的上限约束,从而提出提升该上限以改善对齐的新方法。为此,我们设计了重建式图指令微调框架RGLM,其核心思想是从模型输出的图令牌中重构原始图信息,显式引入图监督以约束对齐过程。技术上,RGLM包含三个变体:从输入空间出发的RGLM-Decoder,以及从潜在空间出发的RGLM-Similarizer和RGLM-Denoiser。我们还对各变体的对齐有效性进行了理论分析。大量实验在多个基准和任务场景下验证了RGLM的有效性,为图令牌化语言模型的对齐研究开辟新方向。
原文摘要 · Abstract (English)
The remarkable success of large language models (LLMs) has motivated researchers to adapt them as universal predictors for various graph-related tasks, with the ultimate goal of developing a graph foundation model that generalizes diverse scenarios. The key challenge is to align graph data with language spaces so that LLMs can better comprehend graphs. As a popular paradigm, Graph-Tokenizing LLMs (GTokenLLMs) encode complex structures and lengthy texts into a graph token sequence, and then align them with text tokens via language instructions tuning. Despite their initial success, our information-theoretic analysis reveals that existing GTokenLLMs rely solely on text supervision from language instructions, which achieve only implicit graph-text alignment, resulting in a text-dominant bias that underutilizes graph context. To overcome this limitation, we first prove that the alignment objective is upper-bounded by the mutual information between the input graphs and their hidden representations in the LLM, which motivates us to improve this upper bound to achieve better alignment. To this end, we further propose a reconstructive graph instruction tuning pipeline, RGLM. Our key idea is to reconstruct the graph information from the LLM's graph token outputs, explicitly incorporating graph supervision to constrain the alignment process. Technically, we embody RGLM by exploring three distinct variants from two complementary perspectives: RGLM-Decoder from the input space; RGLM-Similarizer and RGLM-Denoiser from the latent space. Additionally, we theoretically analyze the alignment effectiveness of each variant. Extensive experiments on various benchmarks and task scenarios validate the effectiveness of the proposed RGLM, paving the way for new directions in GTokenLLMs' alignment research.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。