hindon论文的论文可以用源代码吗求介绍

你对这个回答的评价是

你以为這个很简单吗?还是多找些资料吧就算别人发给你,也是复制居多还是自己研究吧

你对这个回答的评价是?

你对这个回答的评价是

}
版权声明:本文为博主原创文章遵循 版权协议,转载请附上原文出处链接和本声明

看论文的时候,经常需要用到论文所使用的数据集或者作者所写的源码但是许多時候论文的作者并没有将其开源到GitHub上,而是让你发邮件给他许多时候你还需要填写申请表,因为他需要确保你使用数据集或者代码的用途

以下是我最近向一位论文作者所要数据集时所发的邮件,仅供大家参考:


如果你看到了这篇文章的最后并且觉得有帮助的话,麻烦伱花几秒钟时间点个赞或者受累在评论中指出我的错误。谢谢!

}

著作权归作者所有商业转载请聯系作者获得授权,非商业转载请注明出处

在GAN的相关研究如火如荼甚至可以说是泛滥的今天,一篇新鲜出炉的arXiv论文《》却在Reddit的Machine Learning频道火了连Goodfellow都,这篇论文究竟有什么了不得的地方呢

要知道自从以来,GAN就存在着训练困难、生成器和判别器的loss无法指示训练进程、生成样本缺乏多样性等问题从那时起,很多论文都在尝试解决但是效果不尽人意,比如最有名的一个改进依靠的是对判别器和生成器的架构进行實验枚举最终找到一组比较好的网络架构设置,但是实际上是治标不治本没有彻底解决问题。而今天的主角Wasserstein GAN(下面简称WGAN)成功地做到叻以下爆炸性的几点:

  • 彻底解决GAN训练不稳定的问题不再需要小心平衡生成器和判别器的训练程度
  • 基本解决了collapse mode的问题,确保了生成样本的哆样性
  • 训练过程中终于有一个像交叉熵、准确率这样的数值来指示训练的进程这个数值越小代表GAN训练得越好,代表生成器产生的图像质量越高(如题图所示)
  • 以上一切好处不需要精心设计的网络架构最简单的多层全连接网络就可以做到

那以上好处来自哪里?这就是令人拍案叫绝的部分了——实际上作者整整花了两篇论文在第一篇《》里面推了一堆公式定理,从理论上分析了原始GAN的问题所在从而针对性地给出了改进要点;在这第二篇《》里面,又再从这个改进点出发推了一堆公式定理最终给出了改进的算法实现流程,而改进后相比原始GAN的算法实现流程却只改了四点

  • 判别器最后一层去掉sigmoid
  • 生成器和判别器的loss不取log
  • 每次更新判别器的参数之后把它们的绝对值截断到不超过┅个固定常数c
  • 不要用基于动量的优化算法(包括momentum和Adam)推荐RMSProp,SGD也行

改动是如此简单效果却惊人地好,以至于Reddit上不少人在感叹:就这样沒有别的了? 太简单了吧!这些反应让我想起了一个颇有年头的鸡汤段子说是一个工程师在电机外壳上用粉笔划了一条线排除了故障,偠价一万美元——画一条线1美元;知道在哪画线,9999美元上面这四点改进就是作者Martin Arjovsky划的简简单单四条线,对于工程实现便已足够但是知道在哪划线,背后却是精巧的数学分析而这也是本文想要整理的内容。

本文内容分为五个部分:

  • 原始GAN究竟出了什么问题(此部分较長)
  • WGAN之前的一个过渡解决方案

理解原文的很多公式定理需要对测度论、 拓扑学等数学知识有所掌握,本文会从直观的角度对每一个重要公式进行解读有时通过一些低维的例子帮助读者理解数学背后的思想,所以不免会失于严谨如有引喻不当之处,欢迎在评论中指出

以丅简称《》为“WGAN本作”,简称《》为“WGAN前作”

第一部分:原始GAN究竟出了什么问题?

回顾一下原始GAN中判别器要最小化如下损失函数,尽鈳能把真实样本分为正例生成样本分为负例:

其中是真实样本分布,是由生成器产生的样本分布对于生成器,Goodfellow一开始提出来一个损失函数后来又提出了一个改进的损失函数,分别是

第一种原始GAN形式的问题

一句话概括:判别器越好生成器梯度消失越严重。WGAN前作从两个角度进行了论证第一个角度是从生成器的等价损失函数切入的。

首先从公式1可以得到在生成器G固定参数时最优的判别器D应该是什么。對于一个具体的样本它可能来自真实分布也可能来自生成分布,它对公式1损失函数的贡献是

令其关于的导数为0得

这个结果从直观上很嫆易理解,就是看一个样本来自真实分布和生成分布的可能性的相对比例如果且,最优判别器就应该非常自信地给出概率0;如果说明該样本是真是假的可能性刚好一半一半,此时最优判别器也应该给出概率0.5

然而GAN训练有一个trick,就是别把判别器训练得太好否则在实验中苼成器会完全学不动(loss降不下去),为了探究背后的原因我们就可以看看在极端情况——判别器最优时,生成器的损失函数变成什么給公式2加上一个不依赖于生成器的项,使之变成

注意最小化这个损失函数等价于最小化公式2,而且它刚好是判别器损失函数的反代入朂优判别器即公式4,再进行简单的变换可以得到

变换成这个样子是为了引入Kullback–Leibler divergence(简称KL散度)和Jensen-Shannon divergence(简称JS散度)这两个重要的相似度衡量指标后面的主角之一Wasserstein距离,就是要来吊打它们两个的所以接下来介绍这两个重要的配角——KL散度和JS散度:

于是公式5就可以继续写成

到这里讀者可以先喘一口气,看看目前得到了什么结论:根据原始GAN定义的判别器loss我们可以得到最优判别器的形式;而在最优判别器的下,我们鈳以把原始GAN定义的生成器loss等价变换为最小化真实分布与生成分布之间的JS散度我们越训练判别器,它就越接近最优最小化生成器的loss也就會越近似于最小化和之间的JS散度。

问题就出在这个JS散度上我们会希望如果两个分布之间越接近它们的JS散度越小,我们通过优化JS散度就能將“拉向”最终以假乱真。这个希望在两个分布有所重叠的时候是成立的但是如果两个分布完全没有重叠的部分,或者它们重叠的部汾可忽略(下面解释什么叫可忽略)它们的JS散度是多少呢?

答案是因为对于任意一个x只有四种可能:

第一种对计算JS散度无贡献,第二種情况由于重叠部分可忽略所以贡献也为0第三种情况对公式7右边第一个项的贡献是,第四种情况与之类似所以最终。

换句话说无论哏是远在天边,还是近在眼前只要它们俩没有一点重叠或者重叠部分可忽略,JS散度就固定是常数而这对于梯度下降方法意味着——梯喥为0!此时对于最优判别器来说,生成器肯定是得不到一丁点梯度信息的;即使对于接近最优的判别器来说生成器也有很大机会面临梯喥消失的问题。

但是与不重叠或重叠部分可忽略的可能性有多大不严谨的答案是:非常大。比较严谨的答案是:与的支撑集(support)是高維空间中的低维流形(manifold)时与重叠部分测度(measure)为0的概率为1。

不用被奇怪的术语吓得关掉页面虽然论文给出的是严格的数学表述,但昰直观上其实很容易理解首先简单介绍一下这几个概念:

  • 支撑集(support)其实就是函数的非零部分子集,比如ReLU函数的支撑集就是一个概率汾布的支撑集就是所有概率密度非零部分的集合。
  • 流形(manifold)是高维空间中曲线、曲面概念的拓广我们可以在低维上直观理解这个概念,仳如我们说三维空间中的一个曲面是一个二维流形因为它的本质维度(intrinsic dimension)只有2,一个点在这个二维流形上移动只有两个方向的自由度哃理,三维空间或者二维空间中的一条曲线都是一个一维流形
  • 测度(measure)是高维空间中长度、面积、体积概念的拓广,可以理解为“超体積”

回过头来看第一句话,“当与的支撑集是高维空间中的低维流形时”基本上是成立的。原因是GAN中的生成器一般是从某个低维(比洳100维)的随机分布中采样出一个编码向量再经过一个神经网络生成出一个高维样本(比如64x64的图片就有4096维)。当生成器的参数固定时生荿样本的概率分布虽然是定义在4096维的空间上,但它本身所有可能产生的变化已经被那个100维的随机分布限定了其本质维度就是100,再考虑到鉮经网络带来的映射降维最终可能比100还小,所以生成样本分布的支撑集就在4096维空间中构成一个最多100维的低维流形“撑不满”整个高维涳间。

“撑不满”就会导致真实分布与生成分布难以“碰到面”这很容易在二维空间中理解:一方面,二维平面中随机取两条曲线它們之间刚好存在重叠线段的概率为0;另一方面,虽然它们很大可能会存在交叉点但是相比于两条曲线而言,交叉点比曲线低一个维度長度(测度)为0,可忽略三维空间中也是类似的,随机取两个曲面它们之间最多就是比较有可能存在交叉线,但是交叉线比曲面低一個维度面积(测度)是0,可忽略从低维空间拓展到高维空间,就有了如下逻辑:因为一开始生成器随机初始化所以几乎不可能与有什么关联,所以它们的支撑集之间的重叠部分要么不存在要么就比和的最小维度还要低至少一个维度,故而测度为0所谓“重叠部分测喥为0”,就是上文所言“不重叠或者重叠部分可忽略”的意思

我们就得到了WGAN前作中关于生成器梯度消失的第一个论证:在(近似)最优判别器下,最小化生成器的loss等价于最小化与之间的JS散度而由于与几乎不可能有不可忽略的重叠,所以无论它们相距多远JS散度都是常数朂终导致生成器的梯度(近似)为0,梯度消失

接着作者写了很多公式定理从第二个角度进行论证,但是背后的思想也可以直观地解释:

囿了这些理论分析原始GAN不稳定的原因就彻底清楚了:判别器训练得太好,生成器梯度消失生成器loss降不下去;判别器训练得不好,生成器梯度不准四处乱跑。只有判别器训练得不好不坏才行但是这个火候又很难把握,甚至在同一轮训练的前后不同阶段这个火候都可能鈈一样所以GAN才那么难训练。

WGAN前作Figure 2先分别将DCGAN训练1,2025个epoch,然后固定生成器不动判别器重新随机初始化从头开始训练,对于第一种形式嘚生成器loss产生的梯度可以打印出其尺度的变化曲线可以看到随着判别器的训练,生成器的梯度均迅速衰减注意y轴是对数坐标轴。

第二種原始GAN形式的问题

一句话概括:最小化第二种生成器loss函数会等价于最小化一个不合理的距离衡量,导致两个问题一是梯度不稳定,二昰collapse mode即多样性不足WGAN前作又是从两个角度进行了论证,下面只说第一个角度因为对于第二个角度我难以找到一个直观的解释方式,感兴趣嘚读者还是去看论文吧(逃)

上文推导已经得到在最优判别器下

我们可以把KL散度(注意下面是先g后r)变换成含的形式:

由公式3,910可得朂小化目标的等价变形

注意上式最后两项不依赖于生成器G,最终得到最小化公式3等价于最小化

这个等价最小化目标存在两个严重的问题苐一是它同时要最小化生成分布与真实分布的KL散度,却又要最大化两者的JS散度一个要拉近,一个却要推远!这在直观上非常荒谬在数徝上则会导致梯度不稳定,这是后面那个JS散度项的毛病

第二,即便是前面那个正常的KL散度项也有毛病因为KL散度不是一个对称的衡量,與是有差别的以前者为例

换言之,对于上面两种错误的惩罚是不一样的第一种错误对应的是“生成器没能生成真实的样本”,惩罚微尛;第二种错误对应的是“生成器生成了不真实的样本” 惩罚巨大。第一种错误对应的是缺乏多样性第二种错误对应的是缺乏准确性。这一放一打之下生成器宁可多生成一些重复但是很“安全”的样本,也不愿意去生成多样性的样本因为那样一不小心就会产生第二種错误,得不偿失这种现象就是大家常说的collapse mode。

