Towards Diverse and Faithful One-shot Adaption of Generative Adversarial Networks

Yabo Zhang, Mingshuai Yao, Yuxiang Wei, Zhilong Ji, Jinfeng Bai +1 more
7/18/2022
cs.CV

No official code found. Best implementation: YBYBZhang/DiFa (51 stars, MIT, PyTorch, last commit 2022-10-12).

Abstract

One-shot generative domain adaption aims to transfer a pre-trained generator on one domain to a new domain using one reference image only. However, it remains very challenging for the adapted generator (i) to generate diverse images inherited from the pre-trained generator while (ii) faithfully acquiring the domain-specific attributes and styles of the reference image. In this paper, we present a novel one-shot generative domain adaption method, i.e., DiFa, for diverse generation and faithful adaptation. For global-level adaptation, we leverage the difference between the CLIP embedding of reference image and the mean embedding of source images to constrain the target generator. For local-level adaptation, we introduce an attentive style loss which aligns each intermediate token of adapted image with its corresponding token of the reference image. To facilitate diverse generation, selective cross-domain consistency is introduced to select and retain the domain-sharing attributes in the editing latent $\mathcal{W}+$ space to inherit the diversity of pre-trained generator. Extensive experiments show that our method outperforms the state-of-the-arts both quantitatively and qualitatively, especially for the cases of large domain gaps. Moreover, our DiFa can easily be extended to zero-shot generative domain adaption with appealing results. Code is available at https://github.com/1170300521/DiFa.

View on arXivView PDF

Code Implementations(1)

YBYBZhang/DiFaCommunity100%
PyTorchMIT

[NeurIPS 2022] Official pytorch implementation of "Towards Diverse and Faithful One-shot Domain Adaption of Generative Adversarial Networks"

512C++, CudaMay 18, 20224 years agoMIT
generative-adversarial-networkpytorch

Similar Papers

Cite this paper

@article{zhang2022towards,
  title  = {Towards Diverse and Faithful One-shot Adaption of Generative Adversarial Networks},
  author = {Yabo Zhang and Mingshuai Yao and Yuxiang Wei and Zhilong Ji and Jinfeng Bai and Wangmeng Zuo},
  year   = {2022},
  eprint = {2207.08736},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2207.08736v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2207.08736.svg)](https://codeofpaper.com/paper/2207.08736)

Discussion