How to Make LMs Strong Node Classifiers?
Not on the map. Why?
Abstract (arXiv)
Language Models (LMs) are increasingly challenging the dominance of domain-specific models, such as Graph Neural Networks (GNNs) and Graph Transformers (GTs), in graph learning tasks. Following this trend, we propose a novel approach that empowers off-the-shelf LMs to achieve performance comparable to state-of-the-art (SOTA) GNNs on node classification tasks, without requiring any architectural modification. By preserving the LM's original architecture, our approach retains a key benefit of LM instruction tuning: the ability to jointly train on diverse datasets, fostering greater flexibility and efficiency. To achieve this, we introduce two key augmentation strategies: (1) Enriching LMs' input using topological and semantic retrieval methods, which provide richer contextual information, and (2) guiding the LMs' classification process through a lightweight GNN classifier that effectively prunes class candidates. Our experiments on real-world datasets show that backbone Flan-T5 LMs equipped with these augmentation strategies outperform SOTA text-output node classifiers and are comparable to top-performing vector-output node classifiers. By bridging the gap between specialized node classifiers and general LMs, this work paves the way for more versatile and widely applicable graph learning models. We will open-source the code upon publication.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 63 of its 107 references that are in the map. First posted on arXiv 3 Oct 2024.
Builds on
63 of 107 references.
References · 63
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
- 2017Inductive Representation Learning on Large Graphscited by 6,284
- 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,779
- 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,769
- 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,747
- 2017Prototypical Networks for Few-shot Learningcited by 3,001
- 2022Scaling Instruction-Finetuned Language Modelscited by 2,997
- 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,157
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,148
- 2020REALM: Retrieval-Augmented Language Model Pre-Trainingcited by 2,126
38 more
- 2020MiniLM: Deep Self-Attention Distillation for Task-Agnostic Compression of Pre-Trained Transformerscited by 1,414
- 2024From Local to Global: A Graph RAG Approach to Query-Focused Summarizationcited by 1,411
- 2016Variational Graph Auto-Encoderscited by 1,373
- 2020Leveraging Passage Retrieval with Generative Models for Open Domain Question Answeringcited by 1,203
- 2018Deeper Insights into Graph Convolutional Networks for Semi-Supervised Learningcited by 1,099
- 2018Predict then Propagate: Graph Neural Networks meet Personalized PageRankcited by 987
- 2022Atlas: Few-shot Learning with Retrieval Augmented Language Modelscited by 940
- 2023REPLUG: Retrieval-Augmented Black-Box Language Modelscited by 786
- 2024Retrieval-Augmented Generation for AI-Generated Content: A Surveycited by 352
- 2020VirTex: Learning Visual Representations from Textual Annotationscited by 338
- 2024Graph Retrieval-Augmented Generation: A Surveycited by 320
- 2022Generate rather than Retrieve: Large Language Models are Strong Context Generatorscited by 296
- 2023One for All: Towards Training One Graph Model for All Classification Taskscited by 239
- 2023Language is All a Graph Needscited by 219
- 2023Large Language Models on Graphs: A Comprehensive Surveycited by 190
- 2024LLaGA: Large Language and Graph Assistantcited by 177
- 2021Training Graph Neural Networks with 1000 Layerscited by 169
- 2022Learning on Large-scale Text-attributed Graphs via Variational Inferencecited by 153
- 2021Symbolic Music Generation with Diffusion Modelscited by 152
- 2021Node Feature Extraction by Self-Supervised Multi-scale Neighborhood Predictioncited by 137
- 2024GNN-RAG: Graph Neural Retrieval for Large Language Model Reasoningcited by 128
- 2024Graph Chain-of-Thought: Augmenting Large Language Models by Reasoning on Graphscited by 116
- 2024Let Your Graph Do the Talking: Encoding Structured Data for LLMscited by 112
- 2023GraphText: Graph Reasoning in Text Spacecited by 110
- 2022Transformer for Graphs: An Overview from Architecture Perspectivecited by 107
- 2021GNNAutoScale: Scalable and Expressive Graph Neural Networks via Historical Embeddingscited by 90
- 2023Retrieving Multimodal Information for Augmented Generation: A Surveycited by 84
- 2024ZeroG: Investigating Cross-dataset Zero-shot Transferability in Graphscited by 83
- 2023Graph-ToolFormer: To Empower LLMs with Graph Reasoning Ability via Prompt Augmented by ChatGPTcited by 78
- 2018A Retrieve-and-Edit Framework for Predicting Structured Outputscited by 73
- 2023SimTeG: A Frustratingly Simple Approach Improves Textual Graph Learningcited by 73
- 2024Can GNN be Good Adapter for LLMs?cited by 71
- 2024A Survey of Large Language Models for Graphscited by 71
- 2024UniGraph: Learning a Unified Cross-Domain Foundation Model for Text-Attributed Graphscited by 60
- 2024Efficient Tuning and Inference for Large Language Models on Textual Graphscited by 54
- 2023KnowledGPT: Enhancing Large Language Models with Retrieval and Storage Access on Knowledge Basescited by 52
- 2024Can we Soft Prompt LLMs for Graph Learning Tasks?cited by 52
- 2023LLM4DyG: Can Large Language Models Solve Spatial-Temporal Problems on Dynamic Graphs?cited by 40
50 most cited of 63 All 63 (JSON)
Built on by
Citing papers per year: 2024: 2; 2025: 1; 2026: 1.
4 papers.
- 2024Learning Graph Quantized Tokenizerscited by 19
- 2024Graph Learning in the Era of LLMs: A Survey from the Perspective of Data, Models, and Taskscited by 10
- 2025Efficient Text-Attributed Graph Learning through Selective Annotation and Graph Alignmentcited by 0
- 2026Rethinking Message Passing as Retrieval for Text-Attributed Graph Learningcited by 0
- 2026Rethinking Message Passing as Retrieval for Text-Attributed Graph Learning27 Aug 2026
- 2025Efficient Text-Attributed Graph Learning through Selective Annotation and Graph Alignment8 Jun 2025
- 2024Graph Learning in the Era of LLMs: A Survey from the Perspective of Data, Models, and Tasks17 Dec 2024
- 2024Learning Graph Quantized Tokenizers17 Oct 2024