生成对抗网络-条件生成|深度学习(李宏毅)(二十五)

视频地址:
①B站:https://www.bilibili.com/video/BV1JE411g7XF?p=73https://www.bilibili.com/video/BV1JE411g7XF?p=74
②油管:https://www.youtube.com/watch?v=LpyL4nZSuqUhttps://www.youtube.com/watch?v=-3LgL3NXLtI

之前的博客地址:
生成对抗网络-基本概念|深度学习(李宏毅)(二十二)
生成对抗网络-理论部分|深度学习(李宏毅)(二十三)
生成对抗网络-改进方法|深度学习(李宏毅)(二十四)

一、Text-to-Image

  1. 概述

对于原来的GAN来说,只能够输入一个随机的向量,然后得到一个产生的对象,但是我们无法控制产生什么样的对象,而我们期待用GAN来做到这件事,也就是条件生成(Conditional Generation)。以Text-to-Image任务来说,我们希望给Generator输入一段文字,然后让它来产生对应的图片:

Conditional Generation

对于这样的一个任务,我们可以考虑用监督学习的方法来做,也就是给神经网络输入一段文字,来让其输出一张图片,并且要让图片与目标图片越接近越好,以此来达到条件生成的目的:

监督学习

然而这样的做法很可能不太好,试想在训练资料中有多张火车的图片,包括正面的火车和侧面的火车,而按照监督学习的方式,模型会希望在输出“train”时输出的图片要和所有的火车的图片尽可能地相似,那么生成的结果可能就是多张图片的平均,效果看起来可能是很模糊的:

problem

而使用Conditional GAN来完成这个任务就可以避免这个问题,对于Conditional GAN的Generator来说,要同时输入两个对象,一个是从先验分布中采样出来的样本z,一个是当前的条件c,也就是一段文字,然后Generator输出生成的对象x=G(c,z)

Generator

而对于Discriminator来说,如果它只根据输入的图片是否是真实的来打分显然是不够的,这样会造成Generator在学习的过程中逐渐忽略条件c,举例来说,如果Discriminator只关注图片是否真实那么Generator输入“train”时产生一张清晰的猫的图片也能从Discriminator处得到高分。因此,我们同样需要对Discriminator做一些改造,Conditional GAN中Discriminator要有两个输入,一个是条件c,另一个是图片x,仍然输出一个标量代表得分,不过现在这个得分评估两件事,一个是图片x是否真实,另一个是xc是否匹配:

Discriminator

这样的Discriminator的训练样本也要有所变化,对于正样本,必须是正确的文字描述和对应的真实的图片,而负样本分两种,分别是文字描述配上不真实的图片和真实的图片随机匹配错误的文字描述:

训练样本

下图展示了用Conditional GAN做的二次元人物头像生成的效果:

效果
  1. 算法

Initialize:
  初始化D的参数\theta _{d}G的参数\theta _{g}


Step1 学习D
  ①从数据库中随机抽样m个正样本\left \{(c^{1},x^{1}),(c^{2},x^{2}),\cdots ,(c^{m},x^{m})\right \}
  ②从一个分布(比如高斯分布或者均匀分布)中采样m个噪声样本(noise sample)\left \{z^{1},z^{2},\cdots ,z^{m}\right \}
  ③获得G生成的数据\left \{\tilde{x}^{1},\tilde{x}^{2},\cdots ,\tilde{x}^{m}\right \},其中\tilde{x}^{i}=G(c^{i},z^{i})
  ④从数据库中随机抽样m个对象\left \{\hat{x}^{1},\hat{x}^{2},\cdots ,\hat{x}^{m}\right \}
  ⑤目标函数记作\tilde{V},通过最大化(梯度上升)\tilde{V}来更新\theta _{d},也就是\theta _{d}\leftarrow \theta _{d}+\eta \nabla \tilde{V}(\theta _{d}),目标函数\tilde{V}为:
\tilde{V}=\frac{1}{m}\sum_{i=1}^{m}log\; D(c^{i},x^{i})+\frac{1}{m}\sum_{i=1}^{m}log\; (1-D(c^{i},\tilde{x}^{i}))+\frac{1}{m}\sum_{i=1}^{m}log\; (1-D(c^{i},\hat{x}^{i}))


Step2 学习G
  ①同样从一个分布(比如高斯分布或者均匀分布)中采样m个噪声样本(noise sample)\left \{z^{1},z^{2},\cdots ,z^{m}\right \}
  ②从数据库中采样m个条件\left \{c^{1},c^{2},\cdots ,c^{m}\right \}
  ③目标函数同样记作\tilde{V},通过最小化(梯度下降)\tilde{V}来更新\theta _{g},也就是\theta _{g}\leftarrow \theta _{g}-\eta \nabla \tilde{V}(\theta _{g}),目标函数\tilde{V}为:
