Dependency-Aware Discrete Diffusion for Scene Graph Generation
About
Scene graphs (SGs) represent objects and their relationships as structured graphs, enabling applications in image generation, robotics, and 3D understanding. Recent work suggests that conditioning image generation on scene graphs improves compositional fidelity compared to text-only prompting. However, since users typically provide text rather than structured graphs, a key challenge is to generate scene graphs from natural language. Prior work on discrete diffusion has demonstrated success in generating generic graphs such as molecules and circuits, but fails to account for the hierarchical structure and strong dependencies between objects, edges, and relations in scene graphs. We address this limitation by introducing a dependency-aware, hierarchically constrained discrete diffusion model for scene graph generation. Our approach decouples structure and semantics across the forward and reverse processes, enabling the model to capture conditional dependencies. At inference time, we perform training-free conditioning to sample text-aligned scene graphs. We evaluate our method on standard SG benchmarks and demonstrate improvements over both continuous and discrete graph generation baselines across graph and layout metrics. When fed to downstream image generation, our approach yields improved compositional alignment compared to text-to-image models, particularly in multi-object scenarios.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Visual Relation Detection | Visual Genome (test) | R@5087 | 13 | |
| Scene Graph Generation | COCO | N-MMD3.97 | 7 | |
| Layout Generation | Visual Genome | F1-std17.9 | 5 | |
| Layout Generation | COCO | F1-std47.24 | 5 | |
| Compositional text-to-image generation | CompSGBench | SG-IoU56.29 | 4 | |
| Text-to-Image Generation | COCO 24 | FID38.68 | 4 | |
| Text-to-Image Generation | CompSGBench 19 | FID33.85 | 4 | |
| Single Object Completion | Visual Genome (test) | w1 Score16.7 | 3 | |
| Single Object Completion | CompSGBench (test) | w1 Score14.4 | 3 | |
| Single Relation Completion | CompSGBench (test) | w1 Score14 | 3 |