Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Language is All a Graph Needs

About

The emergence of large-scale pre-trained language models has revolutionized various AI research domains. Transformers-based Large Language Models (LLMs) have gradually replaced CNNs and RNNs to unify fields of computer vision and natural language processing. Compared with independent data samples such as images, videos or texts, graphs usually contain rich structural and relational information. Meanwhile, language, especially natural language, being one of the most expressive mediums, excels in describing complex structures. However, existing work on incorporating graph problems into the generative language modeling framework remains very limited. Considering the rising prominence of LLMs, it becomes essential to explore whether LLMs can also replace GNNs as the foundation model for graphs. In this paper, we propose InstructGLM (Instruction-finetuned Graph Language Model) with highly scalable prompts based on natural language instructions. We use natural language to describe multi-scale geometric structure of the graph and then instruction finetune an LLM to perform graph tasks, which enables Generative Graph Learning. Our method surpasses all GNN baselines on ogbn-arxiv, Cora and PubMed datasets, underscoring its effectiveness and sheds light on generative LLMs as new foundation model for graph machine learning. Our code is open-sourced at https://github.com/agiresearch/InstructGLM.

Ruosong Ye, Caiqi Zhang, Runhui Wang, Shuyuan Xu, Yongfeng Zhang• 2023

Related benchmarks

TaskDatasetResultRank
Link PredictionCora (test)--
69
Fraud DetectionDigitalMusic
Macro F1 Score70.39
15
Fraud DetectionInstantVideo
Macro F1 Score65.29
15
Fraud DetectionPromotionAbuse
Macro F1 Score54.61
15
Graph ReasoningGRbench
E-COMMERCE (R-L) Score17.52
9
Link PredictionCloth Amazon-Product (test)
Hit@176.23
5
Link PredictionAmazon-Product Home (test)
Hit@179.82
5
Node ClassificationPubmed GLBench (supervised)
Accuracy71.26
5
Link PredictionAmazon-Product Sport (test)
Hit@162.5
5
Link Predictionogbn-arxiv (test)
Accuracy0.757
5
Showing 10 of 13 rows

Other info

Follow for update