\tilde{V}=\frac{1}{m}\sum_{i=1}^{m}log\left (1-D(G(c^{i},z^{i}))\right )

上面算法中的Step1和Step2也是交替迭代进行的。对于Generator也可以使用梯度上上来极大化\tilde{V}=\frac{1}{m}\sum_{i=1}^{m}logD(G(c^{i},z^{i}))这个目标函数。

  1. Discriminator的两种架构

在Conditional GAN的中Discriminator的架构一般比较常见的是下图中的模式,这里的条件c和对象x都经过各自的一个network得到各自的embedding,然后将这两个embedding输入到另一个network中得到一个得分来衡量x是否真实以及xc是否匹配:

常见的架构

另一种不太常见的架构如下图所示,这里的x首先通过一个network,这个network输出两个东西,一个是衡量x是否真实的得分,另一个则是x的embedding,然后这个x的embedding以及条件c要同时被输入到另一个network中,这个network输出一个得分来衡量xc是否匹配:

不常见的架构

这里的第二种架构的设计相对于第一种来说应该是更加合理的,对于第一种来说,如果Discriminator得出一个低分,那它无法知道为什么给一个低分,不知道是因为不够真实还是不够匹配,而第二种架构显然就可以明显地看出得分的分布。不过在文献中第一种架构是比较常见的,也没有实验来对比这两种架构设计,在实践中可以作为两种不同的方案以供选择。

  1. Stack GAN

在产生比较大的图片时,可能会需要Discriminator拥有更大的capacity,并且生成的效果也可能不会太好,Stack GAN是一种Conditional GAN,它的思路是先产生较小的图,然后再用小图去产生大图。它的架构图如下:

Stack GAN

感兴趣的话可以参照它的原论文:https://arxiv.org/abs/1612.03242

二、Image-to-Image

还可以使用Conditional GAN来做Image-to-Image的生成,比如将轮廓图转换成真实图,黑白图转换成彩图,白天转换成黑夜等:

Image-to-Image

以将轮廓图转换成真实图片为例,如果采用监督学习的方法进行Image-to-Image的生成,做法也就是给神经网络输入一张图片,然后输出生成图片来让它和目标图片尽可能地接近:

监督学习

这样的做法很可能产生和前面Text-to-Image中监督学习方法同样的问题,也就是生成的图片是训练资料的平均,看起来很模糊:

监督学习

因此我们尝试用Conditional GAN的方法来完成这个任务,对于Generator来说会输入轮廓图和一个随机噪声z,输出生成的真实的图片,然后Discriminator会关注Generator的输入和输出,也就是会给Discriminator输入轮廓图和生成图并对其真实度和匹配程度打分,该架构如下:

Conditional GAN

这样的做法其实也可能会遇到一些问题,比如在下图生成的图片中,虽然很清晰但是左上角多了个多余的烟囱,这可能是因为Discriminator会认为左上角加个烟囱好像也没什么不对:

Conditional GAN

我们可以尝试为生成的图片加一些约束来使得生成的图片更加地可控,具体的做法就是将监督学习的做法与Conditional GAN结合起来,也就是在训练Generator时不仅要让Discriminator的得分越高越好,也要让生成的图片与目标图片越接近越好:

结合的方法

这样的话就会产生比较好的效果:

结合的方法

在Image-to-Image的论文里它的Discriminator有经过特别的设计,也就是每次并不是将整张图片输入到Discriminator中,而是只输入图片的一部分:

Patch GAN

Image-to-Image的原论文链接:https://arxiv.org/abs/1611.07004

三、Conditional GAN的其他应用

  1. Speech Enhancement

Conditional GAN也可以用来做语音的增强,也就是将有噪声的语音讯号转换成清晰的信号:

Speech Enhancement

对于传统的监督学习方法,训练数据是清晰的和加噪的声音讯号的pair,做法就是向神经网络中输入加噪的声音讯号,然后让输出的声音讯号与清晰的讯号越接近越好,另外要说的是,声音讯号跟图片差不多也是个矩阵,因此可以套用一些图像处理的方法:

监督学习

而使用Conditional GAN和监督学习相结合的方法就是除了要让Discriminator得分越高越好还让Generator的输出和清晰的讯号尽可能地接近,这是为了避免模型只学习到去噪却会使得讯号的信息改变。而Discriminator会分辨Generator的输入和输出是不是匹配的,也会看Generator的输出是不是clean的:

