Image-to-image translation has gained significant interest due to the success of deep learning models that enforce cycle consistency constraints. However, the recent studies are particularly limited to a subset of domains with significant constraints on style or texture variations. Also, these models show limited performance in multi-domain settings where one image is translated to numerous domains. We propose a Constrained Unsupervised Image-to-Image Generative Adversarial Network (CUNIT-GAN) to address the above problems. It consists of an asymmetric Auto-encoder (AE) based Generator network and a dual-purpose Discriminator network that detects real and fake samples and classifies the input image. This study focuses on enhancing the explainability and representation power of the multidomain latent space through our novel latent contrastive loss, which leads to the clustering of class-level feature embeddings and decoupling of latent space. The effectiveness of CUNIT-GAN is demonstrated through a comprehensive qualitative and quantitative analysis conducted on benchmark multi-domain image datasets.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

CUNIT-GAN: Constraining Latent Space for Unsupervised Multi-domain Image-to-Image Translation via Generative Adversarial Network

  • Krishanu Saini,
  • Anikeit Sethi,
  • Rituraj Singh,
  • Aruna Tiwari,
  • Sumeet Saurav,
  • Sanjay Singh

摘要

Image-to-image translation has gained significant interest due to the success of deep learning models that enforce cycle consistency constraints. However, the recent studies are particularly limited to a subset of domains with significant constraints on style or texture variations. Also, these models show limited performance in multi-domain settings where one image is translated to numerous domains. We propose a Constrained Unsupervised Image-to-Image Generative Adversarial Network (CUNIT-GAN) to address the above problems. It consists of an asymmetric Auto-encoder (AE) based Generator network and a dual-purpose Discriminator network that detects real and fake samples and classifies the input image. This study focuses on enhancing the explainability and representation power of the multidomain latent space through our novel latent contrastive loss, which leads to the clustering of class-level feature embeddings and decoupling of latent space. The effectiveness of CUNIT-GAN is demonstrated through a comprehensive qualitative and quantitative analysis conducted on benchmark multi-domain image datasets.