首页 > 代码库 > Generative Adversarial Nets——解析

Generative Adversarial Nets——解析

摘要

  本文提出了一个通过对抗过程来预测产生式模型的新框架。在新框架中我们同时训练两个模型:一个用来获得数据分布的生成模型G,和一个用来估计样本来自训练数据而不是G的概率的判别模型D,G的训练过程是最大化D产生错误的概率,让其无法判断一个图像是由生成模型产生的,还是来自训练样本.这个框架相当于一个极小化极大的双方博弈。在任意函数G 和D 的空间中存在唯一的解,其中G恢复训练数据分布,并且D处处都等于12。 在G和D 由多层感知器定义的情况下,整个系统可以用反向传播进行训练。在训练或生成样本期间不需要任何马尔科夫链或展开的近似推理网络。

介绍

  深度学习的任务是寻找丰富的层次模型,使其能够在人工智能领域里用来表达各种数据的概率分布,例如自然图像,包含语音的音频波形和自然语言语料库中的符号等。到目前为止,在深度学习领域最成功的的模型之一就是判别式模型,通常它们将高维的丰富的输入映射到类标签上,它主要是基于反向传播和丢弃算法来实现的。由于在最大似然估计和相关策略中出现的许多难以解决的概率计算的困难,以及很难利用在生成上下文时使用分段线性单元的好处,深度生成模型发展得并不是很好。因此,本文提出了一个新的生成模型克服了这些困难。

  本文的对抗生成网络框架中,生成式模型就像是罪犯,专门造假币,判别式模型就像是警察,专门辨别假币,那生成式模型就要努力提高自己的造假技术使造的假币不被发现,而判别式模型就要努力提高自己辨别假币的能力。最终的结果就是两人达到一个平衡的状态,这就是对抗。因为每个人的cost function都依赖于另一个人的参数,但是每个人都不能控制别人的参数,所以这个场景可以认为是一个带有博弈色彩的优化问题。

  这个框架可以针对多种模型和优化算法提供特定的训练算法。在这篇文章中,我们探讨了生成模型通过将随机噪声传输到多层感知机来生成样本的特例,同时判别模型也是通过多层感知机实现的。我们称这个特例为对抗网络。在这种情况下,我们可以仅使用非常成熟的反向传播和丢弃算法来训练两个模型。对于生成模型在生成样本时只使用前向传播算法即可。并且不需要近似推理和马尔可夫链作为前题。

对抗生成网络

先来看一个不太正式但是很直观的图:

技术分享

上图中:

下面的水平线为均匀采样z(噪声数据)的区域,上面的水平线为数据x的部分区域。

蓝色的虚线 表示判别器D(x; Θd)产生的分布

黑色的点线 表示数据x的分布pdata

绿色的实线 表示生成器G(z; Θg),生成的分布pg

向上的箭头表示z通过G转化之后(x‘ = G(z))与分布pg的对应情况

a) 接近收敛的对抗模型对。pg接近于pdata,并且D具有较准确的判别能力

b) 在判别器算法的内部循环中,D收敛到pdata(x) / (pdata(x) +pg)

c) 在D的指导下,G生成的分布更加接近数据分布。

d) 经过几次训练之后D,G都具有不错的能力,并且都已经无法提升自己,此时pg = pdata,D(x) = 1/2

我们说过对于生成模型G,通过最大化D产生错误的概率来对它训练(让它骗过D)。对于判别模型D,通过最大化样本来自训练数据而不是G的概率来对它训练(尽可能的识别G)。

目标函数:

  技术分享

对于生成器G,我们训练后面一项,对于判别器D我们训练整个。

注:(1)优化D的过程在训练的内部所花费的代价是非常高的,并且对于是有限的数据集将会导致overfitting。因此,我们用 k steps来优化 D,one step 来优化 G 。这使得G改变的足够缓慢,那么D将保持在其最佳解附近。

  (2)算法早期G的效果差,D会以高置信值来拒绝生成样本。log(1?D(G(z)))将很快饱和(梯度),所以我们通过最大化log(D(G(z)))而不是最小化log(1?D(G(z))) 来训练G,这个函数在早期能提供非常强的梯度信息。

 算法流程:

技术分享

 

整个算法流程就结束了,接下来我们证明一下算法全局最优为pg = pdata,并且对算法的收敛情况进行讨论。

1. 全局最优为pg = pdata

命题1:固定G,判别器D的最优解为:

技术分享

证明:

  给定G,我们需要最大化目标函数V (G;D)

技术分享

这里x = g(z)是给定的G产生的z和x的映射关系,并且这里的x所在的分布为pg

 对于任意不同时等于0的两个实数(a,b),函数技术分享当y∈(0,1),最大值在a / (a + b)处。证毕

由命题1,公式(1)公式1中的极小化极大问题可以变形为: 

技术分享

目标:最小化C(G),变量为pg

 定理1:当且仅当pg = pdata时,C(G)达到最小,且值为-log4.

证明: 

当pg = pdata时,由公式(2)可以得到DG*(x) = 1/2.

把它代入方程4得到:C(G) = log 1/2 + log 1/2 = -log 4,由此我们可以得到当pg = pdata时,C(G)的值为-log4。但是否为最小呢?我们看到:

      技术分享

用C(G) = V(DG*,G)减去上式我们得到:

  技术分享

其中,KL表示KL散度,也称为相对熵,是一个距离函数。有KL(p||q) = ∑plog(p / q)。

由公式(5)我们能够得到JS散度:

技术分享

其中,JSD表示JS散度

技术分享

由于两个分布之间的Jensen–Shannon散度总是非负的,并且当两个分布相等时,值为0。因此C*=?log(4)C(G)的全局极小值,并且唯一解为pg=pdata,即生成模型能够完美的复制数据的生成过程。

至此,我们证明了算法的全局最优解为pg=pdata

 2.证明Algorithm 1的收敛性

 命题2:当G和D有足够的能力时。在算法1中,给定G,D总能达到它的最优值,并且通过更新pg来提高这个判别准则。 

   技术分享最终pg收敛到pdata.

证明:

如上述准则,考虑V(G,D)=U(pg,D)为关于pg的函数。注意到U(pg,D) 为pg的凸函数。该凸函数上确界内的次导数一定包括该函数最大值处的导数。所以在给定最优的D时,通过梯度下降算法更新pg从而优化G时,pg一定会收敛到最优值。而定理1证明了目标函数只有唯一的全局最优解,所以pg会收敛到pdata。 证毕

实际上优化G时是更新Θg而不是pg本身。尽管缺乏理论证明,但是实践中感知机的优良性能表明这是个好的模型。

 

 

 

Generative Adversarial Nets——解析