Hippocampus's Garden

Under the sea, in the hippocampus's garden...

Awesome StyleGAN Applications | Hippocampus's Garden

Awesome StyleGAN Applications

November 12, 2021  |  10 min read  |  3,122 views

  • このエントリーをはてなブックマークに追加

Since its debut in 2018, StyleGAN attracted lots of attention from AI researchers, artists and even lawyers for its ability to generate super realistic high-resolution images of human faces. At the time of this writing, the original paper [1] has 2,548 citations and its successor StyleGAN2 [2] has 1,065. This means that the number of application works is growing at a dizzying rate. To keep me from getting overwhelmed, I’m writing this post to overlook this research field in an organized way. Of course, it’s impossible to read thousands of papers, so hereafter I’ll focus on the papers that relate to image manipulation including GAN inversion and 3D control.

StyleGANs: A Quick Recap

First of all, let’s briefly recall what StyleGAN was and what the key updates of its subsequent versions were.

StyleGAN

The architecture of the original StyleGAN generator was novel in three ways:

  • Generates images in two-stages; first map the latent code to an intermediate latent space with the mapping network and then feed them to each layer of the synthesis network, rather than directly inputs the latent code to the first layer only.
  • Applies AdaIN [3] layers iteratively to capture the intermediate latent code as “styles”.
  • Injects explicit noise inputs to deal with stochastic variations such as hair and freckles.

These are best described in the figure below.

2021 11 07 22 46 00

Image taken from [1].

With these features combined, StyleGAN can generate images that are almost impossible to discriminate for humans.

2021 11 07 22 46 38

Image taken from [1].

This quality was so amazing that many people rushed to train StyleGAN with their own datasets to generate cats, ukiyoe, Pokémons, and more (see Awesome Pretrained StyleGAN for details).

StyleGAN has another interesting feature called ”style mix”. With two latent codes zA\mathbb{z}_A and zB\mathbb{z}_B (and corresponding wA\mathbb{w}_A and wB\mathbb{w}_B), one can switch the inputs from wB\mathbb{w}_B to wA\mathbb{w}_A in the middle of the synthesis network and get a mixed image that has B’s coarse styles and A’s fine styles.

2021 11 07 22 55 04

Image taken from [1].

StyleGAN2

The human face images generated by StyleGAN look convincing enough, but if you have a careful look, you may notice some unnatural artifacts. In StyleGAN2 [2], some architectural optimizations were made to StyleGAN to facilitate even more realistic generation, though I don’t go into the technical details here.

One of the most important updates of StyleGAN2 is that by regularizing the perceptual path length, it became easier to invert. Now it is possible to encode a given image (either generated or real) into the intermediate style space W\mathcal{W}.

2021 11 07 23 29 09

Image taken from [2].

This paved the way for GAN inversion — projecting an image to the GAN’s latent space where features are semantically disentangled, as is done by VAE. As we’ll see in the next section, StyleGAN2 is currently the most widely used version in terms of the number of application works.

StyleGAN3 (Alias-Free GAN)

In June 2021, the Tero Karras team published Alias-Free GAN (later named StyleGAN3) to address the undesirable aliasing effect that leads to some details glued to the absolute coordinates of the image [4].

A video is worth a thousand words. The official video clearly demonstrates the “texture sticking” issue and how StyleGAN3 solves it perfectly. Now it can be trained on unaligned images like FFHQ-U. For StyleGAN3 applications? There are only a few yet.

StyleGAN Inversion

GAN inversion is a technique to invert a given image back to the latent space, where semantic editing is easily done. For example, when you get a latent code z\mathbb{z} of your portrait x\mathbb{x}, you can generate a younger version by adding the “decrease age” vector n\mathbb{n} and feeding it to the generator.

2021 11 08 22 35 27

Image taken from [5].

There are three approaches to GAN inversion:

  • Optimization-based methods use an iterative algorithm, such as gradient descent, to search for the latent code that minimizes the reconstruction loss. They generally yield better reconstruction results but take a much longer time due to the iterative process.
  • Learning-based methods train an encoder that inverts images to the latent space. They are generally faster and less accurate than their optimization-based counterparts.
  • Hybrid methods of these two. Typically, they first invert a given image to the latent code with a learning-based encoder and then optimize it further. This is expected to reduce the time for optimization and the quality of inversion.