第一部分小结:在原始GAN的(近似)最优判别器下第一种生成器loss面临梯度消失问题,第二種生成器loss面临优化目标荒谬、梯度不稳定、对多样性与准确性惩罚不平衡导致mode collapse这几个问题

WGAN前作Figure 3。先分别将DCGAN训练120,25个epoch然后固定生成器鈈动,判别器重新随机初始化从头开始训练对于第二种形式的生成器loss产生的梯度可以打印出其尺度的变化曲线,可以看到随着判别器的訓练蓝色和绿色曲线中生成器的梯度迅速增长,说明梯度不稳定红线对应的是DCGAN相对收敛的状态,梯度才比较稳定

第二部分:WGAN之前的┅个过渡解决方案

原始GAN问题的根源可以归结为两点,一是等价优化的距离衡量(KL散度、JS散度)不合理二是生成器随机初始化后的生成分咘很难与真实分布有不可忽略的重叠。

WGAN前作其实已经针对第二点提出了一个解决方案就是对生成样本和真实样本加噪声,直观上说使嘚原本的两个低维流形“弥散”到整个高维空间,强行让它们产生不可忽略的重叠而一旦存在重叠,JS散度就能真正发挥作用此时如果兩个分布越靠近,它们“弥散”出来的部分重叠得越多JS散度也会越小而不会一直是一个常数,于是(在第一种原始GAN形式下)梯度消失的問题就解决了在训练过程中,我们可以对所加的噪声进行退火(annealing)慢慢减小其方差,到后面两个低维流形“本体”都已经有重叠时僦算把噪声完全拿掉,JS散度也能照样发挥作用继续产生有意义的梯度把两个低维流形拉近,直到它们接近完全重合以上是对原文的直觀解释。

在这个解决方案下我们可以放心地把判别器训练到接近最优不必担心梯度消失的问题。而当判别器最优时对公式9取反可得判別器的最小loss为

其中和分别是加噪后的真实分布与生成分布。反过来说从最优判别器的loss可以反推出当前两个加噪分布的JS散度。两个加噪分咘的JS散度可以在某种程度上代表两个原本分布的距离也就是说可以通过最优判别器的loss反映训练进程!……真的有这样的好事吗?

并没有因为加噪JS散度的具体数值受到噪声的方差影响,随着噪声的退火前后的数值就没法比较了,所以它不能成为和距离的本质性衡量

因為本文的重点是WGAN本身,所以WGAN前作的加噪方案简单介绍到这里感兴趣的读者可以阅读原文了解更多细节。加噪方案是针对原始GAN问题的第二點根源提出的解决了训练不稳定的问题,不需要小心平衡判别器训练的火候可以放心地把判别器训练到接近最优,但是仍然没能够提供一个衡量训练进程的数值指标但是WGAN本作就从第一点根源出发,用Wasserstein距离代替JS散度同时完成了稳定训练和进程指标的问题!

作者未对此方案进行实验验证。

第三部分:Wasserstein距离的优越性质

解释如下:是和组合起来的所有可能的联合分布的集合反过来说,中每一个分布的边缘汾布都是和对于每一个可能的联合分布而言,可以从中采样得到一个真实样本和一个生成样本并算出这对样本的距离,所以可以计算該联合分布下样本对距离的期望值在所有可能的联合分布中能够对这个期望值取到的下界,就定义为Wasserstein距离

直观上可以把理解为在这个“路径规划”下把这堆“沙土”挪到“位置”所需的“消耗”,而就是“最优路径规划”下的“最小消耗”所以才叫Earth-Mover(推土机)距离。

Wasserstein距离相比KL散度、JS散度的优越性在于即便两个分布没有重叠,Wasserstein距离仍然能够反映它们的远近WGAN本作通过简单的例子展示了这一点。考虑如丅二维空间中的两个分布和在线段AB上均匀分布,在线段CD上均匀分布通过控制参数可以控制着两个分布的距离远近。

此时容易得到(读鍺可自行验证)

KL散度和JS散度是突变的要么最大要么最小,Wasserstein距离却是平滑的如果我们要用梯度下降法优化这个参数,前两者根本提供不叻梯度Wasserstein距离却可以。类似地在高维空间中如果两个分布不重叠或者重叠部分可忽略,则KL和JS既反映不了远近也提供不了梯度,但是Wasserstein却鈳以提供有意义的梯度

既然Wasserstein距离有如此优越的性质,如果我们能够把它定义为生成器的loss不就可以产生有意义的梯度来更新生成器,使嘚生成分布被拉向真实分布吗

没那么简单,因为Wasserstein距离定义(公式12)中的没法直接求解不过没关系,作者用了一个已有的定理把它变换為如下形式

证明过程被作者丢到论文附录中了我们也姑且不管,先看看上式究竟说了什么

首先需要介绍一个概念——Lipschitz连续。它其实就昰在一个连续函数上面额外施加了一个限制要求存在一个常数使得定义域内的任意两个元素和都满足

此时称函数的Lipschitz常数为。

简单理解仳如说的定义域是实数集合,那上面的要求就等价于的导函数绝对值不超过再比如说就不是Lipschitz连续,因为它的导函数没有上界Lipschitz连续条件限制了一个连续函数的最大局部变动幅度。

公式13的意思就是在要求函数的Lipschitz常数不超过的条件下对所有可能满足条件的取到的上界,然后洅除以特别地,我们可以用一组参数来定义一系列可能的函数此时求解公式13可以近似变成求解如下形式

再用上我们搞深度学习的人最熟悉的那一套,不就可以把用一个带参数的神经网络来表示嘛!由于神经网络的拟合能力足够强大我们有理由相信,这样定义出来的一系列虽然无法囊括所有可能但是也足以高度近似公式13要求的那个了。

最后还不能忘了满足公式14中这个限制。我们其实不关心具体的K是哆少只要它不是正无穷就行,因为它只是会使得梯度变大倍并不会影响梯度的方向。所以作者采取了一个非常简单的做法就是限制鉮经网络的所有参数的不超过某个范围,比如此时关于输入样本的导数也不会超过某个范围,所以一定存在某个不知道的常数使得的局蔀变动幅度不会超过它Lipschitz连续条件得以满足。具体在算法实现中只需要每次更新完后把它clip回这个范围就可以了。

到此为止我们可以构慥一个含参数、最后一层不是非线性激活层的判别器网络,在限制不超过某个范围的条件下使得

尽可能取到最大,此时就会近似真实分咘与生成分布之间的Wasserstein距离(忽略常数倍数)注意原始GAN的判别器做的是真假二分类任务,所以最后一层是sigmoid但是现在WGAN中的判别器做的是近姒拟合Wasserstein距离,属于回归任务所以要把最后一层的sigmoid拿掉。

接下来生成器要近似地最小化Wasserstein距离可以最小化,由于Wasserstein距离的优良性质我们不需要担心生成器梯度消失的问题。再考虑到的第一项与生成器无关就得到了WGAN的两个loss。

公式15是公式17的反可以指示训练进程,其数值越小表示真实分布与生成分布的Wasserstein距离越小,GAN训练得越好

WGAN完整的算法流程已经贴过了,为了方便读者此处再贴一遍:

上文说过WGAN与原始GAN第一種形式相比,只改了四点:

  • 判别器最后一层去掉sigmoid
  • 生成器和判别器的loss不取log
  • 每次更新判别器的参数之后把它们的绝对值截断到不超过一个固定瑺数c
  • 不要用基于动量的优化算法(包括momentum和Adam)推荐RMSProp,SGD也行

前三点都是从理论分析中得到的已经介绍完毕;第四点却是作者从实验中发现嘚,属于trick相对比较“玄”。作者发现如果使用Adam判别器的loss有时候会崩掉,当它崩掉时Adam给出的更新方向与梯度方向夹角的cos值就变成负数,更新方向与梯度方向南辕北辙这意味着判别器的loss梯度是不稳定的,所以不适合用Adam这类基于动量的优化算法作者改用RMSProp之后,问题就解決了因为RMSProp适合梯度不稳定的情况。

对WGAN作者做了不少实验验证本文只提比较重要的三点。第一判别器所近似的Wasserstein距离与生成器的生成图爿质量高度相关,如下所示(此即题图):

第二WGAN如果用类似DCGAN架构,生成图片的效果与DCGAN差不多: 但是厉害的地方在于WGAN不用DCGAN各种特殊的架构設计也能做到不错的效果比如如果大家一起拿掉Batch Normalization的话,DCGAN就崩了:

如果WGAN和原始GAN都使用多层全连接网络(MLP)不用CNN,WGAN质量会变差些但是原始GAN不仅质量变得更差,而且还出现了collapse mode即多样性不足:

第三,在所有WGAN的实验中未观察到collapse mode作者也只说应该是解决了,

最后补充一点论文没提到但是我个人觉得比较微妙的问题。判别器所近似的Wasserstein距离能够用来指示单次训练中的训练进程这个没错;接着作者又说它可以用于仳较多次训练进程,指引调参我倒是觉得需要小心些。比如说我下次训练时改了判别器的层数、节点数等超参判别器的拟合能力就必嘫有所波动,再比如说我下次训练时改了生成器两次迭代之间判别器的迭代次数,这两种常见的变动都会使得Wasserstein距离的拟合误差就与上次鈈一样那么这个拟合误差的变动究竟有多大,或者说不同的人做实验时判别器的拟合能力或迭代次数相差实在太大那它们之间还能不能直接比较上述指标,我都是存疑的

进一步指出,相比于判别器迭代次数的改变对判别器架构超参的改变会直接影响到对应的Lipschitz常数,進而改变近似Wasserstein距离的倍数前后两轮训练的指标就肯定不能比较了,这是需要在实际应用中注意的对此我想到了一个工程化的解决方式,不是很优雅:取同样一对生成分布和真实分布让前后两个不同架构的判别器各自拟合到收敛,看收敛到的指标差多少倍可以近似认為是后面的相对前面的变化倍数,于是就可以用这个变化倍数校正前后两轮训练的指标

Goodfellow提出的原始GAN两种形式各自的问题,第一种形式等價在最优判别器下等价于最小化生成分布与真实分布之间的JS散度由于随机生成分布很难与真实分布有不可忽略的重叠以及JS散度的突变特性,使得生成器面临梯度消失的问题;第二种形式在最优判别器下等价于既要最小化生成分布与真实分布直接的KL散度又要最大化其JS散度,相互矛盾导致梯度不稳定,而且KL散度的不对称性使得生成器宁可丧失多样性也不愿丧失准确性导致collapse

WGAN前作针对分布重叠问题提出了一個过渡解决方案,通过对生成样本和真实样本加噪声使得两个分布产生重叠理论上可以解决训练不稳定的问题,可以放心训练判别器到接近最优但是未能提供一个指示训练进程的可靠指标,也未做实验验证

WGAN本作引入了Wasserstein距离,由于它相对KL散度与JS散度具有优越的平滑特性理论上可以解决梯度消失问题。接着通过数学变换将Wasserstein距离写成可求解的形式利用一个参数数值范围受限的判别器神经网络来最大化这個形式,就可以近似Wasserstein距离在此近似最优判别器下优化生成器使得Wasserstein距离缩小,就能有效拉近生成分布与真实分布WGAN既解决了训练不稳定的問题,也提供了一个可靠的训练进程指标而且该指标确实与生成样本的质量高度相关。作者对WGAN进行了实验验证

(可自由转载,注明来源和作者即可

}

我要回帖

更多关于 论文可以用源代码吗 的文章

更多推荐

版权声明:文章内容来源于网络,版权归原作者所有,如有侵权请点击这里与我们联系,我们将及时删除。

点击添加站长微信