BDBM uses a single masked noise-prediction network to model both forward and backward diffusion bridges between two image distributions, enabling bidirectional translation at roughly half the model cost.
The Swiss Army Knife for Image-to-Image Translation: Multi-Task Diffusion Models
1 Pith paper cite this work. Polarity classification is still indexing.
abstract
Recently, diffusion models were applied to a wide range of image analysis tasks. We build on a method for image-to-image translation using denoising diffusion implicit models and include a regression problem and a segmentation problem for guiding the image generation to the desired output. The main advantage of our approach is that the guidance during the denoising process is done by an external gradient. Consequently, the diffusion model does not need to be retrained for the different tasks on the same dataset. We apply our method to simulate the aging process on facial photos using a regression task, as well as on a brain magnetic resonance (MR) imaging dataset for the simulation of brain tumor growth. Furthermore, we use a segmentation model to inpaint tumors at the desired location in healthy slices of brain MR images. We achieve convincing results for all problems.
citation-role summary
citation-polarity summary
fields
cs.CV 1years
2025 1verdicts
CONDITIONAL 1roles
background 1polarities
background 1representative citing papers
citing papers explorer
-
Bidirectional Diffusion Bridge Models
BDBM uses a single masked noise-prediction network to model both forward and backward diffusion bridges between two image distributions, enabling bidirectional translation at roughly half the model cost.