Let’s have a look at some examples.

Image2StyleGAN

Image2StyleGAN [6] is a simple but effective implementation of an optimization-based approach. It minimizes the sum of the perceptual loss and L2 loss to get the embedded latent code.

2021 11 09 08 14 35

Image taken from [6].

The authors noticed that the 512-dimensional intermediate latent space W\mathcal{W} is not informative enough to restore the input image. So, they extended the latent space to W+\mathcal{W}+, a concatenation of 18 different wW\mathbb{w} \in \mathcal{W} vectors, one for each input to the StyleGAN’s AdaIN layers. The resulting W+\mathcal{W}+ space is well disentangled and allows multiple face image editing applications, including style transfer and expression transfer. This feature was further explored in Image2StyleGAN++ [7].

pSp

The learning-based approach pSp (pixel2style2pixel) is a general image-to-image translation framework [8]. The encoder is trained to minimize the reconstruction loss (the sum of L2, LPIPS, identity, and latent code regularization loss) and directly translates the input image to the extended latent code (W+\in \mathcal{W}+).

2021 11 09 08 47 12

Image taken from [8].

The pSp encoder can be trained on images not represented in the StyleGAN domain. In that way, pSp can generate images conditioned on inputs like sketches and segmentation masks.

2021 11 09 09 00 20

Image taken from [8].

e4e

e4e (Encoder for Editing) is a learning-based encoder specifically designed for semantic editing after inversion [9].

First, the authors summarized the variations of the intermediate latent space W\mathcal{W} and the extended W+\mathcal{W}+. They use k\cdot^k notation for the layer extension and explicitly denote W\mathcal{W}_* when it is not the equal distribution to W\mathcal{W}. Now, four variations listed in the table below should be considered. The conventional W+\mathcal{W}+ space is now Wk\mathcal{W}^k or Wk\mathcal{W}^k_* depending on its distribution.

Individual style codes are limited to W\mathcal{W} Same style code in all layers
W\mathcal{W}
Wk\mathcal{W}^k
W\mathcal{W}_*
Wk\mathcal{W}^k_*

Based on the above definition, the authors shed light on distortion-editability tradeoff and distortion-perception tradeoff. That is, inverting to Wk\mathcal{W}^k_* space leads to less distortion (less reconstruction loss) than inverting to W\mathcal{W}, but it is also less realistic and harder to edit results. The figure below depicts the distortion-perception tradeoff.

2021 11 09 09 28 33

Image taken from [9].

e4e employs the pSp’s architecture but has control of these tradeoffs by putting Wk\mathcal{W}^k_* closer to Wk\mathcal{W}^k and W\mathcal{W}_*. The editability-aware encoder achieves good reconstruction and semantic editing at the same time.

2021 11 09 10 02 42

Image taken from [9].

ReStyle

To improve the reconstruction accuracy of pSp and e4e, ReStyle is tasked with predicting a residual of the current estimate to the target [10]. It works with a negligible increase in inference time.

2021 11 09 10 05 38

Image taken from [10].

Semantic Editing

Once the latent code is obtained, we can semantically edit the original image by moving it to a certain direction in the latent space. There are mainly two types of input:

  • Two images (source and reference), possibly with additional instructions including masks or attribute indices.
  • An image and an attribute index.

To put it simply, the former is more like a copy-pasting, while the latter attempts to control the output proactively. Plus, the latter assumes that the edits are made multiple times interactively.

In this area, it is usually assumed that StyleGAN inversion algorithms are given.

Image2StyleGAN++

Image2StyleGAN++ extends Image2StyleGAN to take spatial masks as input to allow local editing [7]. For example, it can perform image crossover:

2021 11 09 20 40 35

Image taken from [7].

Editing in Style

Thanks to the well-disentangled latent space, the k-means clustering of the hidden layer activations of the StyleGAN generator provides an interesting insight: the clusters are semantically meaningful even they were decomposed in an unsupervised way [11].

2021 11 10 00 13 32

Image taken from [11].

Then, the contribution matrix (each element is the contribution of channel cc to cluster kk) can be used to determine the style mixing coefficients. This enables to make a local edit (e.g., swapping eyes).

2021 11 10 00 24 00

Image taken from [11].

