SimTeG: A Frustratingly ple Approach Improves xtual raph Learning

Keyu Duan, Qian Liu, Tat-seng Chua, Shuicheng Yan, Wei Tsang Ooi, Qizhe Xie, Junxian He

Main figure for SimTeG: A Frustratingly ple Approach Improves xtual raph Learning

Abstract

Textual graphs (TGs) are graphs whose nodes correspond to text (sentences or documents), which are widely prevalent. The representation learning of TGs involves two stages: \((i)\) unsupervised feature extraction and \((ii)\) supervised graph representation learning. In recent years, extensive efforts have been devoted to the latter stage, where Graph Neural Networks (GNNs) have dominated. However, the former stage for most existing graph benchmarks still relies on traditional feature engineering techniques. More recently, with the rapid development of language models (LMs), researchers have focused on leveraging LMs to facilitate the learning of TGs, either by jointly training them in a computationally intensive framework (merging the two stages), or designing complex self-supervised training tasks for feature extraction (enhancing the first stage). In this work, we present SimTeG, a frustratingly ple approach for xtual raph learning that does not innovate in frameworks, models, and tasks. Instead, we first perform supervised parameter-efficient fine-tuning (PEFT) on a pre-trained LM on the downstream task, such as node classification. We then generate node embeddings using the last hidden states of finetuned LM. These derived features can be further utilized by any GNN for training on the same task. We evaluate our approach on two fundamental graph representation learning tasks: node classification and link prediction. Through extensive experiments, we show that our approach significantly improves the performance of various GNNs on multiple graph benchmarks. Remarkably, when additional supporting text provided by large language models (LLMs) is included, a simple two-layer GraphSAGE trained on an ensemble of SimTeG \ achieves an accuracy of 77.48\ comparable to state-of-the-art (SOTA) performance obtained from far more complicated GNN architectures. Furthermore, when combined with a SOTA GNN, we achieve a new SOTA of \(78.04 \ Our code is publicly available at \ and the generated node features for all graph benchmarks can be accessed at.

Proposed method: SimTEG

Benchmark Results

Tested on 1 dataset

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

OGB

#ModelAccuracy ↑Paper
1
Jacobi
0.8968
2
GPR
0.8953
3
LEMP4HG
0.8939
4
SAGE
0.8933
15
GLEM
actual #15 vs paper #15
0.8736 ±0.0007
actual #15 vs paper #15
paper 0.8736 ±0.0007
paper 0.8736 ±0.0007 Δ at pub 0.0611
34
X-GIANT
0.8612 ±0.0034
This paper
39
SimTEG
0.8540 ±0.0028
This paper
105
X-OGB
0.8182 ±0.0044
This paper
351
MLP
0.6605 ±0.0020
SOTA performance (not significantly different from the best) Proposed method (this paper) Baseline / comparison (this paper) Global context (other papers) Best at pub — reported value improved over the best known when published