atlas

Research map / Graph learning / Graph self-supervised learning

Harnessing Explanations: LLM-to-LM Interpreter for Enhanced Text-Attributed Graph Representation Learning

ICLR 2024 · Graph learning › Graph self-supervised learning · approach: LLMs and graph foundation models · 16 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes TAPE, which prompts an LLM (GPT-3.5, Llama2) for ranked predictions and explanations on text-attributed graph nodes, fine-tunes a small LM (DeBERTa) to turn the explanations into node features, and trains GNNs on the averaged predictions from these features.

Problem

Node classification on text-attributed graphs (TAGs) where shallow text features are weak and LM-based pipelines are costly, and where LLMs accessed only via text APIs (LMaaS) cannot be fine-tuned or expose logits.

Approach

Use LLM-generated textual explanations and ranked predictions as an intermediate medium: a small LM "interpreter" fine-tuned on those explanations converts them into frozen node features for any downstream GNN, avoiding LM-GNN iterative training and requiring only text-level LLM access.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add explanation and prediction features alongside original-text features (ablation)RevGAT on ogbn-arxiv · ogbn-arxiv testaccuracy: 0.7504 ± 0.0020 -> 0.7750 ± 0.0012Helped
Add original-text and prediction features alongside explanation features (ablation)RevGAT on ogbn-arxiv · ogbn-arxiv testaccuracy: 0.7529 ± 0.0052 -> 0.7750 ± 0.0012Helped
Add original-text and explanation features alongside prediction features (ablation)RevGAT on ogbn-arxiv · ogbn-arxiv testaccuracy: 0.7519 ± 0.0031 -> 0.7750 ± 0.0012Helped
Use Llama2-13b-chat instead of GPT-3.5 to generate predictions and explanationsTAPE with RevGAT · ogbn-arxivaccuracy: 0.7750 ± 0.0012 -> 0.7619 ± 0.0027Hurt
Move the paper title before the abstract in the prompt (ablation)GPT-3.5 zero-shot classification · ogbn-arxivaccuracy: 0.720 -> 0.695Hurt
Instruct the LLM to focus only on content in the actual text (ablation)GPT-3.5 zero-shot classification · ogbn-arxivaccuracy: 0.720 -> 0.695Hurt
Request step-by-step categorization reasoning in the prompt (ablation)GPT-3.5 zero-shot classification · ogbn-arxivaccuracy: 0.720 -> 0.705Hurt
Use TAPE's decoupled training instead of GLEM's iterative trainingDeBERTa-base + RevGAT on ogbn-arxiv · ogbn-arxivtotal training time: 9.18h -> 3.76min; 2.88× speedupHelped

Surprising findings

Limitations

Experimental setup

Datasets: Cora PubMed ogbn-arxiv ogbn-products tape-arxiv23

Benchmarks: Cora PubMed ogbn-arxiv ogbn-products tape-arxiv23

Metrics: accuracy

Baselines: GCN GraphSAGE RevGAT MLP GIANT LM finetune (DeBERTa) GLEM zero-shot GPT-3.5

Builds on: GLEM DeBERTa RevGAT GIANT

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy