notesum.ai

Published at October 22

Large Language Model-based Augmentation for Imbalanced Node Classification on Text-Attributed Graphs

cs.CE

Released Date: October 22, 2024

Authors: Leyao Wang¹, Yu Wang², Bo Ni¹, Yuying Zhao¹, Tyler Derr¹

Aff.: ¹Vanderbilt University; ²University of Oregon

Arxiv: https://arxiv.org/abs/2410.16882v1

Refer to caption

Dataset	Metric	GraphSMOTE	MixupForGraph	GraphENS	LG ${}_{\text{Smote}}$	LG ${}_{\text{Mixup}}$	LG ${}_{\text{Upsampling}}$
Cora	Acc	60.63 $\pm$ 0.50	50.18 $\pm$ 0.28	59.34 $\pm$ 1.10	75.01 $\pm$ 0.23	75.66 $\pm$ 0.23	74.76 $\pm$ 0.42
	F1	61.39 $\pm$ 0.42	47.10 $\pm$ 0.68	57.37 $\pm$ 1.29	73.42 $\pm$ 0.16	74.30 $\pm$ 0.29	73.13 $\pm$ 0.50
	Diff	17.19	43.00	20.30	11.30	13.25	19.76
Pubmed	Acc	67.98 $\pm$ 1.47	68.23 $\pm$ 18.36	70.26 $\pm$ 0.16	75.87 $\pm$ 0.40	74.18 $\pm$ 0.18	73.37 $\pm$ 0.26
	F1	67.19 $\pm$ 1.80	66.50 $\pm$ 19.37	70.16 $\pm$ 0.17	76.20 $\pm$ 0.33	74.35 $\pm$ 0.22	73.43 $\pm$ 0.33
	Diff	36.01	44.00	15.09	10.83	12.96	19.66
Photo	Acc	OOM	24.13 $\pm$ 0.54	27.5 $\pm$ 0.83	59.22 $\pm$ 0.65	58.60 $\pm$ 0.45	66.17 $\pm$ 0.69
	F1	OOM	26.59 $\pm$ 0.95	27.22 $\pm$ 0.80	59.53 $\pm$ 0.341	59.45 $\pm$ 0.31	63.62 $\pm$ 0.38
	Diff	OOM	57.04	24.27	11.62	13.75	11.94
Computer	Acc	OOM	20.11 $\pm$ 1.43	OOM	63.50 $\pm$ 0.29	63.77 $\pm$ 0.36	64.66 $\pm$ 0.78
	F1	OOM	17.58 $\pm$ 1.48	OOM	55.03 $\pm$ 0.16	56.18 $\pm$ 0.35	56.36 $\pm$ 0.66
	Diff	OOM	53.53	OOM	28.66	25.72	23.94
Children	Acc	OOM	17.16 $\pm$ 7.51	OOM	24.54 $\pm$ 0.77	24.50 $\pm$ 1.04	22.99 $\pm$ 0.88
	F1	OOM	9.97 $\pm$ 3.97	OOM	22.41 $\pm$ 0.52	22.16 $\pm$ 0.68	21.81 $\pm$ 0.51
	Diff	OOM	22.15	OOM	36.60	37.51	35.26