Heterogeneous Image GNN: Graph-Conditioned Diffusion for Image Synthesis

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Menneer, Rupert, Margadji, Christos, Pattinson, Sebastian W.
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866929695827689472
author Menneer, Rupert
Margadji, Christos
Pattinson, Sebastian W.
author_facet Menneer, Rupert
Margadji, Christos
Pattinson, Sebastian W.
contents We introduce a novel method for conditioning diffusion-based image synthesis models with heterogeneous graph data. Existing approaches typically incorporate conditioning variables directly into model architectures, either through cross-attention layers that attend to text latents or image concatenation that spatially restrict generation. However, these methods struggle to handle complex scenarios involving diverse, relational conditioning variables, which are more naturally represented as unstructured graphs. This paper presents Heterogeneous Image Graphs (HIG), a novel representation that models conditioning variables and target images as two interconnected graphs, enabling efficient handling of variable-length conditioning inputs and their relationships. We also propose a magnitude-preserving GNN that integrates the HIG into the existing EDM2 diffusion model using a ControlNet approach. Our approach improves upon the SOTA on a variety of conditioning inputs for the COCO-stuff and Visual Genome datasets, and showcases the ability to condition on graph attributes and relationships represented by edges in the HIG.
format Preprint
id arxiv_https___arxiv_org_abs_2502_01309
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Heterogeneous Image GNN: Graph-Conditioned Diffusion for Image Synthesis
Menneer, Rupert
Margadji, Christos
Pattinson, Sebastian W.
Computer Vision and Pattern Recognition
We introduce a novel method for conditioning diffusion-based image synthesis models with heterogeneous graph data. Existing approaches typically incorporate conditioning variables directly into model architectures, either through cross-attention layers that attend to text latents or image concatenation that spatially restrict generation. However, these methods struggle to handle complex scenarios involving diverse, relational conditioning variables, which are more naturally represented as unstructured graphs. This paper presents Heterogeneous Image Graphs (HIG), a novel representation that models conditioning variables and target images as two interconnected graphs, enabling efficient handling of variable-length conditioning inputs and their relationships. We also propose a magnitude-preserving GNN that integrates the HIG into the existing EDM2 diffusion model using a ControlNet approach. Our approach improves upon the SOTA on a variety of conditioning inputs for the COCO-stuff and Visual Genome datasets, and showcases the ability to condition on graph attributes and relationships represented by edges in the HIG.
title Heterogeneous Image GNN: Graph-Conditioned Diffusion for Image Synthesis
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2502.01309