Machine|Paper Notes: Cross-Domain Image Translation Based on GAN
1 GAN
1.1 Introduction
To learn the generator’s distributionpgp gover data x, we define a prior on input noise variablespz(z)p z ( z ) , then represent a mapping to data space asG(z;
θg)G ( z ;
θ g ) , whereGGis a differentiable function represented by a multilayer perceptron with parameters θg θ g . We also define a second multilayer perceptronD(x;
θd)D ( x ;
θ d )that outputs a single scalar.D(x)D ( x )represents the probability thatxxcame from the data rather than pg p g . We trainDDto maximize the probability of assigning the correct label to both training examples and samples from G G . We simultaneously trainGGto minimize log(1?D(G(z))) l o g ( 1 ? D ( G ( z ) ) ) . In other words,DDand G Gplay the following two-player mini-max game with value functionV(G;
D):V ( G ;
D ) :
minGmaxDV(D;
G)=Ex~pdata(x)[logD(x)]+Ez~pz(z)[log(1?D(G(z)))]m i n G m a x D V ( D ;
G ) = E x ~ p d a t a ( x ) [ l o g D ( x ) ] + E z ~ p z ( z ) [ l o g ( 1 ? D ( G ( z ) ) ) ]
1.2 Theory Analysis
2 Cycle GAN(ICCV 2017)
2.1 Task: Cross Domain Image Translation Image-to-image translation is a class of vision and graphics problems where the goal is to learn the mapping between an input image and an output image.
In this paper, we present a method that can learn to do the same: capturing special characteristics of one image collection and figuring out how these characteristics could be translated into the other image collection, all in the absence of any paired training examples.
2.2 Model: Cycle consistent
Loss:
2.3 Experiment
- Dataset: Cityscapes dataset , map and aerial photo on data scraped from Google Maps
- Metrics: AMT perceptual studies, FCN score, Semantic segmentation metrics
- Result:
- On translation tasks that involve color and texture changes, like many of those reported above, the method often succeeds. We have also explored tasks that require geometric changes, with little success.
- Some failure cases are caused by the distribution characteristics of the training datasets.
- We also observe a lingering gap between the results achievable with paired training data and those achieved by our unpaired method.
3 DIAT: Deep Identity-aware Transfer of Facial Attributes 3.1 Task: Identity-aware Transfer of Facial Attributes
Our DIAT and DIAT-A models can provide a unified solution for several representative facial attribute transfer tasks such as expression transfer, accessory removal, age progression, and gender transfer
3.2 Model
In this section, a two-stage scheme is developed to tackle the identity-aware attribute transfer task.
- Face transform network
Loss:
- Face Enhancement Network
Loss:
In DIAT, the perceptual identity loss is defined on the pre-trained VGG-Face. Actually, it may be more effective to define this loss on some CNN trained to attribute transfer. Here we treat identity-preserving and attribute transfer as two related tasks, and define the perceptual identity loss based on the convolutional features of the discriminator. By this way, the network parameters for identity loss will be changed along with the updating of discriminator, and thus we named it as adaptive perceptual identity loss.
3.4 Experiments
Dataset: a subset of the aligned CelebA dataset
4 Unsupervised Cross-Domain Image Generation(ICLR 2017 ) 4.1 Task
Recent achievements replicate some of these capabilities to some degree: Generative Adversarial Networks (GANs) are able to convincingly generate novel samples that match that of a given training set; style transfer methods are able to alter the visual style of images; domain adaptation methods are able to generalize learned functions to new domains even without labeled samples in the target domain and transfer learning is now commonly used to import existing knowledge and to make learning much more efficient.
These capabilities, however, do not address the general analogy synthesis problem that we tackle in this work. Namely, given separated but otherwise unlabeled samples from domainsSSand T Tand a perceptual functionff , learn a mapping G:S→T G : S → Tsuch thatf(x)~f(G(x)f ( x ) ~ f ( G ( x )
As a main application challenge, we tackle the problem of emoji generation for a given facial image. Despite a growing interest in emoji and the hurdle of creating such personal emoji manually, no system has been proposed, to our knowledge, that can solve this problem. Our method is able to produce face emoji that are visually appealing and capture much more of the facial characteristics than the emoji created by well-trained human annotators who use the conventional tools.
4.2 Model
Loss:
- DDis a ternary classification function from the domain T Tto 1,2,3, andDi(x)D i ( x )is the
probability it assigns to classi=1,2,3i = 1 , 2 , 3for an input samplexx - During optimization, LG L Gis minimized overggand LD L Dis minimized overDD
- LCONST L C O N S Tenforces f-constancy forx∈Sx ∈ S , whileLTIDL T I Denforces that for samplesx∈Tx ∈ T
- LTVL T Vis an anisotropic total variation loss, which is added in order to slightly smooth the resulting image
- ffis trained use other datasets before training this model
Dataset:
- Street View House Number (SVHN) dataset to the domain of the MNIST dataset
- FROM PHOTOS TO EMOJI
文章图片
5 StarGAN: Multi-Domain Image-to-Image Translation 5.1 Introduction
Recent studies have shown remarkable success in image-to-image translation for two domains. However, existing approaches have limited scalability and robustness in handling more than two domains, since different models should be built independently for every pair of image domains.
To address this limitation, we propose StarGAN, a novel and scalable approach that can perform image-to-image translations for multiple domains using only a single model
We can further extend to training multiple domains from different datasets.
### 5.2 Model
Loss:
- a domain classification loss of real images(Lrcls L c l s r ) used to optimize D, and a domain classification loss of fake images( LfclsL c l s f ) used to optimize G
- UseLrecL r e cto guarantee that translated images preserve the content of its input images while changing only the domain-related part of the inputs.
5.3.1 Mask Vector
In StarGAN, we use an n-dimensional one-hot vector to represent m, with n being the number of datasets. andcic irepresents a vector for the labels of theii -th dataset. The vector of the known label ci c ican be represented as either a binary vector for binary attributes or a one-hot vector for categorical attributes
5.3.2 Training Strategy When training StarGAN with multiple datasets, we use the domain labelc~c ~defined at above as input to the generator. By doing so, the generator learns to ignore the unspecified labels, which are zero vectors, and
focus on the explicitly given label. The structure of the generator is exactly the same as in training with a single dataset, except for the dimension of the input labelc~c ~ .
5.3.3 CelebA and RaFD dataset demo
5.4 Experiments
Dataset: CelebA, RaFD
Metrics: AMT(human evaluation)
文章图片
Dataset: RaFD dataset (90%/10% splitting for training and test sets)
Metrics: compute the classification error of a facial expression on synthesized images
6 Pix2Pix: Image-to-Image Translation with Conditional Adversarial Networks(use paired data)(CVPR 2017) 6.1 Introduction
We investigate conditional adversarial networks as a general-purpose solution to image-to-image translation problems. These networks not only learn the mapping from input image to output image, but also learn a loss function to train this mapping. This makes it possible to apply the same generic approach to problems that traditionally would require very different loss formulations.
we no longer hand-engineer our mapping functions, and this work suggests we can achieve reasonable results without hand-engineering our loss functions either
(One architecture to different works)
6.2 Model
6.2.1 Generator with skips
6.2.2 Conditional GANs
文章图片
6.2.3 PatchGAN It is well known that the L2 loss and L1produce blurry results on image generation problems . Although these losses fail to encourage high-frequency crispness, in many cases they nonetheless accurately capture the low frequencies .
In order to model high-frequencies, it is sufficient to restrict our attention to the structure in local image patches. Therefore, we design a discriminator architecture – which we term a PatchGAN – that only penalizes structure at the scale of patches. This discriminator tries to classify if each N × N patch in an image is real or fake. We run this discriminator convolutionally across the image, averaging all responses to provide the ultimate output of D
6.2.4 Loss
6.3 Experiments
Dataset:
- Semantic labels$photo, trained on the Cityscapes dataset.
- Architectural labels!photo, trained on CMP Facades
- Map to aerial photo, trained on data scraped from Google Maps.
- BW to color photos, trained on [50 Imagenet large scale visual recognition challenge].
- Edges to photo, trained on data from [64 Generative visual manipulation on the natural image manifold] and [59 Fine-Grained Visual Comparisons with Local Learning ]; binary edges generated using the HED edge detector [57 Holistically-nested edge detection ] plus post processing.
- Sketch to photo: tests edges to photo models on human drawn sketches from [18 How do humans sketch objects].
- Day to night, trained on [32 Transient attributes for high-level understanding and editing of outdoor
scenes ]. - Thermal to color photos, trained on data from [26 Multispectral pedestrian detection: Benchmark dataset and baseline].
- Photo with missing pixels to inpainted photo, trained on Paris StreetView from [13 What makes paris look like paris]
7 Photo-Realistic Single Image Super-Resolution Using a GAN(use paired data to train)(CVPR 2017) 7.1 Task
Despite the breakthroughs in accuracy and speed of single image super-resolution using faster and deeper convolutional neural networks, one central problem remains largely unsolved: how do we recover the finer texture details when we super-resolve at large upscaling factors?
Recent work has largely focused on minimizing the mean squared reconstruction error. The resulting estimates have high peak signal-to-noise ratios, but they are often lacking high-frequency details and are perceptually unsatisfying in the sense that they fail to match the fidelity expected at the higher resolution.
To our knowledge, it is the first framework capable of inferring photo-realistic natural images for 4× upscaling factors. To achieve this, we propose a perceptual loss function which consists of an adversarial loss and a content loss
7.2 Model
文章图片
Loss:
- φi,jφ i , jinlSRVGG/i,jl V G G / i , j S R, we indicate the feature map obtained by the j-th convolution (after activation) before the i-th max pooling layer within the VGG19 network
- D network is optimized by the min-max game
- G network is optimized by the losslSRl S R
Dataset:
- Set5 [Low-complexity single-image super-resolution based on nonnegative neighbor embedding ],
- Set14 [On single image scale-up using sparse-representations ]
- BSD100
- the testing set of BSD300
文章图片
8 Conclusion 8.1 Reason for using GAN
- Difficulties of traditional methods
- How to design effective loss
- How to use unpaired data
- GAN’s advantages
- No need of the specific loss, but a high level goal
- Able to handle unpaired data
- GAN’s disadvantages
- The Generator network often produce insensitive results
- Mode collapse: all inputs are mapped to the same output
?
- GAN Loss: keep high level domain feature
- Keep specific entity feature
- Given separated but otherwise unlabeled samples from domainsSSand T Tand a perceptual functionff , learn a mapping G:S→T G : S → Tsuch thatf(x)~f(G(x))f ( x ) ~ f ( G ( x ) )
- Perceptual Loss
- pre-trained f
- Cycle consistency
- Enhancement network
- Given separated but otherwise unlabeled samples from domainsSSand T Tand a perceptual functionff , learn a mapping G:S→T G : S → Tsuch thatf(x)~f(G(x))f ( x ) ~ f ( G ( x ) )
- Translations for multiple domains using only a single model
- Human evaluation: AMT, MOS
- Visualizing the generated results
- Use a model in the target domain to evaluate: FCN Scores(MNIST classifiers, VGG face classifier)
推荐阅读
- 大数据|python自然语言处理库_8个出色的Python库用于自然语言处理
- 手把手写深度学习|手把手写深度学习(10)(用Pix2Pix GANs实现sketch-to-image跨模态任务(理论基础))
- AI比赛教程|人工智能之自然语言处理技术总结与展望
- 目标检测|目标检测算法的分类和优缺点
- 百度飞桨|基于PaddleX的岩石识别
- java|Meta元宇宙OS要黄(300人研发团队解散,关闭VR/AR操作系统研发)
- 大数据|开始即结束,Meta元宇宙OS要黄()
- 人工智能|Meta 被爆暂停 AR/VR 操作系统项目,元宇宙能否推进()
- java|改名Meta俩月,脸书放弃虚拟现实操作系统(负责人跳槽谷歌)