ToddlerDiffusion: Interactive Structured Image Generation with Cascaded Schrödinger Bridge

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Abdelrahman, Eslam, Zhao, Liangbing, Hu, Vincent Tao, Cord, Matthieu, Perez, Patrick, Elhoseiny, Mohamed
Format: Preprint
Veröffentlicht: 2023
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914964933967872
author Abdelrahman, Eslam
Zhao, Liangbing
Hu, Vincent Tao
Cord, Matthieu
Perez, Patrick
Elhoseiny, Mohamed
author_facet Abdelrahman, Eslam
Zhao, Liangbing
Hu, Vincent Tao
Cord, Matthieu
Perez, Patrick
Elhoseiny, Mohamed
contents Diffusion models break down the challenging task of generating data from high-dimensional distributions into a series of easier denoising steps. Inspired by this paradigm, we propose a novel approach that extends the diffusion framework into modality space, decomposing the complex task of RGB image generation into simpler, interpretable stages. Our method, termed ToddlerDiffusion, cascades modality-specific models, each responsible for generating an intermediate representation, such as contours, palettes, and detailed textures, ultimately culminating in a high-quality RGB image. Instead of relying on the naive LDM concatenation conditioning mechanism to connect the different stages together, we employ Schrödinger Bridge to determine the optimal transport between different modalities. Although employing a cascaded pipeline introduces more stages, which could lead to a more complex architecture, each stage is meticulously formulated for efficiency and accuracy, surpassing Stable-Diffusion (LDM) performance. Modality composition not only enhances overall performance but enables emerging proprieties such as consistent editing, interaction capabilities, high-level interpretability, and faster convergence and sampling rate. Extensive experiments on diverse datasets, including LSUN-Churches, ImageNet, CelebHQ, and LAION-Art, demonstrate the efficacy of our approach, consistently outperforming state-of-the-art methods. For instance, ToddlerDiffusion achieves notable efficiency, matching LDM performance on LSUN-Churches while operating 2$\times$ faster with a 3$\times$ smaller architecture. The project website is available at: https://toddlerdiffusion.github.io/website/
format Preprint
id arxiv_https___arxiv_org_abs_2311_14542
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle ToddlerDiffusion: Interactive Structured Image Generation with Cascaded Schrödinger Bridge
Abdelrahman, Eslam
Zhao, Liangbing
Hu, Vincent Tao
Cord, Matthieu
Perez, Patrick
Elhoseiny, Mohamed
Computer Vision and Pattern Recognition
Diffusion models break down the challenging task of generating data from high-dimensional distributions into a series of easier denoising steps. Inspired by this paradigm, we propose a novel approach that extends the diffusion framework into modality space, decomposing the complex task of RGB image generation into simpler, interpretable stages. Our method, termed ToddlerDiffusion, cascades modality-specific models, each responsible for generating an intermediate representation, such as contours, palettes, and detailed textures, ultimately culminating in a high-quality RGB image. Instead of relying on the naive LDM concatenation conditioning mechanism to connect the different stages together, we employ Schrödinger Bridge to determine the optimal transport between different modalities. Although employing a cascaded pipeline introduces more stages, which could lead to a more complex architecture, each stage is meticulously formulated for efficiency and accuracy, surpassing Stable-Diffusion (LDM) performance. Modality composition not only enhances overall performance but enables emerging proprieties such as consistent editing, interaction capabilities, high-level interpretability, and faster convergence and sampling rate. Extensive experiments on diverse datasets, including LSUN-Churches, ImageNet, CelebHQ, and LAION-Art, demonstrate the efficacy of our approach, consistently outperforming state-of-the-art methods. For instance, ToddlerDiffusion achieves notable efficiency, matching LDM performance on LSUN-Churches while operating 2$\times$ faster with a 3$\times$ smaller architecture. The project website is available at: https://toddlerdiffusion.github.io/website/
title ToddlerDiffusion: Interactive Structured Image Generation with Cascaded Schrödinger Bridge
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2311.14542