StyleFlow

To facilitate more fine-grained and editable encoding, StyleFlow replaces the fully connected layers in the mapping network with attribute-aware continuous normalizing flow (CNF) blocks [12].

2021 11 10 09 27 01

Image taken from [12].

The results are impressive. The two source images are real images (i.e., not generated by StyleGAN), and StyleFlow successfully translates these extreme samples.

2021 11 10 10 03 53

Image taken from [12].

StyleSpace Analysis

Wu et al. quantitatively analyzed how disentangled StyleGAN’s latent spaces are. Although the extended intermediate latent space W+\mathcal{W}+ is the standard choice, they discovered that the style space S\mathcal{S} is actually more disentangled than W+\mathcal{W}+ [13]. As depicted in the figure below, the style space S\mathcal{S} is spanned by a concatenation of affined intermediate latent codes.

2021 11 09 22 24 38

Image taken from [13].

The images below demonstrate how well the style space S\mathcal{S} is disentangled. Each pair is the result of decreasing (-) and increasing (+) a single element of the style code. The numbers in the bottom left corner indicate the layer index and the channel index. Note that the base image is a fake one generated by StyleGAN2.

2021 11 09 21 08 53

Image taken from [13].

Retrieve in Style

Retrieve in Style extended Editing in Style to edit more global features such as pose and hairstyle [14].

2021 11 10 10 14 56

Image taken from [14].

As the title suggests, the authors also provide an algorithm for attribute-guided image retrieval.

3D Control

Among several types of semantic edits, reposing is the most difficult task as it often requires drastic and global changes. Thus, some works explicitly estimate 3D information and use it to perform visually natural reposing.

StyleRig

StyleRig can control human face images generated by StyleGAN like a face rig, by translating the semantic editing on 3D meshes by RigNet [16] to the latent code of StyleGAN [15]. Here, StyleGAN is supposed to be pre-trained and RigNet can be trained in a self-supervised manner.

2021 11 10 21 25 49

Image taken from [16].

Pose with Style

Pose with Style exploits StyleGAN2 to perform human reposing [17]. The target pose information is used as input, and the original image and coordinate information are injected as style. This algorithm is also applicable to the virtual try-on task.

2021 11 10 10 20 19

Image taken from [17].

StyleNeRF

StyleNeRF integrates StyleGAN2 and NeRF (neural radiance field) [19] to generate 3D-consistent high-resolution images [18]. The inputs to the synthesis network are the intermediate latent code w\mathbb{w} and the low-resolution feature map generated by NeRF from the camera pose p\mathbb{p}.

2021 11 10 22 03 21

Image taken from [18]. Left: original NeRF. Right: StyleNeRF.

The figure below shows how StyleNeRF enjoys the benefits from both NeRF’s 3D consistency and StyleGAN2’s perception quality.

2021 11 10 22 17 12

Image taken from [18]. The leftmost is the low-resolution feature map generated by NeRF.

Text-guided Manipulation

Last but not least, text-guided manipulation is an interesting application of StyleGAN.

StyleCLIP

Based upon CLIP (contrastive language-image pretraining) [21], the state-of-the-art text-to-image generator, StyleCLIP finds the optimal direction to move forward in the latent space so that the input image will be edited according to the instruction [20].

A wide variety of text instructions can be accepted. Since CLIP has certain knowledge about proper nouns, it is possible to edit a face to look like “Emma Stone”.

2021 11 10 23 50 29

Image taken from [20].

Concluding Remarks

StyleGAN had such a great impact that its applications are growing fast. I hope this article will give you some sort of perspective on them.

Thanks for reading. As always, feedback is welcomed.

References

