LayerDAG: A Layerwise Autoregressive Diffusion Model for Directed Acyclic Graph Generation
Mufei Li, Viraj Shitole, Eli Chien, Changhai Man, Zhaodong Wang, Srinivas, Ying Zhang, Tushar Krishna, Pan Li
Abstract
Directed acyclic graphs (DAGs) serve as crucial data representations in domains such as hardware synthesis and compiler/program optimization for computing systems. DAG generative models facilitate the creation of synthetic DAGs, which can be used for benchmarking computing systems while preserving intellectual property. However, generating realistic DAGs is challenging due to their inherent directional and logical dependencies. This paper introduces LayerDAG, an autoregressive diffusion model, to address these challenges. LayerDAG decouples the strong node dependencies into manageable units that can be processed sequentially. By interpreting the partial order of nodes as a sequence of bipartite graphs, LayerDAG leverages autoregressive generation to model directional dependencies and employs diffusion models to capture logical dependencies within each bipartite graph. Comparative analyses demonstrate that LayerDAG outperforms existing DAG generative models in both expressiveness and generalization, particularly for generating large-scale DAGs with up to 400 nodes-a critical scenario for system benchmarking. Extensive experiments on both synthetic and real-world flow graphs from various computing platforms show that LayerDAG generates valid DAGs with superior statistical properties and benchmarking performance. The synthetic DAGs generated by LayerDAG enhance the training of ML-based surrogate models, resulting in improved accuracy in predicting performance metrics of real-world DAGs across diverse computing platforms. Our implementation is available at https://github.com/Graph-COM/LayerDAG.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext ad12ac7d-9614-4c64-add3-7b9abf9997eaCited by top-tier papers4
- MDNS: Masked Diffusion Neural Sampler via Stochastic Optimal ControlYuchen Zhu, Wei Guo, Jaemoo Choi, Guan-Horng Liu et al.NeurIPS 2025 · 24 citations
- Generating Directed Graphs with Dual Attention and Asymmetric EncodingAlba Carballo-Castro, Manuel Madeira, Yiming QIN, Dorina Thanou et al.ICLR 2026 · 4 citations
- Hard-Constrained Graph Generation with Discrete-Projection DiffusionXuesong Zhang, Haifeng Sun, Qi Qi, Shengkuan Li et al.ICML 2026
- Diffuse Everything: Multimodal Diffusion Models on Arbitrary State SpacesKevin Rojas, Yuchen Zhu, Sichen Zhu, Felix X.-F. Ye et al.ICML 2025
Builds on22
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- High-Resolution Image Synthesis with Latent Diffusion ModelsRobin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser et al.CVPR 2022 · 13,123 citations
- Scalable Diffusion Models with TransformersWilliam Peebles, Saining XieICCV 2023 · 5,568 citations
- Improved Denoising Diffusion Probabilistic ModelsAlexander Quinn Nichol, Prafulla DhariwalICML 2021 · 5,234 citations
- Structured Denoising Diffusion Models in Discrete State-SpacesJacob Austin, Daniel D. Johnson, Jonathan Ho, Daniel Tarlow et al.NeurIPS 2021 · 2,256 citations
Related papers
- Barking up the right tree: an approach to search over molecule synthesis DAGsJohn Bradshaw, Brooks Paige, Matt J. Kusner, Marwin H. S. Segler et al.NeurIPS 2020 · 71 citations
- Flatten Graphs as Sequences: Transformers are Scalable Graph GeneratorsDexiong Chen, Markus Krimmel, Karsten M. BorgwardtNeurIPS 2025 · 13 citations
- Construction of DAG Models for Autonomous SystemsJing Huang, Kuan Jiang, Weijie Wang, Wei Liang et al.DAC 2025 · 2 citations
- Autoregressive Diffusion Model for Graph GenerationLingkai Kong, Jiaming Cui, Haotian Sun, Yuchen Zhuang et al.ICML 2023 · 105 citations
- Differentiable DAG SamplingBertrand Charpentier, Simon Kibler, Stephan GünnemannICLR 2022 · 51 citations
