Training-free Style Transfer Emerges from h-space in Diffusion models

Mingi Kwon ⋆ Jaeseok Jeong ⋆ Youngjung Uh

Yonsei University, Korea
* denotes equal contribution

Paper (Comming soon) Code (Comming soon)

Teaser


Note that we use fixed pretrained diffusion models without any fine-tuning or gradient guidance.

firenze.jpg

Abstract


Diffusion models (DMs) synthesize high-quality images in various domains. However, controlling their generative process of pre-trained vanilla DMs is not as easy as generative adversarial networks due to the lack of semantic latent space. Recently, Asyrp discovers a hidden latent space of pre-trained vanilla DMs called $ h $-space where changes along certain directions result in attribute changes. In this paper, we dive deeper into Asyrp to discover a secret recipe for content injection and style transfer. 1) cumulatively adding $ h_t $ from different samples along the timesteps leads to content replacement, 2) the distribution of $ h_t $ should be maintained by normalization, 3) the distribution of $ x_t $ should be maintained by calibration, and 4) the content can be harmonized into arbitrary style images even unseen during training. To the best of our knowledge, our method introduces the first training-free feed-forward style transfer only with an unconditional pre-trained frozen generative network. The code will be available online.

DiffStyle


Generative process of DiffStyle

(a) During the editing interval, target content is recursively injected to the asymmetric reverse process. (b) illustrates the green box in (a) in detail. (c) Ours also enables local content injection

Results


Note that we use fixed pretrained diffusion models without any fine-tuning or gradient guidance.

firenze.jpg
CelebA-HQ, LSUN-Church, AFHQ-dog, METFACES, LSUN-bedroom
firenze.jpg
Local editing and Style transfer

More Results


firenze.jpg
AFHQ-dog.
firenze.jpg
LSUN-Bedroom.
firenze.jpg
LSUN-Church
firenze.jpg
Local editing on CelebA-HQ
firenze.jpg
CelebA-HQ
firenze.jpg
METFACES
firenze.jpg
Style transfer with artistic references

© This webpage was in part inspired from this template.