深度学习之风格转换——Style Transfer

sooner高

6823人浏览 · 2018-03-19 20:33:16

sooner高 · 2018-03-19 20:33:16 发布

最近在学习CS20课程——一门讲述tensorflow应用的实践性课程，正好Assignment2讲到了Style Transfer这个东西，这里把我的理解总结一下（代码基于Tensorflow）。

代码：
chiphuyen/stanford-tensorflow-tutorials/assignments/02_style_transfer/
论文：Bringing Impressionism to Life with Neural Style Transfer in Come Swim https://arxiv.org/pdf/1701.04928v1.pdf

简介

这是一个使用两个图片（以A和B表示）作为输入（图A作为内容输入(content input)，图B作为风格输入（style input）。最终的目的是得到一张具有图B的风格和图A的内容的图像。

style picture
content picture
what we got

用一个简单的方程可以表示为：style + content = picture of highly abstracted

步骤简述

使用一个训练好的CNN结构（一般来说做Image classification的就可以，比如VGG, GoogLeNet等等）
我们要有一个概念：CNN中低层特征图保留的内容比较多，而高层更多的是纹理，正是利用这一点，我们将组合内容信息和纹理信息并将两者相结合得到一个全新的图片。
对content和style分别定义损失函数，并将其组合起来作为整个结构的Loss function。
值得注意的是，我们输入时候使用的结构使用的是一样的结构（无论是style pic、content pic还是trianable input），为了避免重复的子计算图装配(To save us from having to assemble the same subgraph multiple times, we will use one variable for all three of them.)，需要使用tf.variable_scope(‘input’)。
跟之前的CNN中通过反向传播调整网络结构参数如w和b不同的是，这里的目标是对一个输入的content pic加入白噪声后的图片进行不断调整，给其加入内容和纹理信息。

损失函数说明

L t o t a l (p \to, a \to, x \to) = α L c o n t e n t (p \to, x \to) + β L s t y l e (a \to, x \to)

其中α和β是权重，论文指出α/β=0.001或者0.0001，这个比例是根据内容和风格的比例来调整的，当你想要风格更浓烈，就提高α/β的值，反之则降低α/β。

1. content loss

content loss比较容易理解，论文里定义如下：

L c o n t e n t (p \to, x \to, l) = 1 2 \sum i, j (F l i, j - P l i, j) 2

假设某一层 l <script type="math/tex" id="MathJax-Element-3">l</script>得到的响应是

F^{l} \in R^{N^{l} * M^{l}}

Nl N l <script type="math/tex" id="MathJax-Element-5">N^l</script>为l层filter的个数，

Ml M l <script type="math/tex" id="MathJax-Element-6">M^l</script>为filter的大小(

height∗width h e i g h t ∗ w i d t h <script type="math/tex" id="MathJax-Element-7">height * width</script>)。

Fli,j F i , j l <script type="math/tex" id="MathJax-Element-8">F^{l}_{i,j}</script>表示的是第

l l <script type="math/tex" id="MathJax-Element-9">l</script>层第

i

j j <script type="math/tex" id="MathJax-Element-11">j</script>的输出。

这里的 $F_{i, j}^{l}$ <script type="math/tex" id="MathJax-Element-12">F_{i,j}^l</script>是层 l <script type="math/tex" id="MathJax-Element-13">l</script>生成图像的特征表示(content representation of the generated image)， $P_{i, j}^{l}$ <script type="math/tex" id="MathJax-Element-14">P_{i,j}^l</script>是层 l <script type="math/tex" id="MathJax-Element-15">l</script>内容图像的特征表示(content representation of the content image)。实际上就是关于 $F$ <script type="math/tex" id="MathJax-Element-16">F</script>和 P <script type="math/tex" id="MathJax-Element-17">P</script>的均方误差。论文中建议使用的layer是conv4_2。

下面翻译自CS20 Assignment2 的note

然而在实践中发现，这个content loss收敛的很慢。所以这里可以将前面的系数1/2换成1/(4s)。s为 $P$ <script type="math/tex" id="MathJax-Element-18">P</script>的所有维度的乘积，比如 P <script type="math/tex" id="MathJax-Element-19">P</script>为[5, 5, 3]。那么s = 5*5*3=75。

2. style loss

style loss有些复杂，在论文中，它有三步计算得来。首先让我们来看style loss的定义：

E_{l} = \frac{1}{4 N_{l}^{2} M_{l}^{2}} \sum_{i, j} (G_{i, j}^{l} - A_{i, j}^{l})^{2}

L s t y l e (a \to, x \to) = \sum l = 0 L w l E l

这里 Nl <script type="math/tex" id="MathJax-Element-22">N_l</script>是层 l <script type="math/tex" id="MathJax-Element-23">l</script>的卷积核（filter）个数， $M_{l}$ <script type="math/tex" id="MathJax-Element-24">M_l</script>为特征图的高和宽的乘积。这里， A <script type="math/tex" id="MathJax-Element-25">A</script>是原图的Gram矩阵， $G$ <script type="math/tex" id="MathJax-Element-26">G</script>为生成图像的Gram矩阵。

Gram矩阵的计算方式是： Gli,j=∑kFli,k∗Flj,k <script type="math/tex" id="MathJax-Element-27">G^{l}_{i,j}=\sum_k F^{l}_{i,k}*F^{l}_{j,k}</script>。即为同一层 l <script type="math/tex" id="MathJax-Element-28">l</script>各个不同的特征图的偏心协方差矩阵。在feature map中，每一个数字都来自于一个特定滤波器在特定位置的卷积，因此每个数字就代表一个特征的强度，而Gram计算的实际上是特征之间的相关性——“哪两个特征是同时出现的，哪两个是此消彼长”。

以层 $l$ <script type="math/tex" id="MathJax-Element-29">l</script>为例，当很多feature map的同一个位置上的数值越小，那么我们得到的Gram中此位置的值就越小；当很多feature map的同一个位置上的数值越大，那么我们得到的Gram中此位置的值就越大。

所以Gram矩阵这里的作用是：在能够在保证内容的情况下，进行风格的渲染

这里的 l <script type="math/tex" id="MathJax-Element-30">l</script>表示了我们想要吸收进生成图像style的特征图，论文中建议取：
这里写图片描述

代码中对应的权值为：
这里写图片描述

试验

定义完毕之后，就回到最开始的Total loss:

L_{t o t a l} (\vec{p}, \vec{a}, \vec{x}) = α L_{c o n t e n t} (\vec{p}, \vec{x}) + β L_{s t y l e} (\vec{a}, \vec{x})

这里再次强调，我们训练的参数不是VGG或者GoogLenet里面的 W <script type="math/tex" id="MathJax-Element-47">W</script>和 $b$ <script type="math/tex" id="MathJax-Element-48">b</script>，而是输入的content pic + noise的内容（为其加上content和style信息）