结合的方法
  1. Video Generation

使用Conditional GAN来做Video Generation的具体做法就是给Generator输入一段影片,然后让它产生接下来影片中要发生的事情,然后把原来的影片和产生的帧接起来输入到Discriminator中让它来区分最后一帧是不是真实的,同时也要让生成的帧与真实的下一帧越接近越好:

Video Generation

四、无监督条件生成

  1. 概述

前面介绍的Conditional GAN的方法全部都是有监督的方法,事实上也可以用GAN来做无监督的条件生成,也就是我们能够实现让一些数据从一个domain转换到另外一个domain,比如我们可以尝试让一张真实的图片转换成梵高画作的风格(风格转换),其实这样的任务也是无法收集到有标签的训练数据的:

风格转换

另外这类无监督条件生成的例子也并非仅限于图像处理,在其他领域也可以应用得到,比如音频性别的转换、文本情感的转换等等:

其他领域应用
  1. 方法的分类

能够做到domain的转换的方法大体上可以分为两类:

  • Approach 1: Direct Transformation

这种方法的Generator的输入和输出没有办法差太多,只能够改变图片的颜色、质地、色调一类浅层的东西,对于将真实的图片转换成梵高画作的风格这样的风格转换的任务是比较适合的:

Direct Transformation
  • Approach 2: Projection to Common Space

第二种方法适用于转换前后差距很大的任务,比如把真人的脸转换成动漫人物的脸,这样的任务不是改改颜色一类浅层的特征就可以办到的。这种方法会首先将输入通过一个Encoder变成一个特征向量,然后用这个特征向量解析得到一个转换后的图像:

Projection to Common Space
  1. 第一种方法

以风格转换的任务为例,现在我们期待要将真实的图片转换成梵高画作的风格,我们有的数据是没有标签的两个domain的大量图片:

训练数据

虽然没有标注数据,但是我们可以通过Discriminator在两个domain之间建立联系,具体的做法是训练一个Discriminator来分辨输入的图片是否是梵高的画作,而Generator的训练目标就是让它输出的图片被Discriminator认为是梵高的画作:

架构

不过这么做可能会遇到一个问题,也就是由于Generator没有受到任何限制,所以可能会直接无视输入而产生一个与输入完全不同的图片来尝试骗过Discriminator

issue
  • 采用浅层网络

为了解决上述问题,第一种方法就是采用较渐层的Generator,只要采用比较浅的Generator就可以直接无视这个问题,因为对于比较简单的Generator来说,它的输出不会对输入改变太多。然而如果Generator是深层的就有可能使输入受到很大的改变从而产生上述问题。

  • 采用两个Encoder

第二种解决方案就是让Generator的输入和输出都通过一个训练好的Encoder变成相应的code,在训练的时候要让这两个code尽可能地接近,这样就可以保证Generator不会让输入和输出相差太多:

Encoder
  • CycleGAN

CycleGAN的做法是让Generator(这个Generator记作G_{X\rightarrow Y})输出的图片再经过另一个Generator(这个Generator记作G_{Y\rightarrow X})还原成真实的图,然后让还原的这张图能够和原来的输入越接近越好,这个就叫做cycle consistency,在训练G_{X\rightarrow Y}时另一个G_{Y\rightarrow X}和Discriminator(这里的Discriminator记作D_Y)是固定住的:

CycleGAN

那么G_{Y\rightarrow X}是怎么训练的呢?事实上这个Generator的功能是输入一张梵高画作风格的图,输出真实风格的图,它的训练和G_{X\rightarrow Y}的训练是一样的,对于G_{Y\rightarrow X}的输出也要通过G_{X\rightarrow Y}还原成梵高画作风格的图,同时要与原来的输入越接近越好,并且G_{Y\rightarrow X}的Discriminator(这个Discriminator不妨记作D_X)的作用是判定输入的图片是不是属于真实的风格。CycleGAN的整体架构如下图:

CycleGAN

总结一下,也就是说CycleGAN需要训练两个Generator和两个Discriminator,他们之间相互支持训练能够保证Generator的输入和输出不会千差万别并且风格也能得到转换。

CycleGAN论文链接:https://arxiv.org/abs/1703.10593

另外研究表明CycleGAN有一个有趣的现象,比如在下图中,两个Generator会将原图转换成中间风格的图然后再转回来,然而中间的图的房顶上没有黑点,但是经过G_{Y\rightarrow X}后又产生了黑点,这说明CycleGAN把原图中的信息隐藏起来了:

CycleGAN

参考论文链接:https://arxiv.org/abs/1712.02950

  • StarGAN

