Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Dual Diffusion Implicit Bridges for Image-to-Image Translation

About

Common image-to-image translation methods rely on joint training over data from both source and target domains. The training process requires concurrent access to both datasets, which hinders data separation and privacy protection; and existing models cannot be easily adapted for translation of new domain pairs. We present Dual Diffusion Implicit Bridges (DDIBs), an image translation method based on diffusion models, that circumvents training on domain pairs. Image translation with DDIBs relies on two diffusion models trained independently on each domain, and is a two-step process: DDIBs first obtain latent encodings for source images with the source diffusion model, and then decode such encodings using the target model to construct target images. Both steps are defined via ordinary differential equations (ODEs), thus the process is cycle consistent only up to discretization errors of the ODE solvers. Theoretically, we interpret DDIBs as concatenation of source to latent, and latent to target Schrodinger Bridges, a form of entropy-regularized optimal transport, to explain the efficacy of the method. Experimentally, we apply DDIBs on synthetic and high-resolution image datasets, to demonstrate their utility in a wide variety of translation tasks and their inherent optimal transport properties.

Xuan Su, Jiaming Song, Chenlin Meng, Stefano Ermon• 2022

Related benchmarks

TaskDatasetResultRank
Object DetectionBDD100K (Nighttime)
AP10.67
26
Image-to-Image Translationedges -> handbags (test)
FID186.8
15
Surface Normal EstimationBedroom Images In-domain
L1 Error0.1849
11
Image-to-Image TranslationEdges -> Handbags 64 x 64 (test)
FID186.8
11
Image-to-Image TranslationDIODE Outdoor 256 x 256 (test)
FID242.3
11
SegmentationRetinal OCT
Accuracy88.1
10
SegmentationChest MRI to CT
Accuracy88.9
10
SegmentationCardiac MRI
Accuracy94.6
10
Image SynthesisChest MRI to CT (test)
FID97.98
9
Image-to-Image TranslationMRI to CT Out-of-domain
FID48.15
9
Showing 10 of 24 rows

Other info

Follow for update