atlas

Research map / Graph learning / Graph self-supervised learning

Pard: Permutation-Invariant Autoregressive Diffusion for Graph Generation

NeurIPS 2024 · Graph learning › Graph self-supervised learning · approach: Diffusion-based graph generation · 3 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.

Pard generates graphs block-by-block autoregressively, with each block modeled by a shared permutation-equivariant discrete diffusion model using a PPGN-transformer hybrid, achieving state-of-the-art molecular and generic graph generation without extra features.

Problem

Graph generation: autoregressive models are sensitive to node ordering, while diffusion models need thousands of denoising steps and extra structural features to perform well.

Approach

Exploit a unique structural partial order on graph nodes (via k-hop degree structure) to split graph generation into permutation-invariant blocks; model each block's conditional with a shared discrete diffusion model using an equivariant PPGN-Transformer, with a causal block-ID masking scheme enabling parallel training.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
add autoregressive block generation (K_h=1) instead of full diffusion without AR (ablation)Pard · QM9Validity: 93.8 -> 97.1Helped
add autoregressive block generation (K_h=1) instead of full diffusion without AR (ablation)Pard · QM9Molecule stability: 76.4 -> 86.1Helped
add autoregressive block generation (K_h=1) instead of full diffusion without AR (ablation)Pard · QM9Uniqueness: 96.9 -> 96.5Hurt
increase no-AR diffusion steps from 140 to 280 (ablation)Pard without AR · QM9Validity: 93.8 -> 94.3Helped
increase no-AR diffusion steps from 140 to 490 (ablation)Pard without AR · QM9Validity: 93.8 -> 95.2Helped
increase no-AR diffusion steps to 490 rather than use AR with K_h=1 and 140 total steps (ablation)Pard · QM9Validity: 97.1 vs 95.2Hurt
replace PPGNTransformer with a Transformer-only backbone (ablation)Pard · QM9Validity: 97.1 -> 26.3Hurt
replace PPGNTransformer with a PPGN-only backbone (ablation)Pard · QM9Molecule stability: 86.2 -> 84.67Hurt
replace PPGN-only backbone with PPGNTransformer (ablation)Pard · QM9Molecule stability: 84.67 -> 86.2Helped
use Pard rather than DiGress (marg. + feat.)molecular graph generation · QM9Molecule stability: 79.8 -> 86.1Helped
use Pard rather than GDSSmolecular graph generation · ZINC250kFCD: 14.66 -> 1.98Helped
use Pard rather than DiGressmolecular graph generation · MOSESSNN: 0.52 -> 0.56Helped

Surprising findings

Limitations

Experimental setup

Datasets: QM9 ZINC250k MOSES Community-small Caveman Cora Breast Grid

Benchmarks: QM9 ZINC250k MOSES Community-small Caveman Cora Breast Grid

Metrics: validity uniqueness novelty atom stability molecule stability FCD SNN scaffold similarity filters MMD degree

Baselines: DiGress GDSS GraphARM SwinGNN EDP-GNN GraphRNN GRAN ConGress JT-VAE GraphINVENT

Builds on: DiGress PPGN GRIT D3PM GraphARM

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