HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Stochastic Conditional Diffusion Models for Robust Semantic Image Synthesis

Juyeon Ko; Inho Kong; Dogyun Park; Hyunwoo J. Kim

Stochastic Conditional Diffusion Models for Robust Semantic Image Synthesis

Abstract

Semantic image synthesis (SIS) is a task to generate realistic images corresponding to semantic maps (labels). However, in real-world applications, SIS often encounters noisy user inputs. To address this, we propose Stochastic Conditional Diffusion Model (SCDM), which is a robust conditional diffusion model that features novel forward and generation processes tailored for SIS with noisy labels. It enhances robustness by stochastically perturbing the semantic label maps through Label Diffusion, which diffuses the labels with discrete diffusion. Through the diffusion of labels, the noisy and clean semantic maps become similar as the timestep increases, eventually becoming identical at $t=T$. This facilitates the generation of an image close to a clean image, enabling robust generation. Furthermore, we propose a class-wise noise schedule to differentially diffuse the labels depending on the class. We demonstrate that the proposed method generates high-quality samples through extensive experiments and analyses on benchmark datasets, including a novel experimental setup simulating human errors during real-world applications. Code is available at https://github.com/mlvlab/SCDM.

Code Repositories

mlvlab/scdm
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
conditional-image-generation-on-celebamask-hqSCDM
FID: 17.4
LPIPS: 0.418
mIoU: 77.2
image-to-image-translation-on-ade20k-labelsSCDM
FID: 26.9
LPIPS: 0.530
mIoU: 49.4
image-to-image-translation-on-coco-stuffSCDM
FID: 15.3
LPIPS: 0.519
mIoU: 38.1
noisy-semantic-image-synthesis-on-noisySCDM
FID: 32.4
mIoU: 44.7
noisy-semantic-image-synthesis-on-noisy-1SCDM
FID: 31.2
mIoU: 40.1
noisy-semantic-image-synthesis-on-noisy-2SCDM
FID: 28.1
mIoU: 45.1

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Stochastic Conditional Diffusion Models for Robust Semantic Image Synthesis | Papers | HyperAI