本章涉及到的知识点清单:
1、数学期望的定义
2、KL散度的定义
3、零和博弈
4、GAN的工作原理
5、GAN的目标函数
6、求解D的最优解
7、反求解G使得G和D的概率分布差异最小
8、案例之GAN实现拟合二次函数
推导GAN公式之前,需要预备一些数学期望和KL散度的知识点
一、数学期望的定义
期望:在概率论中,将实验中每次可能产生的结果的概率乘以其结果的总和,反映随机变量平均取值的大小。根据其随机变量的取值范围不同,分为离散型和连续型对于连续型随机变量x,其概率密度函数为f(x),则X的数学期望E(x)可以表示成微积分的形式
二、KL散度的定义
KL散度:在信息论中,用生成的概率分布Q来拟合逼近真实的概率分布P时,所产生的信息损耗,即描述两个概率分布的差异,其本身是非对称的
设x是连续型随机变量,其真实概率分布为P(x),拟合分布概率为Q(x),则P对Q的KL散度为
三、零和博弈
GAN被称为对抗式神经网络,启发自博弈论中的二人零和博弈
零和博弈:指参与博弈的双方,在严格的竞争下,一方的收益必然意味着另一方的损失,博弈过程中,双方的各自收益和损失的相加总和永远为零,双方完全不存在合作的可能。就好比下棋一样,你和对手的每一步棋都是向着自己最有利的方向走,最终只有一方赢一方输,而下棋的总成绩永远为零
显然,GAN也是由博弈双方组成,分别为生成网络G(Generator)和判别网络D(Discriminator)
四、GAN的工作原理
上图中,x是真实数据,Pdata(x)是x的概率分布,z是噪点数据,P(z)是z的概率分布,其工作过程为:
(1):从噪声z进行随机抽样,传入G网络,生成新数据G(z)和其概率分布Pg(G(z))
(2):将真实数据和G生成的新数据一起传入D网络进行真假判别,通过sigmoid函数来输出判定类别
(3):迭代优化D和G损失函数,根据D来调整G
(4):直到D和G达到收敛,即D无法判断G产生数据的真假性,即Pg(G(z))已经非常逼近Pdata(x)
至此,我们可以抽象看出GAN的目的,将随机噪声z通过G网络得到一个和真实数据分布Pdata(x)差不多的生成分布Pg(G(z)),这个过程就是G和D相互博弈的过程
五、GAN的目标函数
定义GAN的目标函数为V(G,D),在博弈过程中,G希望减少V的值让自己生成的分布无法识别,而D希望增大V的值让自己可以高效的判别出数据的真假类别,则V(G,D)的表达式为
其中E表示真实数据x和噪点数据z的数学期望
G网络是一个生成器,可以是全连接神经网络、卷积神经网络等等,通过噪点分布P(z),一般是高斯分布,得到一个生成数据的分布Pg(x),我们希望Pg(x)非常靠近Pdata(x),来拟合逼近真实分布
D网络是一个判别函数,需要解决传统的二分类问题,其职责就是有效的区分真实分布和生成分布,即衡量Pg(x)和Pdata(x)之间的差距,并通过反复的迭代训练
六、求解D的最优解
从目标函数出发,由于V是连续的,我们将V写成微积分的形式来表示期望
设G(z)生成的数据是x,分别求出噪点z和噪点的微分dz表达式
带入z和dz,可以得到
我们定义Pg(x)表示z的生成分布,则
带入目标函数可得
现在要求V(D,G)关于D的最大值,则固定G来求D的偏导数
七、反求解G使得G和D的概率分布差异最小
从D(x)的最优解D*(X)的表达式可以看到,我们期望当G产生出来的拟合分布和真实分布一致时,即
在这个条件下,D*(x)=1/2,即此时D网络已经无法直接分辨出G产生出来的数据的真假性了
那么当D满足最优解后,此时的G的解是什么呢?我们只需要带入D*(x)反过来求解G即可
我们对上述积分表达式进行等效处理,在log里面的分式上,分子分母同时除以2(分式不变原理),然后保持分母不变,将分子的1/2利用对数的乘法原理提到外面,则上式可以等效变形为
我们引入连续函数的KL散度将上式积分式整理成散度表达式
根据KL散度的定义,当拟合分布Pg(x)完全等于真实分布Pdata(x)时,KL=0,所以G网络的最小值是-log4
由此证明了当D网络逼近其最优解的同时,G网络也无限逼近其最小值
八、案例之GAN实现拟合二次函数
有G网络和D网络的意义,我们编写如下代码来拟合二次函数,其中G网络只是一个全连接网络,利用梯度下降来反向传播更新其权重
迭代5000次后的博弈结果为
从结果上可以看到,G网络生成的分布(绿色)已经非常逼近真实分布(蓝色),且D网络的判别能力逼近50%,G网络的最优值逼近-log4=1.38629达到了很好的收敛效果
案例代码见 https://github.com/privateEye-zzy/gan/tree/master
作者:PrivateEye_zzy
链接:https://www.jianshu.com/p/cbcb94d28585
來源:简书