arXiv:2410.11370cs.CLcs.IR2024-10被引 4

让大模型更懂图数据,提升零样本推理能力

Enhance Graph Alignment for Large Language Models

  • 用对齐下游任务的模板设计自监督任务,解决语义错位问题
  • 在4个数据集上显著提升有监督学习与零样本性能
  • 适合需要通用图理解能力的研究者和工业应用

图结构数据广泛存在于现实世界中。近年来,由于强大的涌现能力,大型语言模型(LLMs)在建模图数据方面展现出良好前景。有效将图数据应用于LLMs的关键在于将其转化为模型可理解的格式。图转令牌方法通过指令微调将图转换为令牌序列,并与文本令牌对齐,其中自监督指令微调使模型获得图的通用知识,而有监督微调则使其专用于下游图任务。尽管已有初步成功,我们发现现有方法存在自监督任务与下游任务之间的错位,导致自监督微调对下游任务产生负向迁移。为解决此问题,我们提出图对齐大模型(GALLM),通过使用与下游任务对齐的模板实现任务对齐。在自监督微调阶段,引入一种新型文本匹配任务,使用与下游任务一致的模板;在任务特定微调阶段,提出两种类别提示方法,从附加解释中学习监督信息并进一步使用对齐模板。在四个数据集上的实验评估表明,模型在有监督学习、跨数据集泛化能力以及零样本能力方面均有显著提升,凸显其作为图基础模型的潜力。

原文摘要 · Abstract (English)

Graph-structured data is prevalent in the real world. Recently, due to the powerful emergent capabilities, Large Language Models (LLMs) have shown promising performance in modeling graphs. The key to effectively applying LLMs on graphs is converting graph data into a format LLMs can comprehend. Graph-to-token approaches are popular in enabling LLMs to process graph information. They transform graphs into sequences of tokens and align them with text tokens through instruction tuning, where self-supervised instruction tuning helps LLMs acquire general knowledge about graphs, and supervised fine-tuning specializes LLMs for the downstream tasks on graphs. Despite their initial success, we find that existing methods have a misalignment between self-supervised tasks and supervised downstream tasks, resulting in negative transfer from self-supervised fine-tuning to downstream tasks. To address these issues, we propose Graph Alignment Large Language Models (GALLM) to benefit from aligned task templates. In the self-supervised tuning stage, we introduce a novel text matching task using templates aligned with downstream tasks. In the task-specific tuning stage, we propose two category prompt methods that learn supervision information from additional explanation with further aligned templates. Experimental evaluations on four datasets demonstrate substantial improvements in supervised learning, multi-dataset generalizability, and particularly in zero-shot capability, highlighting the model's potential as a graph foundation model.

图神经网络大模型零样本

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