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

Collaborative Diffusion for Multi-Modal Face Generation and Editing

About

Diffusion models arise as a powerful generative tool recently. Despite the great progress, existing diffusion models mainly focus on uni-modal control, i.e., the diffusion process is driven by only one modality of condition. To further unleash the users' creativity, it is desirable for the model to be controllable by multiple modalities simultaneously, e.g., generating and editing faces by describing the age (text-driven) while drawing the face shape (mask-driven). In this work, we present Collaborative Diffusion, where pre-trained uni-modal diffusion models collaborate to achieve multi-modal face generation and editing without re-training. Our key insight is that diffusion models driven by different modalities are inherently complementary regarding the latent denoising steps, where bilateral connections can be established upon. Specifically, we propose dynamic diffuser, a meta-network that adaptively hallucinates multi-modal denoising steps by predicting the spatial-temporal influence functions for each pre-trained uni-modal model. Collaborative Diffusion not only collaborates generation capabilities from uni-modal diffusion models, but also integrates multiple uni-modal manipulations to perform multi-modal editing. Extensive qualitative and quantitative experiments demonstrate the superiority of our framework in both image quality and condition consistency.

Ziqi Huang, Kelvin C.K. Chan, Yuming Jiang, Ziwei Liu• 2023

Related benchmarks

TaskDatasetResultRank
Many-to-one MRI Synthesis (T2, FLAIR -> T1)BRATS (test)
PSNR23.59
21
Many-to-one MRI Synthesis (T1, FLAIR -> T2)BRATS (test)
PSNR22.96
21
Nanoparticles distribution predictionB16 tumor model dataset (external val)
SSIM (%)85.85
13
NPs distribution predictionNPs distribution dataset 1.0 (Internal val)
SSIM92.96
13
MRI Synthesis (T1, T2 -> FLAIR)BRATS (test)
PSNR22.9
13
Facial Image GenerationDISFA
FID5.949
11
Facial Image GenerationBP4D
FID7.992
11
Facial Conditional GenerationFacial Conditional Generation Text + Face Parsing
FID122
4
Face GenerationCelebA-HQ (val)
FID111.4
3
Showing 9 of 9 rows

Other info

Code

Follow for update