GAN的数学推导和案例应用

本章涉及到的知识点清单:
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网络只是一个全连接网络,利用梯度下降来反向传播更新其权重

G网络和D网络

迭代5000次后的博弈结果为

在这里插入图片描述

在这里插入图片描述

从结果上可以看到,G网络生成的分布(绿色)已经非常逼近真实分布(蓝色),且D网络的判别能力逼近50%,G网络的最优值逼近-log4=1.38629达到了很好的收敛效果

案例代码见 https://github.com/privateEye-zzy/gan/tree/master

作者:PrivateEye_zzy
链接:https://www.jianshu.com/p/cbcb94d28585
來源:简书

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值