Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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

About

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.

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

Related benchmarks

TaskDatasetResultRank
Node ClassificationInstagram
Accuracy67.56
140
Node ClassificationFreebase
Macro F159.62
94
Question AnsweringWebQSP--
59
Graph Question AnsweringExplaGraphs
Accuracy85.39
43
Homogeneous Node ClassificationCora
Micro F186.83
26
Homogeneous Node ClassificationarXiv
Micro-F167.8
26
Heterogeneous Node ClassificationIMDB
Micro-F176.57
24
Link PredictionSports
Recall@53.89
15
Link PredictionBaby
Recall@105.09
15
Link PredictionBaby
Recall@53.13
15
Showing 10 of 14 rows

Other info

Follow for update