Improving Graph Generation by Restricting Graph Bandwidth
Nathaniel Lee Diamant, Alex M. Tseng, Kangway V. Chuang, Tommaso Biancalani, Gabriele Scalia
Abstract
Deep graph generative modeling has proven capable of learning the distribution of complex, multiscale structures characterizing real-world graphs. However, one of the main limitations of existing methods is their large output space, which limits generation scalability and hinders accurate modeling of the underlying distribution. To overcome these limitations, we propose a novel approach that significantly reduces the output space of existing graph generative models. Specifically, starting from the observation that many real-world graphs have low graph bandwidth, we restrict graph bandwidth during training and generation. Our strategy improves both generation scalability and quality without increasing architectural complexity or reducing expressiveness. Our approach is compatible with existing graph generative methods, and we describe its application to both autoregressive and one-shot models. We extensively validate our strategy on synthetic and real datasets, including molecular graphs. Our experiments show that, in addition to improving generation efficiency, our approach consistently improves generation quality and reconstruction accuracy. The implementation is made available 1 .
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers13
- Efficient and Scalable Graph Generation through Iterative Local ExpansionAndreas Bergmeister, Karolis Martinkus, Nathanaël Perraudin, Roger WattenhoferICLR 2024 · 38 citations
- Generative Modelling of Structurally Constrained GraphsManuel Madeira, Clément Vignac, Dorina Thanou, Pascal FrossardNeurIPS 2024 · 20 citations
- A Simple and Scalable Representation for Graph GenerationYunhui Jang, Seul Lee, Sungsoo AhnICLR 2024 · 14 citations
- Flatten Graphs as Sequences: Transformers are Scalable Graph GeneratorsDexiong Chen, Markus Krimmel, Karsten M. BorgwardtNeurIPS 2025 · 13 citations
- Bures-Wasserstein Flow Matching for Graph GenerationKeyue Jiang, Jiahao Cui, Xiaowen Dong, Laura ToniICLR 2026 · 10 citations
Builds on9
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Improved Denoising Diffusion Probabilistic ModelsAlexander Quinn Nichol, Prafulla DhariwalICML 2021 · 5,234 citations
- Strategies for Pre-training Graph Neural NetworksWeihua Hu, Bowen Liu, Joseph Gomes, Marinka Zitnik et al.ICLR 2020 · 1,744 citations
- GraphAF: a Flow-based Autoregressive Model for Molecular Graph GenerationChence Shi, Minkai Xu, Zhaocheng Zhu, Weinan Zhang et al.ICLR 2020 · 532 citations
- GraphGen: A Scalable Approach to Domain-agnostic Labeled Graph GenerationNikhil Goyal, Harsh Vardhan Jain, Sayan RanuWWW 2020 · 110 citations
Related papers
- Scalable Deep Generative Modeling for Sparse GraphsHanjun Dai, Azade Nazi, Yujia Li, Bo Dai et al.ICML 2020 · 95 citations
- HiGen: Hierarchical Graph Generative NetworksMahdi KaramiICLR 2024 · 6 citations
- SBGD: Improving Graph Diffusion Generative Model via Stochastic Block DiffusionJunwei Su, Shan WuICML 2025
- SPECTRE: Spectral Conditioning Helps to Overcome the Expressivity Limits of One-shot Graph GeneratorsKarolis Martinkus, Andreas Loukas, Nathanaël Perraudin, Roger WattenhoferICML 2022 · 109 citations
- Order Matters: Probabilistic Modeling of Node Sequence for Graph GenerationXiaohui Chen, Xu Han, Jiajing Hu, Francisco J. R. Ruiz et al.ICML 2021 · 40 citations
