ICLR 2023

Label-free Node Classification on Graphs with Large Language Models (LLMs)

Zhikai Chen, Haitao Mao, Hongzhi Wen, Haoyu Han, Wei-dong Jin, Haiyang Zhang, Hui Liu, Jiliang Tang

Main figure for Label-free Node Classification on Graphs with Large Language Models (LLMs)

Abstract

In recent years, there have been remarkable advancements in node classification achieved by Graph Neural Networks (GNNs). However, they necessitate abundant high-quality labels to ensure promising performance. In contrast, Large Language Models (LLMs) exhibit impressive zero-shot proficiency on text-attributed graphs. Yet, they face challenges in efficiently processing structural data and suffer from high inference costs. In light of these observations, this work introduces a label-free node classification on graphs with LLMs pipeline, LLM-GNN. It amalgamates the strengths of both GNNs and LLMs while mitigating their limitations. Specifically, LLMs are leveraged to annotate a small portion of nodes and then GNNs are trained on LLMs' annotations to make predictions for the remaining large portion of nodes. The implementation of LLM-GNN faces a unique challenge: how can we actively select nodes for LLMs to annotate and consequently enhance the GNN training? How can we leverage LLMs to obtain annotations of high quality, representativeness, and diversity, thereby enhancing GNN performance with less cost? To tackle this challenge, we develop an annotation quality heuristic and leverage the confidence scores derived from LLMs to advanced node selection. Comprehensive experimental results validate the effectiveness of LLM-GNN on text-attributed graphs from various domains. In particular, LLM-GNN can achieve an accuracy of 74.9\

Benchmark Results

Tested on 2 datasets

Tables show presented results in global context (best 3 papers + GCN + MLP).

Classic

#ModelAccuracy ↑Paper
1
GCN
0.9302 ±0.0011
2
RSE + RGCN (70b)
0.8852 ±0.0019
3
LEMP+T
0.8825
9
SAGE
0.8771
237
MLP
0.7469 ±0.0085
244
Hybrid (One-shot)
0.7367 ±0.0262
This paper
268
PS-Pagerank-W
0.6986 ±0.0029
This paper
269
PS-RIM-W
0.6984 ±0.0019
This paper
270
Pagerank-W
0.6961 ±0.0034
This paper
272
DA-GraphPart
0.6872 ±0.0101
This paper
273
GraphPart-W
0.6843 ±0.0092
This paper
275
DA-RIM-W
0.6822 ±0.0025
This paper
276
DA-Pagerank-W
0.6819 ±0.0032
This paper
277
DA-RIM
0.6782 ±0.0032
This paper
280
GraphPart
0.6728 ±0.0087
This paper
281
PS-Degree-W
0.6721 ±0.0029
This paper
282
DA-Pagerank
0.6715 ±0.0044
This paper
284
DA-GraphPart-W
0.6713 ±0.0123
This paper
285
PageRank
0.6713 ±0.0046
This paper
286
PS-GraphPart-W
0.6690 ±0.0105
This paper
288
RIM
0.6672 ±0.0016
This paper
289
PS-FeatProp
0.6609 ±0.0035
This paper
291
Degree-W
0.6561 ±0.0031
This paper
292
PS-Density-W
0.6557 ±0.0045
This paper
293
Degree
0.6538 ±0.0035
This paper
294
Density
0.6496 ±0.0053
This paper
295
PS-FeatProp-W
0.6472 ±0.0019
This paper
296
FeatProp
0.6408 ±0.0012
This paper
297
Density-W
0.6380 ±0.0069
This paper
298
FeatProp-W
0.6380 ±0.0021
This paper
299
DA-Degree
0.6340 ±0.0051
This paper
300
DA-Density-W
0.6326 ±0.0062
This paper
301
DA-Degree-W
0.6314 ±0.0055
This paper
302
DA-Density
0.6308 ±0.0045
This paper
303
PS-Random-W
0.6260 ±0.0094
This paper
304
Random-W
0.6142 ±0.0154
This paper
305
Random
0.6069 ±0.0173
This paper
307
DA-AGE
0.5939 ±0.0021
This paper
308
DA-AGE-W
0.5919 ±0.0024
This paper
309
C-Density
0.5777 ±0.0085
This paper
310
PS-AGE-W
0.5613 ±0.0011
This paper
311
AGE
0.5551 ±0.0012
This paper
313
AGE-W
0.5515 ±0.0014
This paper

OGB

#ModelAccuracy ↑Paper
1
Jacobi
0.8968
2
GPR
0.8953
3
LEMP4HG
0.8939
4
SAGE
0.8933
205
one-shot
0.7867 ±0.0450
This paper
250
Hybrid (One-shot)
0.7567 ±0.0613
This paper
260
Zero-shot
0.7533 ±0.0499
This paper
261
Most Voting
0.7533 ±0.0499
This paper
272
PS-FeatProp
0.7491 ±0.0017
This paper
286
PS-FeatProp-W
0.7454 ±0.0024
This paper
290
FeatProp-W
0.7432 ±0.0014
This paper
296
FeatProp
0.7404 ±0.0015
This paper
299
topk
0.7400 ±0.0510
This paper
318
Random-W
0.7196 ±0.0059
This paper
319
Degree-W
0.7196 ±0.0023
This paper
322
PS-Random-W
0.7162 ±0.0054
This paper
324
PS-Degree-W
0.7157 ±0.0025
This paper
329
Degree
0.7122 ±0.0021
This paper
334
Random
0.7040 ±0.0060
This paper
339
Pagerank-W
0.6975 ±0.0029
This paper
340
PS-Pagerank-W
0.6969 ±0.0045
This paper
342
PageRank
0.6920 ±0.0032
This paper
343
PS-AGE-W
0.6862 ±0.0015
This paper
351
MLP
0.6605 ±0.0020
354
AGE
0.6563 ±0.0015
This paper
358
AGE-W
0.6492 ±0.0019
This paper
373
DA-AGE
0.6003 ±0.0011
This paper
374
DA-AGE-W
0.5995 ±0.0023
This paper
381
DA-Pagerank-W
0.5557 ±0.0024
This paper
384
DA-Pagerank
0.5477 ±0.0036
This paper
386
DA-Degree-W
0.4790 ±0.0045
This paper
387
DA-Degree
0.4441 ±0.0053
This paper
390
Density-W
0.2097 ±0.0015
This paper
391
Density
0.2022 ±0.0011
This paper
392
PS-Density-W
0.1915 ±0.0018
This paper
393
DA-Density-W
0.0852 ±0.0011
This paper
394
DA-Density
0.0850 ±0.0032
This paper
395
C-Density
0.0829
This paper
SOTA performance (not significantly different from the best) Baseline / comparison (this paper) Global context (other papers)