One-Shot Adaptation of GAN in Just One CLIP

Gihyun Kwon, Jong Chul Ye
3/17/2022
cs.CVcs.AIcs.LG

Official code: cyclomon/OneshotCLIP (66 stars, MIT, PyTorch, last commit 2023-06-05).

Abstract

There are many recent research efforts to fine-tune a pre-trained generator with a few target images to generate images of a novel domain. Unfortunately, these methods often suffer from overfitting or under-fitting when fine-tuned with a single target image. To address this, here we present a novel single-shot GAN adaptation method through unified CLIP space manipulations. Specifically, our model employs a two-step training strategy: reference image search in the source generator using a CLIP-guided latent optimization, followed by generator fine-tuning with a novel loss function that imposes CLIP space consistency between the source and adapted generators. To further improve the adapted model to produce spatially consistent samples with respect to the source generator, we also propose contrastive regularization for patchwise relationships in the CLIP space. Experimental results show that our model generates diverse outputs with the target texture and outperforms the baseline models both qualitatively and quantitatively. Furthermore, we show that our CLIP space manipulation strategy allows more effective attribute editing.

View on arXivView PDF

Code Implementations(1)

PyTorchMIT

Official Source code of "One-Shot Adaptation of GAN in Just One CLIP" IEEE Transactions on Pattern Anaylsis and Machine Intelligence (TPAMI)

661C++, CudaMar 10, 20223 years agoMIT
clippytorchstylegan

Similar Papers

Cite this paper

@article{kwon2022oneshot,
  title  = {One-Shot Adaptation of GAN in Just One CLIP},
  author = {Gihyun Kwon and Jong Chul Ye},
  year   = {2022},
  eprint = {2203.09301},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2203.09301}
}

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/2203.09301.svg)](https://codeofpaper.com/paper/2203.09301)

Discussion