Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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

About

Can we model Non-Euclidean graphs as pure language or even Euclidean vectors while retaining their inherent information? The Non-Euclidean property have posed a long term challenge in graph modeling. Despite recent graph neural networks and graph transformers efforts encoding graphs as Euclidean vectors, recovering the original graph from vectors remains a challenge. In this paper, we introduce GraphsGPT, featuring an Graph2Seq encoder that transforms Non-Euclidean graphs into learnable Graph Words in the Euclidean space, along with a GraphGPT decoder that reconstructs the original graph from Graph Words to ensure information equivalence. We pretrain GraphsGPT on $100$M molecules and yield some interesting findings: (1) The pretrained Graph2Seq excels in graph representation learning, achieving state-of-the-art results on $8/9$ graph classification and regression tasks. (2) The pretrained GraphGPT serves as a strong graph generator, demonstrated by its strong ability to perform both few-shot and conditional graph generation. (3) Graph2Seq+GraphGPT enables effective graph mixup in the Euclidean space, overcoming previously known Non-Euclidean challenges. (4) The edge-centric pretraining framework GraphsGPT demonstrates its efficacy in graph domain tasks, excelling in both representation and generation. Code is available at \href{https://github.com/A4Bio/GraphsGPT}{GitHub}.

Zhangyang Gao, Daize Dong, Cheng Tan, Jun Xia, Bozhen Hu, Stan Z. Li• 2024

Related benchmarks

TaskDatasetResultRank
Graph property detectionPlanar Small House of Graphs (test)
F1 Score90
17
Graph property detectionTree Small synthetic (test)
F1 Score98
17
Graph property detectionHam Large House of Graphs (test)
F1 Score90
17
Graph property detectionClaw Small House of Graphs (test)
F1 Score92
17
Graph property detectionPlanar Medium House of Graphs (test)
F1 Score97
17
Graph property detectionPlanar Large House of Graphs (test)
F1 Score91
17
Graph property detectionClaw Huge House of Graphs (test)
F1 Score73
17
Graph property detectionTree Large synthetic (test)
F1 Score89
17
Graph property detectionHam Huge House of Graphs (test)
F1 Score81
17
Graph property detectionClaw Large House of Graphs (test)
F1 Score81
17
Showing 10 of 14 rows

Other info

Follow for update