[1] Tero Karras, Samuli Laine, Timo Aila. ”A Style-Based Generator Architecture for Generative Adversarial Networks“. CVPR. 2019.
[2] Tero Karras, Samuli Laine, Miika Aittala, Janne Hellsten, Jaakko Lehtinen, Timo Aila. ”Analyzing and Improving the Image Quality of StyleGAN “. CVPR. 2020.
[3] Xun Huang, Serge Belongie. ”Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization“. ICCV. 2017.
[4] Tero Karras, Miika Aittala, Samuli Laine, Erik Härkönen, Janne Hellsten, Jaakko Lehtinen, Timo Aila. ”Alias-Free Generative Adversarial Networks“. NeurIPS. 2021.
[5] Weihao Xia, Yulun Zhang, Yujiu Yang, Jing-Hao Xue, Bolei Zhou, Ming-Hsuan Yang. ”GAN Inversion: A Survey”. 2021.
[6] Rameen Abdal, Yipeng Qin, Peter Wonka. ”Image2StyleGAN: How to Embed Images Into the StyleGAN Latent Space?“. ICCV. 2019.
[7] Rameen Abdal, Yipeng Qin, Peter Wonka. ”Image2StyleGAN++: How to Edit the Embedded Images?“. CVPR. 2020.
[8] Elad Richardson, Yuval Alaluf, Or Patashnik, Yotam Nitzan, Yaniv Azar, Stav Shapiro, Daniel Cohen-Or. ”Encoding in Style: A StyleGAN Encoder for Image-to-Image Translation“. CVPR. 2021.
[9] Omer Tov, Yuval Alaluf, Yotam Nitzan, Or Patashnik, Daniel Cohen-Or. ”Designing an Encoder for StyleGAN Image Manipulation“. SIGGRAPH. 2021.
[10] Yuval Alaluf, Or Patashnik, Daniel Cohen-Or. ”ReStyle: A Residual-Based StyleGAN Encoder via Iterative Refinement“. ICCV. 2021.
[11] Edo Collins, Raja Bala, Bob Price, Sabine Süsstrunk. ”Editing in Style: Uncovering the Local Semantics of GANs“. CVPR. 2020.
[12] Rameen Abdal, Peihao Zhu, Niloy Mitra, Peter Wonka. ”StyleFlow: Attribute-conditioned Exploration of StyleGAN-Generated Images using Conditional Continuous Normalizing Flows“. SIGGRAPH. 2021.
[13] Zongze Wu, Dani Lischinski, Eli Shechtman. ”StyleSpace Analysis: Disentangled Controls for StyleGAN Image Generation“. CVPR. 2021.
[14] Min Jin Chong, Wen-Sheng Chu, Abhishek Kumar, David Forsyth. ”Retrieve in Style: Unsupervised Facial Feature Transfer and Retrieval”. 2021.
[15] Ayush Tewari, Mohamed Elgharib, Gaurav Bharaj, Florian Bernard, Hans-Peter Seidel, Patrick Pérez, Michael Zollhöfer, Christian Theobalt. ”StyleRig: Rigging StyleGAN for 3D Control over Portrait Images“. CVPR. 2020.
[16] Zhan Xu , Yang Zhou , Evangelos Kalogerakis , Chris Landreth , Karan Singh. ”RigNet: Neural Rigging for Articulated Characters“. SIGGRAPH. 2020.
[17] Badour AlBahar, Jingwan Lu, Jimei Yang, Zhixin Shu, Eli Shechtman, Jia-Bin Huang. ”Pose with Style: Detail-Preserving Pose-Guided Image Synthesis with Conditional StyleGAN“. SIGGRAPH Asia. 2021.
[18] Jiatao Gu, Lingjie Liu, Peng Wang, Christian Theobalt. ”StyleNeRF: A Style-based 3D-Aware Generator for High-resolution Image Synthesis”. 2021.
[19] Ben Mildenhall, Pratul P. Srinivasan, Matthew Tancik, Jonathan T. Barron, Ravi Ramamoorthi, Ren Ng. ”NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis“. ECCV. 2020.
[20] Or Patashnik, Zongze Wu, Eli Shechtman, Daniel Cohen-Or, Dani Lischinski. ”StyleCLIP: Text-Driven Manipulation of StyleGAN Imagery“. ICCV. 2021.
[21] Alec Radford, Jong Wook Kim, Chris Hallacy, Aditya Ramesh, Gabriel Goh, Sandhini Agarwal, Girish Sastry, Amanda Askell, Pamela Mishkin, Jack Clark, Gretchen Krueger, Ilya Sutskever. ”Learning Transferable Visual Models From Natural Language Supervision“. ICML. 2021.


  • このエントリーをはてなブックマークに追加
[object Object]

Written by Shion Honda. If you like this, please share!

Shion Honda

Hippocampus's Garden © 2024, Shion Honda. Built with Gatsby