对抗生成网络

概述

对抗生成网络包含两个模型,一个是生成模型(generative model),一个是判别模型(discriminative model)。生成模型的任务是生成看起来自然真实的、和原始数据相似的实例。判别模型的任务是判断给定的实例看起来是自然真实的还是人为伪造的(真实实例来源于数据集,伪造实例来源于生成模型)。

论文中的类比解释:生成模型像“一个造假团伙,试图生产和使用假币”,而判别模型像“检测假币的警察”。生成器(generator)试图欺骗判别器(discriminator),判别器则努力不被生成器欺骗。模型经过交替优化训练,两种模型都能得到提升,但最终我们要得到的是效果提升到很高很好的生成模型(造假团伙),这个生成模型(造假团伙)所生成的产品能达到真假难分的地步。

GAN网络整体示意如下:

对抗生成网络-01.png

有两个网络,G(Generator)和D(Discriminator)。Generator是一个生成图片的网络,它接收一个随机的噪声z,通过这个噪声生成图片,记做G(z)。Discriminator是一个判别网络,判别一张图片是不是“真实的”。它的输入是xx代表一张图片,输出D(x)代表x为真实图片的概率,如果为1,就代表100%是真实的图片,而输出为0,就代表不可能是真实的图片。

上面的 判别式模型拟合的是条件类别概率为P(x|y),生成模型拟合的是联合概率分布P(x,y)

模型优化训练

再正式了解GAN模型优化训练过程前,先了解一下纳什均衡,纳什均衡是博弈中的局面:对于每个参与者来说,只要其他人不改变策略,他就无法改善自己的状况。转换到GAN中,生成模型G恢复了训练数据的分布,判别器就无法再判别出正确的结果,即判别器开始乱猜。这是双方利益达到了最大化,不再改变策略,也就不再更新自己的权重了。

GAN的目标函数为:
\min _{G} \max _{D} V(D, G)=\mathbb{E}_{\boldsymbol{x} \sim p_{\text {data }}(\boldsymbol{x})}[\log D(\boldsymbol{x})]+\mathbb{E}_{\boldsymbol{z} \sim p_{\boldsymbol{z}}(\boldsymbol{z})}[\log (1-D(G(\boldsymbol{z})))]
上面的公式中, z为随机噪声;p_z随机噪声z服从的概率分布;G(z;\theta_g)生成器为输入噪声z输出的假图像;p_g为生成器生成的假图像服从的概率分布;x\sim p_\text{data}为真实数据服从的概率分布;D(x;\theta_d)判别器输入的是图像,输出该图像来自x的概率;\theta_g为生成器G的参数,\theta_d为判别器D的参数。公式的第一项\log D(\boldsymbol{x})的期望,希望输入真实数据时,输出越大越好。公式中的第二项\log (1-D(G(\boldsymbol{z})))的期望,希望输入假数据时,输出越小越好。

因此对抗体现在判别器D希望能够正确的判别出图像是否来自于x,即生成图像的判别结果趋近于0,真实图像的判别结果趋近于1;生成器G希望输出的图像分布和真实分布接近,即生成图像的判别结果很小。

在训练生成器时损失\log (1-D(G(\boldsymbol{z})))的图像为下图,我们希望判别器对生成器生成的图像判别为正确的,因此希望生成器的损失越小越好。

对抗生成网络-02.jpg

那么他在训练的是时候是怎么训练的呢?从下面的伪代码中,我们可以发现在训练的过程中,首先固定生成器G参数并训练多轮判别器,然后固定判别器D参数并训练一轮生成器参数,由此交替迭代,使得对方的错误最大化,最终生成器G能估测出样本数据的分布。

对抗生成网络-03.png

论文最后给出了对抗训练的过程图:

对抗生成网络-04.png

训练相关理论

命题一:当生成器G参数固定时,最优的判别器D
D_G^*({\boldsymbol x})=\frac{p_{\text{data}}({\boldsymbol x})}{p_{\text{data}}({\boldsymbol x})+p_{\text{g}} ({\boldsymbol x})}

它接收一个任意一个图片,输出最有判别器认为他是真实数据的概率。上式中,p_{\text{data}}({\boldsymbol x})+p_{\text{g}}({\boldsymbol x})的结果不一定为1。

证明:根据期望的定义\mathbb{E}_{x\sim p}[f(x)]=\int_x p(x)f(x)dx,可以由上文中GAN的目标函数转化为:
V(G, D) =\int_{\boldsymbol{x}} p_{\text {data }}(\boldsymbol{x}) \log (D(\boldsymbol{x})) dx+\int_{\boldsymbol{z}} p_{\boldsymbol{z}}(\boldsymbol{z}) \log (1-D(g(\boldsymbol{z}))) dz
上式中积分变量不一致,利用测度论Radon-Nikodym Theorem(无意识统计学家定律)来切换积分变量,转化为:
V(G, D) =\int_{\boldsymbol{x}} p_{\text {data }}(\boldsymbol{x}) \log (D(\boldsymbol{x}))+p_{g}(\boldsymbol{x}) \log (1-D(\boldsymbol{x})) dx

无意识统计学家定律是用于切换积分变量:

