深度学习的常见模型(GAN)

GAN的来源

14年Goodfellow提出Generative adversarial nets即生成式对抗网络,它要解决的问题是如何从训练样本中学习出新样本,训练样本是图片就生成新图片,训练样本是文章就输出新文章等等。

GANs简单的想法就是用两个模型, 一个生成模型,一个判别模型。判别模型用于判断一个给定的图片是不是真实的图片(从数据集里获取的图片),生成模型的任务是去创造一个看起来像真的图片一样的图片,有点拗口,就是说模型自己去产生一个图片,可以和你想要的图片很像。而在开始的时候这两个模型都是没有经过训练的,这两个模型一起对抗训练,生成模型产生一张图片去欺骗判别模型,然后判别模型去判断这张图片是真是假,最终在这两个模型训练的过程中,两个模型的能力越来越强,最终达到稳态。

GAN的基本组成

GAN 模型中的两位博弈方分别由生成式模型(generative model)和判别式模型(discriminative model)充当。

生成模型: G 捕捉样本数据的分布,用服从某一分布(均匀分布,高斯分布等)的噪声 z 生成一个类似真实训练数据的样本,追求效果是越像真实样本越好;

判别模型: D 是一个二分类器,估计一个样本来自于训练数据(而非生成数据)的概率,如果样本来自于真实的训练数据,D 输出大概率,否则,D 输出小概率。

可以做如下类比:生成网络 G 好比假币制造团伙,专门制造假币,判别网络 D 好比警察,专门检测使用的货币是真币还是假币,G 的目标是想方设法生成和真币一样的货币,使得 D 判别不出来,D 的目标是想方设法检测出来 G 生成的假币。

示例图片
上图是GAN网络的思想导图,我们用1代表真实数据,0来代表生成的假数据。对于判别器D来说,对于真实数据,它要尽可能判别正确输出值1;而对于生成器G,根据随机噪音向量z生成假数据也输入判别器D,对于这些假数据,判别器要尽可能输出值0。

GAN的训练过程可以看成一个博弈的过程,也可以看成2个人在玩一个极大极小值游戏,可以用如下公式表示:

$$
\min \limits_{G}\max\limits_{D}GAN(D,G)=E_{x\sim P_{data}(x)}[logD(x)]+E_{z\sim p_z(z)}[log(1-D(G(z)))]
$$
其本质上是两个优化问题,把拆解就如同下面两个公式:

优化D:
$$
\max\limits_{D}GAN(D,G)=E_{x\sim P_{data}(x)}[logD(x)]+E_{z\sim p_z(z)}[log(1-D(G(z)))]
$$
优化G:
$$
\min\limits_{G}GAN(D,G)=E_{z\sim p_z(z)}[log(1-D(G(z)))]
$$

当优化D时,生成器确定,我们要让判别器尽可能输出高的值,所以要最大化公式(2)的值;当优化G的时候,判别器确定,我们要使判别器判断错误,尽可能使D(G(z))的值更大,所以要最小化公式(3)的值。

GAN的训练过程

示例图片
上图是GAN的训练过程,解析见图中右边文字。

GAN的算法流程和动态求解过程如下图所示:

示例图片
示例图片
一开始我们确定G,最大化D,让点沿着D变大的方向移动(红色箭头),然后我们确定D,最小化G,让点沿着G变小的方向移动(蓝色箭头)。循环上述若干步后,达到期望的鞍点(理想最优解)。

GAN的网络结构

判别器(卷积)

卷积层大家应该都很熟悉了,为了方便说明,定义如下:

二维的离散卷积(N=2)
方形的特征输入(i1=i2=i)
方形的卷积核尺寸(k1=k2=k )
每个维度相同的步长(s1=s2=s)
每个维度相同的padding (p1=p2=p)
下图(左)表示参数为 (i=5,k=3,s=2,p=1)的卷积计算过程,从计算结果可以看出输出特征的尺寸为 (o1=o2=o=3);下图(右)表示参为 (i=6,k=3,s=2,p=1)的卷积计算过程,从计算结果可以看出输出特征的尺寸为 (o1=o2=o=3)。

示例图片
从上述2个例子我们可以总结出卷积层输入特征和输出特征尺寸和卷积核参数的关系为:
$$
o=\lfloor\frac{i+2p-k}{s}\rfloor+1
$$

生成器(反卷积)

在介绍反卷积之前,我们先来看一下卷积运算和矩阵运算之间的关系。例有如下运算(i=4,k=3,s=1,p=0),输出为o=2。

示例图片
通过上述的分析,我们已经知道卷积层的前向操作可以表示为和矩阵C相乘,那么我们很容易得到卷积层的反向传播就是和C的转置相乘。

反卷积和卷积的关系如下:
示例图片
右上图表示的是参数为( i′=2,k′=3,s′=1,p′=2)的反卷积操作,其对应的卷积操作参数为 (i=4,k=3,s=1,p=0)。我们可以发现对应的卷积和非卷积操作其 (k=k′,s=s′),但是反卷积却多了p′=2。通过对比我们可以发现卷积层中左上角的输入只对左上角的输出有贡献,所以反卷积层会出现 p′=k−p−1=2。通过示意图,我们可以发现,反卷积层的输入输出在 s=s′=1的情况下关系为:
$$
o′=i′-k′+2p′+1=i′+(k-1)-2p
$$

GAN的优点
●GAN是一种生成式模型,相比较其他生成模型(玻尔兹曼机和GSNs)只用到了反向传播
●相比其他所有模型, GAN可以产生更加清晰,真实的样本
●GAN采用的是一种无监督的学习方式训练,可以被广泛用在无监督学习和半监督学习领域

GAN的缺点
●训练GAN需要达到纳什均衡,有时候可以用梯度下降法做到,有时候做不到.我们还没有找到很好的达到纳什均衡的方法,所以训练GAN相比VAE或者PixelRNN是不稳定的
●GAN不适合处理离散形式的数据,比如文本
●GAN存在训练不稳定、梯度消失、模式崩溃的问题

实例DCGAN网络
网络结构 (判别器)
示例图片

网络结构 (生成器)
示例图片

二次元动漫人脸(共50个epoch)
数据集:51223张动漫人脸
示例图片
图左为原始数据集,图右为训练过程

训练过程生成效果图如下:

示例图片
示例图片

真实人脸(共100个epoch)
数据集:CelebA 是香港中文大学的开放数据集,包含10,177个名人身份的202,599张人脸图片。(选取了25600张),数据集如下:
示例图片

训练过程生成效果图如下:

示例图片
示例图片

------ 本文结束------
bluesliuf wechat
坚持技术分享,欢迎大家扫码交流!