Contrastive Learning for Unpaired Image-to-Image Translation

Image-to-image translation, where each patch in the output reflects the content of the corresponding patch in the input, independent of domain

Released in: Contrastive Learning for Unpaired Image-to-Image Translation

Source: Contrastive Learning for Unpaired Image-to-Image Translation

Contributor:

Summary

Author states that in image translation settings, each patch in the output should reflect the content of the corresponding patch in the input, independent of domain. To achieve this, they propose a straightforward method — maximizing mutual information between the two, using a framework based on contrastive learning.

The method encourages two elements (corresponding patches) to map to a similar point in a learned feature space, relative to other elements (other patches) in the dataset, referred to as negatives.  explore several critical design choices for making contrastive learning effective in the image synthesis setting. Notably, we use a multilayer, patch-based approach, rather than operate on entire images. Furthermore, we draw negatives from within the input image itself, rather than from the rest of the dataset. We demonstrate that our framework enables one-sided translation in the unpaired image-to-image translation setting, while improving quality and reducing training time. In addition, our method can even be extended to the training setting where each “domain” is only a single image.

2020

Year Released

Key Links & Stats

taesungp / contrastive-unpaired-translation

Contrastive Learning for Unpaired Image-to-Image Translation

License

Contrastive Learning for Unpaired Image-to-Image Translation

@inproceedings{park2020cut, title={Contrastive Learning for Unpaired Image-to-Image Translation}, author={Taesung Park and Alexei A. Efros and Richard Zhang and Jun-Yan Zhu}, booktitle={European Conference on Computer Vision}, year={2020} }

ML Tasks

  1. Domain Adaptation
  2. Image Generation

ML Platform

  1. Pytorch

Modalities

  1. General

Verticals

  1. General

CG Platform

  1. Not Applicable

Related organizations

UC Berkeley

Adobe Research