atlas

Research map / Graph learning / Graph self-supervised learning

A Graph is Worth $K$ Words: Euclideanizing Graph using Pure Transformer

ICML 2024 · Graph learning › Graph self-supervised learning · approach: Diffusion-based graph generation · 1 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 GraphsGPT, a pure-transformer encoder-decoder (Graph2Seq encoder, GraphGPT decoder) that maps molecular graphs into a fixed set of learnable Graph Words and reconstructs them, pretrained on ~100M ZINC molecules with an edge-centric GPT-style objective.

Problem

Graphs are non-Euclidean, so it is hard to encode them as Euclidean vectors that can be decoded back to the original graph, which limits autoregressive pure-transformer modeling for graph representation, generation and manipulation.

Approach

Flatten a molecular graph via DFS into a flexible token sequence, encode it into K Euclidean Graph Words with a pure transformer, and decode the graph autoregressively using an edge-centric generation scheme with block-wise causal attention and learnable position codebooks that are randomly shuffled.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
pretrain Graph2Seq-1W with GPT-style whole-graph reconstructionGraph2Seq-1W · MoleculeNet property prediction (9 tasks)performance across tasks: non-trivial improvement over 8 out of 9 datasetsHelped
mix Graph Words during training (ablation)GraphsGPT · MoleculeNet HIVROC-AUC: 79.4 -> 79.8 (+0.4)Helped
mix Graph Words during training (ablation)GraphsGPT · MoleculeNet BBBPROC-AUC: 72.8 -> 73.4 (+0.6)Helped
mix Graph Words during training (ablation)GraphsGPT · MoleculeNet BaceROC-AUC: 83.4 -> 85.4 (+2.0)Helped
mix Graph Words during training (ablation)GraphsGPT · MoleculeNet Tox21ROC-AUC: 76.9 -> 77.2 (+0.3)Helped
mix Graph Words during training (ablation)GraphsGPT · MoleculeNet ToxCastROC-AUC: 65.4 -> 65.5 (+0.1)Helped
mix Graph Words during training (ablation)GraphsGPT · MoleculeNet SiderROC-AUC: 68.2 -> 68.9 (+0.7)Helped
increase Graph Word sampling variance from s=0.25 to s=2.0 (ablation)GraphGPT-1W few-shot generation · MOSES unconditional generationNovelty: 0.255 -> 1.0Helped
increase Graph Word sampling variance from s=0.25 to s=2.0 (ablation)GraphGPT-1W few-shot generation · MOSES unconditional generationValidity: 0.995 -> 0.972Hurt
pretrain GraphGPT-1W-C (ablation)conditional molecule generation · QED, SA, and logP targets (average)MAD: 0.127 -> 0.069Helped
pretrain GraphGPT-1W-C (ablation)conditional molecule generation · QED, SA, and logP targets (average)Validity: 0.984 -> 0.995Helped
pretrain GraphGPT-1W-C (ablation)conditional molecule generation · QED, SA, and logP targets (average)SD: 0.118 -> 0.261Hurt

Surprising findings

Limitations

Experimental setup

Datasets: ZINC MoleculeNet MOSES ZINC-C ZINC-250K GuacaMol

Benchmarks: MoleculeNet MOSES ZINC-250K ZINC-C

Metrics: ROC-AUC RMSD validity uniqueness novelty IntDiv MAD SD NSPDK FCD

Baselines: GIN InfoGraph GPT-GNN EdgePred ContextPred GraphLoG GraphMAE Mole-BERT GraphMVP MolGPT G-mixup GraphAF

Builds on: TokenGT GPT-GNN MolGPT G-mixup

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