E XEMPLAR G UIDED U NSUPERVISED I MAGE - TOI MAGE T RANSLATION WITH S EMANTIC C ONSISTENCY
A BSTRACT
Image-to-image translation has recently received signiﬁcant attention due to advances in deep learning.
Most works focus on learning either a one-to-one mapping in an unsupervised way or a many-to-many mapping in a supervised way.
However, a more practical setting is many-to-many mapping in an unsupervised way, which is harder due to the lack of supervision and the complex inner- and cross-domain variations.
To alleviate these issues, we propose the Exemplar Guided & Semantically Consistent Image-to-image Translation (EGSC-IT) network which conditions the translation process on an exemplar image in the target domain.
We assume that an image comprises of a content component which is shared across domains, and a style component speciﬁc to each domain.
Under the guidance of an exemplar from the target domain we apply Adaptive Instance Normalization to the shared content component, which allows us to transfer the style information of the target domain to the source domain.
To avoid semantic inconsistencies during translation that naturally appear due to the large inner- and cross-domain variations, we introduce the concept of feature masks that provide coarse semantic guidance without requiring the use of any semantic labels.
Experimental results on various datasets show that
EGSC-IT does not only translate the source image to diverse instances in the target domain, but also preserves the semantic consistency during the process.
[Figure]
[Figure]
I NTRODUCTION
Image-to-image (I2I) translation refers to the task of mapping an image from a source domain to a target domain, e.g . semantic maps to real images, gray-scale to color images, low-resolution to high-resolution images, and so on.
The recent advances in deep learning have greatly improved the quality of I2I translation methods for a number of applications, including super-resolution (Dong et al., 2014), colorization (Zhang et al., 2016), inpainting (Pathak et al., 2016), attribute transfer (Lee et al., 2018), style transfer (Gatys et al., 2016), and domain adaptation (Hoffman et al., 2018; Liu et al., 2017).
Most of these works (Isola et al., 2017; Wang et al., 2018; Zhu et al., 2017b) have been very successful in these cross-domain I2I translation tasks because they rely on large datasets of paired training data as supervision.
However, for many tasks it is not easy, or even possible, to obtain such paired data that show how an image in the source domain should be translated to an image in the target domain, e.g . in cross-city street view translation or male-female face translation.
For this unsupervised setting, Zhu et al.
(2017a) proposed to use a cycle-consistency loss, which assumes that a mapping from domain A to B, followed by its reverse operation approximately yields an identity
F A
F B x B
Z
E A
G B Exemplar-specific branch A ( ES A )
Exemplar-specific branch B ( ES B )
Domain A Domain B
InstanceNormalization content style semantic x AB
Gaussian prior γ B, β B x
[Figure]
Z
E A
G B
Guide Branch A ( GB A )
Guide Branch B ( GB B )
Domain A
Domain B
Transfer
Featuremask
AdaINparam content style semantic x
A x AB
Gaussianprior γ B, β B
AdaIN
Transfer
Weightssharing x B
Z
E A
G B
Domain A
Domain B
Transfer
Featuremask
AdaINparam c A
AdaIN param
Feature mask x A x AB
Gaussianprior 휸 B , 휷 B
AdaIN
Transfer
Weightssharing m A c̃
A x B
Z
E A G B
Domain A
Domain B Transfer Featuremask AdaINparam c
A
AdaIN param
Feature mask x A x AB
Gaussianprior 휸 B, 휷 B
AdaIN
Transfer
Weightssharing m
A c̃ A
Eq.
(3) f
A f B Eq.
(1) representation m domain function, that
F ( G (
A ))
≈ x A . Liu et al. (2017) further proposed a shared-latent space constraint, which assumes that a pair of corresponding images ( x A , x B ) from domains A and B respectively can be mapped to the same representation z in a shared latent space Z. Note that, all the aforementioned methods assume that there is a deterministic one-to-one mapping between the two domains, i.e . each image in A is translated to only a single image in B. By doing so, they fail to capture the multimodal nature of the image distribution within the target domain, e.g . different color and style of shoes in sketch-to-image translation and different seasons in synthetic-to-real street view translation.
In this work, we propose Exemplar Guided & Semantically Consistent I2I Translation (EGSCIT) to explicitly address this issue.
As shown in concurrent works (Gonzalez-Garcia et al., 2018;
Huang et al., 2018; Lee et al., 2018), we assume that an image is composed of two disentangled representations.
In our case, ﬁrst a domain-shared representation that models the content in the image, and second a domain-speciﬁc representation that contains the style information.
However, for a multimodal domain with complex inner-variations, as the ones we target in this paper, e.g . street views of day-and-night or different seasons, it is difﬁcult to have a single static representation which covers all variations in that domain.
Moreover, it is unclear which style (time-of-day/season) to pick during the image translation process.
To handle such multimodal I2I translations, some approaches (Almahairi et al., 2018; Gonzalez-Garcia et al., 2018; Lee et al., 2018) incorporate noise vectors as additional inputs to the generator, but as shown in (Isola et al., 2017; Zhu et al., 2017b)
this could lead to mode collapsing issues.
Instead, we propose to condition the image translation process on an arbitrary image in the target domain, i.e . an exemplar.
By doing so, EGSC-IT does not only enable multimodal ( i.e . many-to-many) image translations, but also allows for explicit control over the translation process, since by using different exemplars as guidance we are able to translate an input image into images of different styles within the target domain – see Fig. 1.
To instantiate this idea, we adopt the weight sharing architecture proposed in UNIT (Liu et al., 2017), but instead of having a single latent space shared by both domains, we propose to decompose the latent space into two components according to the two disentangled representations presented above.
That is, a domain-shared component that focuses on the image content, and a domain-speciﬁc component that captures the style information associated with the exemplar.
In our particular case, the domain-shared content component contains semantic information, such as the objects’ category, shape and spatial layout, while the domain-speciﬁc style component contains the style information, such as the color and texture, to be translated from a target domain exemplar to an image in the source domain.
To realize this translation, we apply adaptive instance normalization (AdaIN) (Huang & Belongie, 2017) to the shared content component of the source domain image using the AdaIN parameters computed from the target domain exemplar.
However, directly applying AdaIN to the feature maps of the shared content component would mix up all objects and scenes in the image, making the image translation prone to failure when an image contains diverse objects and scenes.
To tackle this problem, existing works (Gatys et al., 2017; Hoffman et al., 2018; Li et al., 2018;
Luan et al., 2017) use semantic labels as an additional form of supervision.
However, ground-truth semantic labels are not easy to obtain for most tasks as they require labor-intensive annotations.
Instead, to maintain the semantic consistency during image translation without using any semantic labels we propose to compute feature masks.
One can think of feature masks as attention modules that approximately decouple different semantic categories in an unsupervised way under the guidance of perceptual losses and adversarial loss.
In particular, one feature mask corresponding to a certain semantic category is applied to one feature map of the shared content component, and consequently the AdaIN for that channel is only required to capture and model the style difference for that category, e.g . sky’s style in two domains.
To the best of our knowledge, this is the ﬁrst line of work that addresses the semantic consistency issue under this setting.
See Fig.
2 for an overview of EGSC-IT.
Our contribution is three-fold.
i)
We propose a novel approach for the I2I translation task, which enables multimodal ( i.e . many-to-many) mappings and allows for explicit style control over the translation process.
ii)
We introduce the concept of feature masks for the unsupervised, multimodal
I2I translation task, which provides coarse semantic guidance without using any semantic labels.
iii)
Evaluation on different datasets show that our method is robust to mode collapse and can generate results with semantic consistency, conditioned on a given exemplar image.
R ELATED WORK
I2I translation.
I2I translation is used to learn a mapping from one image ( i.e . source domain) to another ( i.e . target domain).
Recently, with the advent of generative models (Goodfellow et al., 2014; Kingma & Welling, 2013), there have been a lot of works on this topic.
Isola et al.
(2017) proposed pix2pix to learn the mapping from input images to output images using a U-Net neural network in an adversarial way.
Wang et al.
(2018) extended the method to pix2pixHD, to turn semantic label maps into high-resolution photo-realistic images.
Zhu et al.
(2017b) extended pix2pix to BicycleGAN, which can model multimodal distributions and produce both diverse and realistic results.
All these methods, however, require paired training data as supervision which may be difﬁcult or even impossible to collect in many scenarios, such as synthetic-to-real street view translation or face-to-cartoon translation (Royer et al., 2017).
Recently, several unsupervised methods have been proposed to learn the mappings between two image collections without paired training data.
Note that, this is an ill-posed problem since there are inﬁnitely many mappings existing between two unpaired image domains.
To address this ill-posed problem, different constraints have been added to the network to regularize the learning process (Kim et al., 2017;
Liu et al., 2017; Royer et al., 2017; Yi et al., 2017; Zhu et al., 2017a).
One popular constraint is cycle-consistency, which enforces the network to learn deterministic mappings for various applications.
Going one step further, Liu et al.
(2017) proposed a shared-latent space constraint which encourages a pair of images from different domains to be mapped to the same representation in the latent space.
In a similar vein, Royer et al.
(2017) proposed to enforce a feature-level constraint with a latent embedding reconstruction loss.
However, we argue that these constraints are not well suited for complex domains with large inner-domain variations, as also mentioned in (Almahairi et al., 2018; Gonzalez-Garcia et al., 2018; Lee et al., 2018; Lin et al., 2018).
Unlike these methods, to address this problem we propose to add a target domain exemplar as guidance during image translation through AdaIN (Huang & Belongie, 2017).
As explained in the previous section, the
AdaIN technique is utilized to transfer the style component from the target domain exemplar to the shared content component of the source domain image.
This allows multimodal ( i.e . many-to-many) translations and can produce images of desired styles with explicit control over the translation process.
Concurrent to our work, MUNIT (Huang et al., 2018), also proposed to use AdaIN to transfer style information from the target domain to the source domain.
Unlike MUNIT, before applying AdaIN to the shared content component we compute feature masks to decouple different semantic categories and preserve the semantic consistency during the translation process.
In particular, by applying feature masks to the feature maps of the shared content component, each channel can specialize and model the style difference only for a single semantic category, which is crucial when handling domains with complex scenes.
Style transfer.
Style transfer aims at transferring the style information from an exemplar image to a content image, while preserving the content information.
The seminal work by Gatys et al.
(2016) proposed to transfer style information by matching the feature correlations, i.e . Gram matrices,
[Table]
in the convolutional layers of a deep neural network (DNN) following an iterative optimization process.
In order to improve the speed and ﬂexibility [Table] several feed-forward neural networks have been proposed.
Huang & Belongie (2017) proposed a simple but effective method, called AdaIN [Table] which aligns the mean and variance of the content image features with those of the style image features
Li et al.
(2017) proposed the whitening and coloring transform (WCT) algorithm [Table] which directly matches the features’ covariance in the content image to those in the given style image.
However, due to the lack of semantic consistency during translation [Table] these stylization methods usually generate non-photorealistic images, suffering from the “spills over” problem (Luan et al., 2017).
To address this, semantic label maps are used as additional supervision to help style transfer between corresponding semantic regions (Gatys et al., 2017;
Li et al., 2018; Luan et al., 2017).
Unlike these works, we propose to compute feature masks to approximately model such semantic information without using any semantic labels that are very hard to collect.
Table 1 summarizes the features of the most related works.
As can be seen, our method using the combination of AdaIN and feature masks under the guidance of perceptual loss is, to the best of our knowledge, the ﬁrst to achieve multimodal I2I translations in the unsupervised setting with high semantic consistency , without requiring any ground-truth semantic labels.
M ETHOD
Our goal is to learn a many-to-many mapping between two domains in an unsupervised way, which is guided by the style of an exemplar while retaining the semantic consistency at the same time.
For example, a synthetic street view image can be translated to either a day-time or night-time realistic scene, depending on the exemplar.
To realize this, similarly to concurrent works (Gonzalez-Garcia et al., 2018;
Huang et al., 2018; Lee et al., 2018)
we assume that an image can be decomposed into two disentangled components.
In our case, that is, one modeling the shared content between domains, i.e . domain-shared content component, and another modeling the style information speciﬁc to exemplars in the target domain, i.e . domain-speciﬁc style component.
In what follows, we present our EGSC-IT framework, the architecture of its networks, and the learning procedure.
3. F RAMEWORK
For simplicity, we present EGSC-IT in the A → B direction – see Fig. 2.
Each image domain ( i.e . source and target) is modeled by a VAE-GAN Larsen et al.
(2016), which includes an encoder
E A , a generator G A , and a discriminator D A .
For the B → A direction, the translation process as well as the notation are analogous.
Weight sharing for domain-shared content.
To learn the content component of an image pair that is shared across source and target domains we employ the weight sharing strategy proposed in UNIT (Liu et al., 2017).
The latter assumes that the two domains, A and B, share a common latent space, and any image pair from the two domains, x A and x B , can be mapped to the same latent representation in this shared-latent space z .
They achieve this by simply sharing the weights of the last layer in E A and E B as well as the ﬁrst layer in G A and G B .
For more details about the weight-sharing strategy we refer the reader to the original UNIT paper.
m A m B m A m B x
A c
A x AA x B c B x BB x A c A x AB x B c B x BA γ A , β A γ B , β B x B γ B , β B x A γ A , β A
Exemplar-based AdaIN for domain-speciﬁc style.
The shared content component contains semantic information, such as the objects’ category, shape and spatial layout, but no style information, e.g .
their color and texture.
Inspired by Huang & Belongie (2017), who showed that AdaIN’s afﬁne parameters have a big inﬂuence on the output image’s style, we propose to apply AdaIN to the shared content component before the decoding stage.
In particular, the exemplar from the target domain is fed to another network (see Fig.
2, blue line)
to compute a set of feature maps f B , which are expected to contain the style information of the target domain.
As in (Huang & Belongie, 2017), means and variances are calculated for each channel of f B and used as AdaIN’s afﬁne parameters,
[Equation]
where µ ( · ) and δ ( · ) respectively denote a function to compute the mean and variance across spatial dimensions.
The shared content component is ﬁrst normalized by these afﬁne parameters, as shown in
Eq. 2, and then decoded to a target-domain image using the target domain generator.
Since different afﬁne parameters normalize the feature statistics in different ways, by using different exemplar images in the target domain as input we can translate an image in the source domain to different sub-styles in the target domain.
Therefore, EGSC-IT does not only allow for multimodal I2I translations, but at the same time enables the user to have explicit style control over the translation process.
Feature masks for semantic consistency.
Directly applying AdaIN to the shared content component does not give satisfying results.
The reason is that one channel in the shared content component is likely to contain information from multiple objects and scenes.
The difference of these objects and scenes between the two domains is not always uniform, due to the large inner- and cross-domain variations.
As such, applying AdaIN over a feature map with complex semantics is prone to mix styles of different objects and scenes together, hence failing to give semantically-consistent translations.
To tackle this problem, existing works use semantic labels as an additional form of supervision.
However, ground-truth semantic labels are not easy to obtain for most tasks as they require laborintensive annotations.
Instead, we propose to compute feature masks (see Fig. 2, red line) to make an approximate estimation of semantic categories without using any ground-truth semantic labels.
The feature masks m A , which can be regarded as attention modules, are computed by applying a nonlinear activation function and a threshold to feature maps f A , i.e . m
A =
(1 − η ) · σ ( f A ) + η , where η is a threshold and σ is the sigmoid function.
Feature masks contain substantial semantic information, which can be used to retain the semantic consistency during translation, e.g . translating the source sky into the target sky style without affecting the other scene elements.
The new normalized representation ˜ c
A is ˜ c
A = AdaIN( m A ◦ c A , γ B , β B ) , where ◦ denotes the Hadamard product.
During training, there are four types of information ﬂow – see Fig. 3.
For the reconstruction ﬂow x A → x AA , the shared content component c A , feature masks m A , and AdaIN parameters γ A , β A are all computed from x A (and vice versa for x B → x BB ).
For the translation ﬂow x A → x AB , the shared content component c A and feature masks are computed from x A , while AdaIN’s afﬁne parameters γ B and β B are computed from the target domain exemplar x B (and vice versa for x B → x BA ).
3. N ETWORK ARCHITECTURE
The overall framework can be divided into several sub-networks 1 . 1) Two Encoders, E A and E B .
Each one consists of several strided convolutional layers and several residual blocks to compute the shared content component.
A feature mask network and an AdaIN network, F A and F B for
A → B translation (vise versa for B → A) have the same architecture as the Encoder above except for the weight-sharing layers.
3) Two Generators, G A and G B , are almost symmetric to the Encoders except that the up-sampling is done by transposed convolutional layers.
4) Two Discriminators, D A and D B , are fully-convolutional networks containing a stack of convolutional layers.
A VGG sub-network (Simonyan & Zisserman, 2015), V GG , that contains the ﬁrst few layers (up to relu5_1) of a pre-trained VGG-19 (Simonyan & Zisserman, 2015), which is used to calculate perceptual losses.
Note that, although we use UNIT as our baseline framework to build upon, this is not a hard restriction.
In theory, UNIT can be replaced with any baseline framework with similar functionality.
3. L EARNING
The learning procedure of EGSC-IT contains VAEs, GANs, cycle-consistency and perceptual losses.
To make the training more stable, we ﬁrst pre-train the feature mask network and AdaIN network for each domain separately within a VAE-GAN architecture, and use the encoder part as ﬁxed feature extractors, i.e . F A and F B , for the remaining training.
The overall loss is shown in Eq.
L
[Equation]
where the VAEs, GANs and cycle-consistency losses are identical to the ones used in Liu et al.
The perceptual loss consists of the content loss captured by V GG 19 feature maps φ containing localized spatial information, and the style loss captured by the Gram matrix containing non-localized style information similar to (Gatys et al., 2016; Johnson et al., 2016), is as follows, L [Equation] G
[Equation]
) where λ c and λ s are the weights for content and style losses, which depend on the dataset domain variations and tasks.
The content loss L c A ( E A , G A ) and style loss L s A ( E A , G A ) are deﬁned as, L c A ( E A , G A [Equation] [ (cid:107)
[Equation] ] , L [Equation] G
[Equation]
Gram
[Equation]
Gram [Equation] [Equation] We use the ﬁrst convolutional layer of the ﬁve blocks in V GG 19 to extract the feature maps.
L c B ( E B , G B ) and L s B ( E B , G B ) are deﬁned likewise.
For the content losses L c A and L c B , a linear weighting scheme is adopted to help the network focus more on the high-level semantic information.
In both content and style losses we use the L1 distance, which in our experiments outperforms L2.
Now that we have introduced all losses, we can explain how these losses help to achieve I2I translation, multimodal translation, and semantic consistency.
I2I translation : L VAE , L GAN and L CC help to maintain the shared latent space by relating the two different domains and ﬁnding the optimal translation between the two in an unsupervised way.
Multimodal translation : L S and L GAN help to encourage x AB to look not only like the main mode of variation in domain B, but also like an exemplar from domain B, x B , since the domain space is actually supported by each data sample.
Semantic consistency : L C encourages the network to utilize the feature mask information for semantic consistency, without relying on hard correspondences between semantic labels as existing works do.
E XPERIMENTS
We evaluate EGSC-IT’s translation ability, i.e .
how well it generates domain-realistic-looking and semantically consistent images, both qualitatively and quantitatively on three tasks with progressively increasing visual complexity: 1) single-digit translation; 2) multi-digit translation; 3) street view translation.
We ﬁrst perform an ablation study on various components of EGSC-IT on the single-digit translation task.
Then, we present results on more challenging translation tasks, and evaluate EGSCIT quantitatively on the semantic segmentation task.
In supplementary material, we apply EGSC-IT to the face gender translation task and perform the ablation study on the street-view translation task.
Single-digit translation.
We set up a controlled experiment on the MNIST-Single dataset, which is created based on the handwritten digits dataset MNIST LeCun et al.
The MNIST-Single dataset consists of two different domains as shown in Fig.
For domain A of both training/test sets, the foreground and background are randomly set to black or white but different from each other.
For domain B of training set, the foreground and background for digits from 0 to 4 are randomly assigned
A B AB BA
MUNIT
EG-UNIT ab ba
EG-UNIT w/o feature mask ab ba
CycleGAN ab ba
UNIT ab ba ab ba
EG-UNITw/o AdaIN EG-UNIT w/o perceptual loss ab ba ab ba
A B
EGSC-IT
AB BA
EGSC-IT w/o feature mask CycleGAN
AB BA
UNIT
AB BA AB BA
EGSC-ITw/o AdaIN
EGSC-IT
AB BA AB BA w/o
GT
AB BA
GT
Neural ST
AB BA
AdaIN* [13] AB BA AdaIN [13] AB BA
WCT* [23] AB BA WCT [23] AB BA alpha=1.
alpha=0.
MUNIT
AB BA
Neural ST [7]
AdaIN* [13] AdaIN [13]
WCT* [23] WCT [23]
WCT*
AB BA
AdaIN*
AB BA
[Table]
a color from (cid:8) red, green, blue (cid:9) , and the foreground and background for digits from 5 to 9 are ﬁxed to red and green , respectively.
For domain B of testing set, the foreground and background of all digits are randomly assigned a color from (cid:8) red, green, blue (cid:9) .
Such data imbalance is designed on purpose to test the translation diversity and generalization ability.
In particular, for diversity, we want to check whether a method would suffer from the mode collapse issue and translate the images to the dominant mode, i.e . (red, green) , while for generalization, we want to check whether the model can be applied to new styles in the target domain that never appear in the training set, e.g . translate number 6 from black foreground and white background to blue foreground and red background.
We ﬁrst analyze the importance of three main components of EGSC-IT, i.e . feature masks, AdaIN, and perceptual loss, on the MNIST-Single dataset.
As shown in Fig.
4, EGSC-IT can successfully transfer the source image into the style of the exemplar image.
Ablating the feature mask from
EGSC-IT, leads to incorrect foreground and background shape, indicating that feature masks can indeed provide semantic information to transfer the corresponding local regions.
Without AdaIN, the network suffers from the mode collapse issue in A → B translation, i.e .
all samples are transferred to the dominant mode with red foreground and green background, indicating that the exemplar’s style information can help the network to learn many-to-many mappings and avoid the mode collapse issue.
Without perceptual losses L P , colors of foreground and background are incorrect, which shows that perceptual losses can encourage the network to learn semantic knowledge, in this case foreground and background, without ground-truth semantic labels.
As for other I2I translation methods, CycleGAN (Zhu et al., 2017a) and UNIT (Liu et al., 2017) can only do deterministic image translations and suffer from mode collapse issue, such as white ↔ green and black ↔ red for CycleGAN in Fig. 4.
MUNIT (Huang et al., 2018) can successfully transfer the style of the exemplar image, but the foreground and background are mixed and the digit’s shape is not kept well.
These qualitative observations are in accordance with the quantitative results in Tab. 2, where our full EGSC-IT obtains higher SSIM scores than all other alternatives.
In addition, we compare with other style transfer methods, Neural ST (Gatys et al., 2016), AdaIN (Huang & Belongie, 2017), and WCT (Li et al., 2017).
In each case, we resize the input image to 512 × 512 resolution and choose the best performing hyper-parameters.
Note how style transfer methods can transfer the style successfully but fail to keep semantic consistency.
Quantitative results for style transfer methods are in supplementary material.
To verify EGSC-IT’s ability to match the target domain distributions of real data and translated results, we visualize them using t-SNE embeddings (Maaten & Hinton, 2008) in Fig.
The t-SNE embeddings are calculated from the translated images with PCA dimension reducing.
Our method can match the distributions well, while others either collapse to few modes or mismatch the distributions.
Multi-digit translation.
The MNIST-Multiple dataset is another controlled experiment designed to mimic the complexity in real-world scenarios.
It is used to test whether the network understands the semantics, i.e . digits, in an image and translates each digit accordingly.
Each image in MNISTMultiple contains all ten digits, which are randomly placed in 4 × 4 grids.
Two domains are designed: in domain A, the foreground and background are randomly set to black or white , but different from
[Figure]
A B
MUNIT EG-UNIT ab ba
EG-UNIT w/o feature mask ab ba
CycleGAN ab ba
UNIT ab ba ab ba
EG-UNIT w/o AdaIN
EG-UNIT w/o perceptual loss ab ba ab ba
[Figure]
each other; in domain B, the background is randomly assigned to either black or white and each foreground digit is assigned to a certain color, but with a little saturation and lightness perturbation.
Our goal is to encourage the network to understand the semantic information, i.e .
the different digits and backgrounds, when translate an image from domain A to domain B.
That is, a successfully translated image should have the content of domain A, the digit class, and the style of domain B, the digit and background colors respectively.
This experiment is quite challenging, but we observe that our model can still obtain good results without the need for ground-truth semantic labels or paired data.
For example, in Figure 6 top row the digits 1,2,3,4,6 can be successfully translated given the criteria described above.
As seen in Fig. 6, MUNIT can not translate the foreground color with semantic consistency, and the colors look more “fake”.
[Table]
Street view translation.
We carry out a synthetic ↔ real experiment for street view translation between GTA5 (Richter et al., 2016) and Berkeley Deep Drive (BDD)
(Xu et al., 2017) datasets.
The street view datasets are more complex than the digit ones (different illumination/weather conditions, complex environments).
As shown in Fig.
7, our method can successfully translate the images from the source to the target domain according to the exemplar’s style.
For small variations, e.g . day → day (ﬁrst row), MUNIT can keep up, however for large variations, e.g . day → night and vice versa (second row), which is exactly the problem we examine in this paper, only
EGSC-IT can successfully translate details like the proper sky color and illumination condition w.r.t.
the exemplar.
Similar to FCN-score used by Isola et al.
(2017), we also use the semantic segmentation performance to quantitatively evaluate the image translation
[Figure]
[Table]
Cityscapes GTA5→Cityscapess GTA5←Cityscapes
BDD GTA5→BDD
GTA5←BDD
MUNIT
GTA5→BDD GTA5←BDD
GTA5→Cityscapes GTA5←Cityscapes
EGSC-IT quality.
We ﬁrst translate images in GTA5 dataset to an arbitrary image in BDD dataset.
We only generate images of size 256 × 512 due to the limitation on GPU memory.
Then, we train a single-scale
Deeplab model (Chen et al., 2018) on the translated images and test it on BDD test set.
The mean
Intersection over Union (mIoU) scores in Tab.
3 show that training with our translated synthetic images can improve the segmentation results, which indicates that our method can indeed reasonably translate the source GTA5 image to the target domain style with semantic consistency and reduce the domain difference successfully.
D ISCUSSIONS
Since our method does not use any semantic segmentation labels nor paired data, there are some artifacts in the results for some hard cases.
For example, as to the street view translation, day → night and night → day ( e.g . Fig.
7 bottom row) are more challenging than day → day ( e.g . Fig.
7 top row).
As a result, it is sometimes hard for our model to understand the semantics in such cases.
In the future, it would be interesting to extend our method to the semi-supervised setting in order to beneﬁt from the presence of some fully-labeled data.
C ONCLUSION
We introduced the EGSC-IT framework to learn a multimodal mapping across domains in an unsupervised way.
Under the guidance of an exemplar from the target domain, we showed how to combine AdaIN with feature masks in order to transfer the style of the exemplar to the source image, while retaining semantic consistency at the same time.
Numerous quantitative and qualitative results demonstrate the effectiveness of our method in this particular setting.
A CKNOWLEDGMENTS
We gratefully acknowledge the support of Toyota Motors Europe, FWO Structure from Semantics project, and KU Leuven GOA project CAMETRON.
A S UPPLEMENTARY MATERIAL
This supplementary material includes more experimental results (§A.1), as well as additional implementation details regarding the network architecture and training (§A.2).
A. E XTRA EXPERIMENT RESULTS
Face gender translation.
The Large-scale CelebFaces Attributes (CelebA) dataset (Liu et al., 2015) is a large- scale face attributes dataset with more than 200K celebrity images.
We divide the aligned face images into male and female domains, containing 84,434 and 118,165 images respectively.
We perform face gender translation on this dataset to show how the proposed method can be generalized to tasks with attributes as styles.
From Fig.
8, we observe that EGSC-IT can translate the face gender successfully, and most importantly transfer the style of hair, skin and background according to the given exemplar image, unlike MUNIT.
In addition, we also provide the male → female face translation results matrix Fig.
We observe that the output image’s content is consistent with the source image’s and its style is consistent with the target image’s.
Such observation can reﬂect how the latent space changes given different input images, i.e .
the the content latent is only related to source image since the style information is combined in the decoder part through feature mask and AdaIN techniques.
Letter-digit translation.
To further evaluate the generalization ability of our method, we use EMNIST (Cohen et al., 2017) for three translation tasks: 1) black-white letter ↔ colored digit; 2) black-white digit ↔ colored letter; 3) black-white letter ↔ colored letter.
EMNIST dataset, a set of handwritten character digits, has the same image format and dataset structure with MNIST dataset.
We apply the same process to EMNIST letter images as that for MNIST single-digit images in the main paper.
As shown in 10, our model trained with only single-digit data can successfully generalize to the letter data.
Single-digit translation.
For the quantitative comparison, we also report the SSIM socre of style transfer meth- ods in Tab. 4.
In addition, the larger size version of the single-digit translation t-SNE embedding visualization as shown in Fig.
[Table]
Multi-digit translation.
The setting of this experiment was presented in the main paper.
Here, we provide more details on the results.
As seen in Fig.
12, both CycleGAN and UNIT can not translate the foreground and background color accordingly, and the colors in CycleGAN look more “fake”.
This is due to the fact that CycleGAN and UNIT only learn a one-to-one mapping.
These observations are consistent with the SSIM score in Tab. 5, where both CycleGAN and UNIT have much lower SSIM scores.
Differently, MUNIT can not translate the foreground color with semantic consistency, and the colors look more “fake”.
These observations are consistent with the visualization of t-SNE embeddings.
As to the SSIM score, MUNIT seems comparable to ours although visually it is performing worse.
The probable reason is that MUNIT can mostly translate the background successfully which occupies the majority of the image.
Street view translation.
We also provide a larger size version of the results for GTA5 ↔ BDD translation as shown in Fig.
In addition, we also provide the ablation study results in Fig.
We observe that: 1) removing feature mask will lead to color mismatches or inaccuracies ( e.g . Fig.
15(a) 1st row 3rd col); 2) removing AdaIN will reduce the model to unimodality ( e.g .
all images are translated to a sunny day with blue sky, see Fig.
15(a) 4th col) since the output image’s style is not guided by the exemplar image; 3) removing perceptual loss will lead to incorrect style ( e.g . Fig.
15(b) 5th col)
and the color will spread even given the feature mask since there is no perceptual feedback during training ( e.g . Fig.
15(a) 5th col).
[Table]
A. I MPLEMENTATION DETAILS
The network architecture and training parameters are listed in Tab. 6.
We set the number of down-sampling and up-sampling convolutional layers n 1 = 1 in single-digit translation and n 1 = 3 in other translation experiments.
Following UNIT (Liu et al., 2017), the number of residual blocks in Encoder and Generator is set to n 2 = 4 with one sharing layer, and the number of convolutional layers in discriminator is set to n 3 = 5 .
The threshold parameter η is used to adjust how much the feature mask affects the information ﬂow.
Setting η = 0 , i.e . using the feature maps (paper Fig.
2 top branch) as feature mask directly, leads to useful information being dropped out and artifacts in the results.
Setting η = 1 , i.e . not using feature mask at all, leads to results without semantic consistency (see paper Fig. 4).
After experimenting with different values, we ﬁxed it to 0 .
We use the Adam Kingma & Ba (2015) optimizer with β 1 = 0 .
5 and β 2 = 0 .
The learning rate is polynomially decayed with a power of 0.9, as mentioned in (Chen et al., 2018).
In order to keep training stable, we update encoder and generator 5 times, and discriminator 1 time in each iteration.
The loss weights in L UNIT are following (Liu et al., 2017), and λ c , λ s are chosen according to the dataset variations and tasks.
For data augmentation, we do left-right ﬂip and random crop.
In addition, we set a low λ c for face gender translation, since we need to change the shape and add/remove hair in this translation task.
[Table]
[Figure]
[Figure]
Female→Male 182659 182657 182645 182669182644
Male→Female
EGSC-IT
MUNIT MUNIT
EGSC-IT
[Figure]
[Figure]
emnist_BW_letter_mnist
A
B ab ba emnist_BW_mnist_letter
A
B ab ba emnist_BW_letter_letter
A
B ab ba
A
B
AB
BA
B
AB
BA A
B
AB
BA A
[Figure]
MUNIT
EGSC-IT EGSC-IT w/o feature mask
CycleGAN
UNIT
EGSC-IT w/o AdaIN w/o
A→B
B→A pre30_early10_lr200_initpca_methodbarnes_hut
EGSC-IT
IT w/o feature mask
EGSC-IT w/o AdaIN
EGSC-IT w/o
A→B
B→A pre30_early40_lr200_initrandom_methodbarnes_hut
EGSC-IT
A→B B→A
A B
MUNIT EG-UNIT ab ba
EG-UNIT w/o feature mask ab ba
CycleGAN ab ba
UNIT ab ba ab ba
EG-UNIT w/o AdaIN
EG-UNIT w/o perceptual loss ab ba ab ba
[Figure]
A→B B→A
MUNIT
EGSC-IT CycleGAN
UNIT
MUNIT
EG-UNIT EG-UNIT w/o feature mask
CycleGAN
UNIT
EG-UNIT w/o AdaIN
EG-UNIT w/o
Multi-digit
A→B
Multi-digit B→A
PCA50 + random init
[Figure]
A→B B→A
MUNIT
EGSC-IT CycleGAN
UNIT
BDD GTA
Cityscapes
GTA5→Cityscapes GTA5→BDD GTA5←Cityscapes GTA5←BDD EG-UNIT MUNIT MUNIT EG-UNIT MUNIT EG-UNIT MUNIT EG-UNIT GTA
Cityscapes
GTA5→Cityscapes GTA5←Cityscapes s BDD GTA5→BDD GTA5←BDD
GTA
BDD
GTA5→BDD
GTA5←BDD
MUNIT
GTA5→BDD
GTA5←BDD
EGSC-IT
GTA5 Cityscapes GTA5→Cityscapes s
GTA5←Cityscapes
BDD GTA5→BDD GTA5←BDD
MUNIT
GTA5→BDD GTA5←BDD
GTA5→Cityscapes GTA5←Cityscapes
EGSC-IT
Data
[Figure]