arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.04780cs.CLcs.AI
收藏

面向通用图挖掘的大语言模型图导向指令微调

Graph-oriented Instruction Tuning of Large Language Models for Generic Graph Mining

Engineering Research Center of Big Data Intelligence, Ministry of Education(教育部大数据智能工程研究中心);Fujian Key Laboratory of Network Computing and Intelligent Information Processing(福建省网络计算与智能信息处理重点实验室);College of Computer and Data Science, Fuzhou University(福州大学计算机与数据科学学院);Department of Computer Science, Emory University(埃默里大学计算机科学系)

Yanchao Tan, Hang Lv, Pengxiang Zhan, Shiping Wang, Carl Yang

AI总结:

提出MuseGraph框架,通过紧凑图描述、基于思维链的指令生成和图感知指令微调,将GNN与LLM结合,实现跨任务和数据集的高效图挖掘。

摘要:

具有丰富属性的图对于建模互联实体和增强各种实际应用中的预测至关重要。传统的图神经网络(GNN)通常需要针对不同的图任务和数据集进行重新训练。尽管大语言模型(LLM)的出现为自然语言处理带来了新范式,但它们在通用图挖掘(即训练单个模型同时处理多样任务和数据集)方面的潜力仍未充分探索。为此,我们的新颖框架MuseGraph无缝地将GNN和LLM的优势整合到一个基础模型中,用于跨任务和数据集的图挖掘。该框架首先采用紧凑的图描述,在语言令牌限制内封装关键图信息。然后,我们提出了一种基于思维链(CoT)指令包的多样化指令生成机制,以从GPT-4等高级LLM中提取推理能力。最后,我们设计了一种图感知的指令微调策略,以促进多个任务和数据集之间的相互增强,同时防止LLM生成能力的灾难性遗忘。我们的实验结果表明,在五个图任务和十个数据集上取得了显著改进,展示了MuseGraph在提高图导向下游任务准确性的同时增强LLM生成能力的潜力。

英文摘要:

Graphs with abundant attributes are essential in modeling interconnected entities and enhancing predictions across various real-world applications. Traditional Graph Neural Networks (GNNs) often require re-training for different graph tasks and datasets. Although the emergence of Large Language Models (LLMs) has introduced new paradigms in natural language processing, their potential for generic graph mining, training a single model to simultaneously handle diverse tasks and datasets, remains under-explored. To this end, our novel framework MuseGraph, seamlessly integrates the strengths of GNNs and LLMs into one foundation model for graph mining across tasks and datasets. This framework first features a compact graph description to encapsulate key graph information within language token limitations. Then, we propose a diverse instruction generation mechanism with Chain-of-Thought (CoT)-based instruction packages to distill the reasoning capabilities from advanced LLMs like GPT-4. Finally, we design a graph-aware instruction tuning strategy to facilitate mutual enhancement across multiple tasks and datasets while preventing catastrophic forgetting of LLMs' generative abilities. Our experimental results demonstrate significant improvements in five graph tasks and ten datasets, showcasing the potential of our MuseGraph in enhancing the accuracy of graph-oriented downstream tasks while improving the generation abilities of LLMs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