z\rightarrow x单射时,对噪声z采样相当于对图像采样,即\mathbb{E}_{z\sim p_z(z)}\log(1-D(G(z)))=\mathbb{E}_{x\sim p_G(x)}\log(1-D(x))

上式的形式为可以看作为交叉熵y\rightarrow a\log(y)+b\log(1-y)y=\frac{a}{a+b}时在y\in[a,b]上取最大值。

对于上面的a,b为常数,y为自变量。交叉熵为f(y)=a\log{y}+b\log{(1-y)},对其求导为f'(y)=\frac{a}{y}-\frac{b}{1-y},导数为0时y=\frac{a}{a+b}。若a+b\neq0a,b\in(0,1)y=\frac{a}{a+b}f''(y)的结果为f''(\frac{a}{a+b})=-\frac{a}{(\frac{a}{a+b})^2}-\frac{b}{(1-\frac{a}{a+b})^2}<0。根据上面的一阶导数为0、二阶导数为负,y取得最大值。

定理1:当且仅当p_g=p_{\text{data}}时,生成器价值函数C(G)取得全局最小值-\log4

证明:在上文判别器D已取到最优化时,训练器G使判别器D的价值函数最小化:
\begin{aligned} C(G)&=\underset{D}{\max}V(G,D)\\ &=\mathbb{E}_{\boldsymbol{x} \sim p_{\text {data }}}[\log D^*_G(\boldsymbol{x})]+\mathbb{E}_{\boldsymbol{z} \sim p_{\boldsymbol{z}}}[\log (1-D^*_G(G(\boldsymbol{z})))]\\ &=\mathbb{E}_{\boldsymbol{x} \sim p_{\text {data }}}[\log \frac{p_{\text{data}}(\boldsymbol{x})}{p_{\text{data}}(\boldsymbol{x})+p_{\text{g}}(\boldsymbol{x})}]+\mathbb{E}_{\boldsymbol{x} \sim p_{\boldsymbol{g}}}[\log \frac{p_{\text{g}}(\boldsymbol{x})}{p_{\text{data}}(\boldsymbol{x})+p_{\text{g}}(\boldsymbol{x})}] \end{aligned}
对上面式子做如下变换:
\begin{aligned} C(G)&=-\log4+\log4+\mathbb{E}_{\boldsymbol{x} \sim p_{\text {data }}}[\log \frac{p_{\text{data}}(\boldsymbol{x})}{p_{\text{data}}(\boldsymbol{x})+p_{\text{g}}(\boldsymbol{x})}]+\mathbb{E}_{\boldsymbol{x} \sim p_{\boldsymbol{g}}}[\log \frac{p_{\text{g}}(\boldsymbol{x})}{p_{\text{data}}(\boldsymbol{x})+p_{\text{g}}(\boldsymbol{x})}]\\ &=-\log4+\mathbb{E}_{\boldsymbol{x} \sim p_{\text {data }}}[\log \frac{p_{\text{data}}(\boldsymbol{x})}{p_{\text{data}}(\boldsymbol{x})+p_{\text{g}}(\boldsymbol{x})}+\log2]+\mathbb{E}_{\boldsymbol{x} \sim p_{\boldsymbol{g}}}[\log \frac{p_{\text{g}}(\boldsymbol{x})}{p_{\text{data}}(\boldsymbol{x})+p_{\text{g}}(\boldsymbol{x})}+\log2]\\ &=-\log4+\mathbb{E}_{\boldsymbol{x} \sim p_{\text {data }}}[\log \frac{p_{\text{data}}(\boldsymbol{x})}{\frac{p_{\text{data}}(\boldsymbol{x})+p_{\text{g}}(\boldsymbol{x})}{2}}]+\mathbb{E}_{\boldsymbol{x} \sim p_{\boldsymbol{g}}}[\log \frac{p_{\text{g}}(\boldsymbol{x})}{\frac{p_{\text{data}}(\boldsymbol{x})+p_{\text{g}}(\boldsymbol{x})}{2}}]\\ &=-\log4+KL(p_{\text{data}}||\frac{p_{\text{data}}+p_{\text{g}}}{2})+KL(p_{\text{g}}||\frac{p_{\text{data}}+p_{\text{g}}}{2})\\ &=-\log4+JSD(p_{\text{data}}||p_{\text{g}}) \end{aligned}

KL散度的定义为:
\begin{gathered} {KL}(\mathrm{P} \| \mathrm{Q})=\sum_{i \in X} P(i) *(\log (\frac{P(i)}{Q(i)}) \\ {KL}(\mathrm{P} \| \mathrm{Q})=\int_{x} P(x) *(\log \frac{P(i)}{Q(i)}) d x \end{gathered}
KL散度存在非负性和不对称性。

JS散度的定义:设M=\frac{1}{2}(P+Q),则:
JSD(P||Q)=\frac{1}{2}KL(P||M)+\frac{1}{2}KL(Q||M)

因此当p_{\text{data}}=p_{\text{g}}时,得到全局最小值-\log4

代码链接:github

参考文献:

  1. Generative Adversarial Nets
  2. 生成对抗网络(GAN)
  3. 生成对抗网络GAN开山之作论文精读
  4. 通俗理解生成对抗网络GAN
  5. KL散度相关知识点总结
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容