SwinGNN: Rethinking Permutation Invariance in Diffusion Models for Graph Generation
On the map: Diffusion models → Flow matching, time series and science
Abstract (arXiv)
Diffusion models based on permutation-equivariant networks can learn permutation-invariant distributions for graph data. However, in comparison to their non-invariant counterparts, we have found that these invariant models encounter greater learning challenges since 1) their effective target distributions exhibit more modes; 2) their optimal one-step denoising scores are the score functions of Gaussian mixtures with more components. Motivated by this analysis, we propose a non-invariant diffusion model, called textitSwinGNN, which employs an efficient edge-to-edge 2-WL message passing network and utilizes shifted window based self-attention inspired by SwinTransformers. Further, through systematic ablations, we identify several critical training and sampling techniques that significantly improve the sample quality of graph generation. At last, we introduce a simple post-processing trick, textiti.e., randomly permuting the generated graphs, which provably converts any graph generative model to a permutation-invariant one. Extensive experiments on synthetic and real-world protein and molecule datasets show that our SwinGNN achieves state-of-the-art performances. Our code is released at https://github.com/qiyan98/SwinGNN.
Key facts. Cited by 26 papers in the map (data through 7 Oct 2026). Placed in Flow matching, time series and science, one of 14 directions in Diffusion models. Builds on 54 of its 80 references that are in the map. First posted on arXiv 4 Jul 2023.
Builds on
54 of 80 references.
Closest ancestors · 5
Same direction, newest first
- 2022DiGress: Discrete Denoising diffusion for graph generation29 Sep
- 2022Equivariant Diffusion for Molecule Generation in 3D31 Mar
- 2022GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generation6 Mar
- 2022Score-based Generative Modeling of Graphs via the System of Stochastic Differential Equations5 Feb
- 2020Permutation Invariant Graph Generation via Score-Based Generative Modeling2 Mar
Other references · 49
- 2017Attention Is All You Needcited by 59,932
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2018How Powerful are Graph Neural Networks?cited by 4,355
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,736
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 2,994
- 2022Imagen Video: High Definition Video Generation with Diffusion Modelscited by 1,765
37 more
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
- 2016Variational Graph Auto-Encoderscited by 1,373
- 2018Unified Perceptual Parsing for Scene Understandingcited by 1,265
- 2022Diffusion Models: A Comprehensive Survey of Methods and Applicationscited by 1,107
- 2020Improved Techniques for Training Score-Based Generative Modelscited by 1,036
- 2018Weisfeiler and Leman Go Neural: Higher-order Graph Neural Networkscited by 878
- 2018Junction Tree Variational Autoencoder for Molecular Graph Generationcited by 647
- 2021Argmax Flows and Multinomial Diffusion: Learning Categorical Distributionscited by 566
- 2018GraphRNN: Generating Realistic Graphs with Deep Auto-regressive Modelscited by 406
- 2018MolGAN: An implicit generative model for small molecular graphscited by 392
- 2019Provably Powerful Graph Networkscited by 385
- 2018GraphVAE: Towards Generation of Small Graphs Using Variational Autoencoderscited by 375
- 2022Analog Bits: Generating Discrete Data using Diffusion Models with Self-Conditioningcited by 362
- 2018Invariant and Equivariant Graph Networkscited by 339
- 2020GraphAF: a Flow-based Autoregressive Model for Molecular Graph Generationcited by 301
- 2019Efficient Graph Generation with Graph Recurrent Attention Networkscited by 231
- 2018Fr'echet ChemNet Distance: A metric for generative models for molecules in drug discoverycited by 191
- 2021GraphDF: A Discrete Flow Model for Molecular Graph Generationcited by 177
- 2019On the equivalence between graph isomorphism testing and function approximation with GNNscited by 172
- 2021Weisfeiler and Leman go Machine Learning: The Story so farcited by 121
- 2019GraphNVP: An Invertible Flow Model for Generating Molecular Graphscited by 115
- 2021Energy-Based Learning for Scene Graph Generationcited by 104
- 2019Graph Normalizing Flowscited by 103
- 2022SPECTRE: Spectral Conditioning Helps to Overcome the Expressivity Limits of One-shot Graph Generatorscited by 96
- 2023Diffusion-Based 3D Human Pose Estimation with Multi-Hypothesis Aggregationcited by 81
- 2018Generative Code Modeling with Graphscited by 59
- 2020Categorical Normalizing Flows via Continuous Transformationscited by 36
- 2021Top-N: Equivariant set and graph generation without exchangeabilitycited by 36
- 2019Neural Turtle Graphics for Modeling City Road Layoutscited by 29
- 2022Towards Better Out-of-Distribution Generalization of Neural Algorithmic Reasoning Taskscited by 29
- 2014Near-optimal-sample estimators for spherical Gaussian mixturescited by 26
- 2021Beyond In-Place Corruption: Insertion and Deletion In Denoising Probabilistic Modelscited by 23
- 2018Practical bounds on the error of Bayesian posterior approximations: A nonasymptotic approachcited by 12
- 2023What functions can Graph Neural Networks compute on random graphs? The role of Positional Encodingcited by 11
- 2022Micro and Macro Level Graph Modeling for Graph Variational Auto-Encoderscited by 8
- 2022What Functions Can Graph Neural Networks Generate?cited by 5
- 2022GraphPNAS: Learning Distribution of Good Neural Architectures via Deep Graph Generative Modelscited by 5
Built on by
Citing papers per year: 2023: 3; 2024: 9; 2025: 8; 2026: 6.
26 papers.
Sort:
By area:
- 2024Utilizing Image Transforms and Diffusion Models for Generative Modeling of Short and Long Time Seriescited by 42
- 2023Efficient and Scalable Graph Generation through Iterative Local Expansioncited by 35
- 2025MoFlow: One-Step Flow Matching for Human Trajectory Forecasting via Implicit Maximum Likelihood Estimation based Distillationcited by 31
- 2024Pard: Permutation-Invariant Autoregressive Diffusion for Graph Generationcited by 25
- 2024Generative Modelling of Structurally Constrained Graphscited by 17
- 2023A Simple and Scalable Representation for Graph Generationcited by 15
- 2024Improving Molecular Graph Generation with Flow Matching and Optimal Transportcited by 10
- 2024Equivariant Denoisers Cannot Copy Graphs: Align Your Graph Diffusion Modelscited by 7
- 2025TrajFlow: Multi-modal Motion Prediction via Flow Matchingcited by 7
- 2024Reviving Life on the Edge: Joint Score-Based Graph Generation of Rich Edge Attributescited by 6
- 2024Generating Physical Dynamics under Priorscited by 6
- 2025FragFM: Hierarchical Framework for Efficient Molecule Generation via Fragment-Level Discrete Flow Matchingcited by 5
14 more
- 2026Rethinking Diffusion Models with Symmetries through Canonicalization with Applications to Molecular Graph Generationcited by 5
- 2024Joint Generative Modeling of Grounded Scene Graphs and Images via Diffusion Modelscited by 4
- 2025DDPS: Discrete Diffusion Posterior Sampling for Paths in Layered Graphscited by 4
- 2025RETRO SYNFLOW: Discrete Flow Matching for Accurate and Diverse Single-Step Retrosynthesiscited by 4
- 2025Diffusion-Free Graph Generation with Next-Scale Predictioncited by 3
- 2023Generative AI-Based Probabilistic Constellation Shaping With Diffusion Modelscited by 2
- 2026ACDZero: MCTS Agent for Mastering Automated Cyber Defensecited by 2
- 2024SeaDAG: Semi-autoregressive Diffusion for Conditional Directed Acyclic Graph Generationcited by 1
- 2025Graph VQ-Transformer (GVT): Fast and Accurate Molecular Generation via High-Fidelity Discrete Latentscited by 1
- 2025Subgraph Generation for Generalizing on Out-of-Distribution Linkscited by 0
- 2026Balancing Symmetry and Efficiency in Graph Flow Matchingcited by 0
- 2026Modular Diffusion Models for Structured Visual Recognitioncited by 0
- 2026From topology learning to graph generation: A unifying perspectivecited by 0
- 2026GLASS: Global Latent Aggregation with Slot-based Set Decoding for Scalable All-Atom Crystal Generationcited by 0
- 2026GLASS: Global Latent Aggregation with Slot-based Set Decoding for Scalable All-Atom Crystal Generation29 Sep 2026
- 2026From topology learning to graph generation: A unifying perspective2 Sep 2026
- 2026Modular Diffusion Models for Structured Visual Recognition21 Jun 2026
- 2026Balancing Symmetry and Efficiency in Graph Flow Matching20 Feb 2026
- 2026Rethinking Diffusion Models with Symmetries through Canonicalization with Applications to Molecular Graph Generation16 Feb 2026
- 2026ACDZero: MCTS Agent for Mastering Automated Cyber Defense5 Jan 2026
- 2025Graph VQ-Transformer (GVT): Fast and Accurate Molecular Generation via High-Fidelity Discrete Latents2 Dec 2025
- 2025Subgraph Generation for Generalizing on Out-of-Distribution Links15 Jul 2025
- 2025TrajFlow: Multi-modal Motion Prediction via Flow Matching10 Jun 2025
- 2025RETRO SYNFLOW: Discrete Flow Matching for Accurate and Diverse Single-Step Retrosynthesis4 Jun 2025
- 2025DDPS: Discrete Diffusion Posterior Sampling for Paths in Layered Graphs29 Apr 2025
- 2025Diffusion-Free Graph Generation with Next-Scale Prediction30 Mar 2025
14 more
- 2025MoFlow: One-Step Flow Matching for Human Trajectory Forecasting via Implicit Maximum Likelihood Estimation based Distillation13 Mar 2025
- 2025FragFM: Hierarchical Framework for Efficient Molecule Generation via Fragment-Level Discrete Flow Matching19 Feb 2025
- 2024Improving Molecular Graph Generation with Flow Matching and Optimal Transport8 Nov 2024
- 2024Utilizing Image Transforms and Diffusion Models for Generative Modeling of Short and Long Time Series25 Oct 2024
- 2024SeaDAG: Semi-autoregressive Diffusion for Conditional Directed Acyclic Graph Generation21 Oct 2024
- 2024Generating Physical Dynamics under Priors1 Sep 2024
- 2024Generative Modelling of Structurally Constrained Graphs25 Jun 2024
- 2024Equivariant Denoisers Cannot Copy Graphs: Align Your Graph Diffusion Models27 May 2024
- 2024Reviving Life on the Edge: Joint Score-Based Graph Generation of Rich Edge Attributes6 Feb 2024
- 2024Pard: Permutation-Invariant Autoregressive Diffusion for Graph Generation6 Feb 2024
- 2024Joint Generative Modeling of Grounded Scene Graphs and Images via Diffusion Models2 Jan 2024
- 2023Efficient and Scalable Graph Generation through Iterative Local Expansion14 Dec 2023
- 2023A Simple and Scalable Representation for Graph Generation4 Dec 2023
- 2023Generative AI-Based Probabilistic Constellation Shaping With Diffusion Models15 Nov 2023
- 2024Utilizing Image Transforms and Diffusion Models for Generative Modeling of Short and Long Time Seriescited by 42
- 2025MoFlow: One-Step Flow Matching for Human Trajectory Forecasting via Implicit Maximum Likelihood Estimation based Distillationcited by 31
- 2024Pard: Permutation-Invariant Autoregressive Diffusion for Graph Generationcited by 25
- 2024Generative Modelling of Structurally Constrained Graphscited by 17
- 2024Improving Molecular Graph Generation with Flow Matching and Optimal Transportcited by 10
- 2024Equivariant Denoisers Cannot Copy Graphs: Align Your Graph Diffusion Modelscited by 7
- 2025TrajFlow: Multi-modal Motion Prediction via Flow Matchingcited by 7
- 2025FragFM: Hierarchical Framework for Efficient Molecule Generation via Fragment-Level Discrete Flow Matchingcited by 5
- 2026Rethinking Diffusion Models with Symmetries through Canonicalization with Applications to Molecular Graph Generationcited by 5
- 2024Joint Generative Modeling of Grounded Scene Graphs and Images via Diffusion Modelscited by 4
- 2025DDPS: Discrete Diffusion Posterior Sampling for Paths in Layered Graphscited by 4
- 2025RETRO SYNFLOW: Discrete Flow Matching for Accurate and Diverse Single-Step Retrosynthesiscited by 4
12 most cited of 17 All 17 (JSON)
- 2026Modular Diffusion Models for Structured Visual Recognition21 Jun 2026
- 2026Balancing Symmetry and Efficiency in Graph Flow Matching20 Feb 2026
- 2026Rethinking Diffusion Models with Symmetries through Canonicalization with Applications to Molecular Graph Generation16 Feb 2026
- 2025Graph VQ-Transformer (GVT): Fast and Accurate Molecular Generation via High-Fidelity Discrete Latents2 Dec 2025
- 2025TrajFlow: Multi-modal Motion Prediction via Flow Matching10 Jun 2025
- 2025RETRO SYNFLOW: Discrete Flow Matching for Accurate and Diverse Single-Step Retrosynthesis4 Jun 2025
- 2025DDPS: Discrete Diffusion Posterior Sampling for Paths in Layered Graphs29 Apr 2025
- 2025MoFlow: One-Step Flow Matching for Human Trajectory Forecasting via Implicit Maximum Likelihood Estimation based Distillation13 Mar 2025
- 2025FragFM: Hierarchical Framework for Efficient Molecule Generation via Fragment-Level Discrete Flow Matching19 Feb 2025
- 2024Improving Molecular Graph Generation with Flow Matching and Optimal Transport8 Nov 2024
- 2024Utilizing Image Transforms and Diffusion Models for Generative Modeling of Short and Long Time Series25 Oct 2024
- 2024SeaDAG: Semi-autoregressive Diffusion for Conditional Directed Acyclic Graph Generation21 Oct 2024
12 newest of 17 All 17 (JSON)
- 2023Efficient and Scalable Graph Generation through Iterative Local Expansioncited by 35
- 2023A Simple and Scalable Representation for Graph Generationcited by 15
- 2024Reviving Life on the Edge: Joint Score-Based Graph Generation of Rich Edge Attributescited by 6
- 2024Generating Physical Dynamics under Priorscited by 6
- 2025Diffusion-Free Graph Generation with Next-Scale Predictioncited by 3
- 2026ACDZero: MCTS Agent for Mastering Automated Cyber Defensecited by 2
- 2025Subgraph Generation for Generalizing on Out-of-Distribution Linkscited by 0
- 2026From topology learning to graph generation: A unifying perspectivecited by 0
- 2026GLASS: Global Latent Aggregation with Slot-based Set Decoding for Scalable All-Atom Crystal Generationcited by 0
- 2026GLASS: Global Latent Aggregation with Slot-based Set Decoding for Scalable All-Atom Crystal Generation29 Sep 2026
- 2026From topology learning to graph generation: A unifying perspective2 Sep 2026
- 2026ACDZero: MCTS Agent for Mastering Automated Cyber Defense5 Jan 2026
- 2025Subgraph Generation for Generalizing on Out-of-Distribution Links15 Jul 2025
- 2025Diffusion-Free Graph Generation with Next-Scale Prediction30 Mar 2025
- 2024Generating Physical Dynamics under Priors1 Sep 2024
- 2024Reviving Life on the Edge: Joint Score-Based Graph Generation of Rich Edge Attributes6 Feb 2024
- 2023Efficient and Scalable Graph Generation through Iterative Local Expansion14 Dec 2023
- 2023A Simple and Scalable Representation for Graph Generation4 Dec 2023
Similar, not cited
Close in wording, no citation either way
- 2023Autoregressive Diffusion Model for Graph Generation
- 2023Generative Diffusion Models on Graphs: Methods and Applications
- 2025Permutation-Invariant Spectral Learning via Dyson Diffusion
- 2024Advancing Graph Generation through Beta Diffusion
- 2023Graph Generation with Diffusion Mixture
- 2023Efficient and Degree-Guided Graph Generation via Discrete Diffusion Modeling
- 2026Principled Latent Diffusion for Graphs via Laplacian Autoencoders
- 2023SaGess: Sampling Graph Denoising Diffusion Model for Scalable Graph Generation