刚才CycleGAN讨论的是两个domain互相转换的情况,而如果我们期待能够在多个domain之间都能互相转换,那我们就需要在两两个domain之间训练各自的Generator:

cross-domain models

而StarGAN这种方法可以只训练一个Generator就完成多个domain之间的互相转换。在StarGAN中,它的Discriminator会输入一张真实的图片或者生成的图片,然后会判定这张图片是real的还是fake的,并且判断这张图片属于哪一个domain:

Discriminator

它的Generator会输入来自某个domain的原始的图片和想要转换到的domain,然后输出转换后的图片,得到的这张图片既要输入给Discriminator来尝试骗过他,也要与原始图片的domain标签结合在一起重新输入给同一个Generator,然后Generator输出的重构的图片要和原始图片尽可能地接近,这个其实类似上面的cycle consistency:

Generator

StarGAN的domain的标签其实不是一个数字而是一个0-1向量,下图展示了对人脸特征的转换效果:

特征转换

下面是对人脸表情的转换效果:

表情转换

StarGAN论文链接:https://arxiv.org/abs/1711.09020

  1. 第二种方法

对于二次元人脸转换的问题,我们手里只有无监督的两个domain的数据:

数据

第二种方法是期望能够通过Encoder来提取两个domain的输入图片的特征并且投影成一个隐层向量,然后Decoder能够通过这个隐层的向量来生成一张某个domain的图片。通过这种方式,我们期望能够做到将真实的人脸通过真实图片的Encoder编码成向量,然后使用这个向量和二次元人脸的Decoder来生成转换后的图片:

第二种方法

首先我们想到的做法是按照自编码器的方法分别训练两对Encoder和Decoder,同时也要为每个Decoder输出的图片训练一个Discriminator来鉴别是否属于这个domain:

架构

事实上这种做法显然是有问题的,因为两个自编码器是单独训练的,他们的隐层向量的构建没有任何联系,很有可能隐层向量的每一维度对于两个domain来说表示完全不相同的特征,这可能造成真实人脸通过Encoder得到的隐层向量通过二次元人脸的Decoder后生成的图片跟原始图片完全不相干:

问题

我们有许多不同的方法来解决上述问题:

  • 共享参数

我们可以尝试将两个Encoder的最后几层以及Decoder的前几层的参数共享,通过这种方法我们就可以期待隐层向量的每一维度对于两个domain来说都表示相同的特征:

共享参数

对于Encoder来说,这种共享参数最极端的状况就是Encoder的参数是完全共享的,也就是只使用一个Encoder,不过这个Encoder也要输入一个flag来表明输入的图片来自哪一个domain。

  • Domain Discriminator

另一种方法就是加一个Domain Discriminator,这个Discriminator输入Encoder的隐层向量并且判断这个隐层向量来自哪一个domain,如果现在Encoder产生的隐层向量可以骗过Discriminator的话,那我们就可以期待隐层向量的每一维度对于两个domain来说都表示相同的特征:

Domain Discriminator
  • Cycle Consistency

这里也可以应用Cycle Consistency的方法,具体也就是将原始图片输入Encoder后得到隐层向量,然后用二次元人脸的Decoder把这个向量解析成图片,然后这个图片再通过二次元人脸的Encoder得到隐层向量,这个隐层向量再通过真实人脸的Decoder解析成图片,然后使得这个图片与原始图片越接近越好:

Cycle Consistency

这个技术用在ComboGAN里,论文链接:https://arxiv.org/abs/1712.06909

  • Semantic Consistency

Semantic Consistency的方法与Cycle Consistency不同的是,Cycle Consistency计算的是图片与图片像素级别的相似度,而Semantic Consistency计算的是两个Encoder获得的隐层向量之间的相似度:

Semantic Consistency

这种方法的好处就是可以考虑Semantic上的相似度而不是Cycle Consistency只考虑表象上的相似度。

这个技术用在XGAN里,论文链接:https://arxiv.org/abs/1711.05139

  1. Voice Conversion

对于声音转换这个任务,在过去的方法中就是收集两个人的声音,让他们说一些一样的话然后按照监督学习的方法去实现:

过去的方法

而如果使用GAN的方法,就可以让两个人说不一样的话甚至不同的语言,然后用上述无监督的方法来完成这个任务:

GAN的方法

这方面的内容就不再赘述。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,080评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,422评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,630评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,554评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,662评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,856评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,014评论 3 408
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,752评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,212评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,541评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,687评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,347评论 4 331
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,973评论 3 315
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,777评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,006评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,406评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,576评论 2 349

推荐阅读更多精彩内容