机器学习与深度学习系列连载: 第四部分 对抗网络GAN (五) 对抗网络 Wasserstein GAN

对抗网络 Wasserstein GAN

1. 传统Traditional GAN的问题

1. JS 距离衡量存在问题

在大多数例子中 P G P_G PG P d a t a P_{data} Pdata 中间是不重叠的

机器学习与深度学习系列连载: 第四部分 对抗网络GAN (五) 对抗网络 Wasserstein GAN_第1张图片

生成器的目的就是要让 P G P_G PG接近 P d a t a P_{data} Pdata,那么GAN中的JS divergence 无法有一个过渡的过程,在 P G P_G PG接近 P d a t a P_{data} Pdata接近之前,他们的距离适中是一个固定的值

机器学习与深度学习系列连载: 第四部分 对抗网络GAN (五) 对抗网络 Wasserstein GAN_第2张图片
我们需要一个将 P G P_G PG P d a t a P_{data} Pdata距离衡量的一种方式

2. “挖掘机距离”Earth Mover’s Distance

这里的“挖掘机”可不是蓝翔哦,不过在概率密度分布上,和真实的挖掘机是一样的,“铲土”!
机器学习与深度学习系列连载: 第四部分 对抗网络GAN (五) 对抗网络 Wasserstein GAN_第3张图片

如何铲土呢:

我们看P和Q两个概率分布:(图片来自 https://vincentherrmann.github.io/blog/wasserstein)
P通过铲土的形式(方式1),将概率密度函数的图形,向Q靠拢。
机器学习与深度学习系列连载: 第四部分 对抗网络GAN (五) 对抗网络 Wasserstein GAN_第4张图片

但是铲土的方式由很多,这种放方式也行( 方式2)

机器学习与深度学习系列连载: 第四部分 对抗网络GAN (五) 对抗网络 Wasserstein GAN_第5张图片

它们之间的区别是,方式1的运土的距离比方式2的短

我们的目标就是要找到,运土过程中的最短距离:
机器学习与深度学习系列连载: 第四部分 对抗网络GAN (五) 对抗网络 Wasserstein GAN_第6张图片

  • 横轴一行加起来就是P对应的概率分布(土的多少)
  • 纵轴一列加起来就是Q对应的概率分布(土的多少)
    我们有:
    在这里插入图片描述
    我们需要找到,一个最佳plan
    机器学习与深度学习系列连载: 第四部分 对抗网络GAN (五) 对抗网络 Wasserstein GAN_第7张图片

机器学习与深度学习系列连载: 第四部分 对抗网络GAN (五) 对抗网络 Wasserstein GAN_第8张图片

找到最佳的迁移方案往往需要遍历所有的方案,这种办法显然是效率最低的。

3. WGAN

WGAN 就是用 Wasserstein distance 来衡量 P d a t a P_{data} Pdata P G P_{G} PG

衡量的方式是:

机器学习与深度学习系列连载: 第四部分 对抗网络GAN (五) 对抗网络 Wasserstein GAN_第9张图片

D 属于 1-Lipschitz 函数,其实可以理解为D是平滑的。
机器学习与深度学习系列连载: 第四部分 对抗网络GAN (五) 对抗网络 Wasserstein GAN_第10张图片

由于为了让V(G,D)最大,V的公式让real最大,让generated最小,但是需要限制。如果generated的结果接近无权小,V的虽然最大,但是没有达到GAN的最初目的。
机器学习与深度学习系列连载: 第四部分 对抗网络GAN (五) 对抗网络 Wasserstein GAN_第11张图片

所以有Weight Clipping 的技巧(Martin Arjovsky, et al., arXiv, 2017):

强制性让所有的参数w在-c到c之间,if w > c, w = c;
if w < -c, w = -c

4. Improved WGAN(WGAN-GP)

一种和Lipschitz 函数等价的计算方式:在x所有区域内,D(x)的导数都小于或等于1

在这里插入图片描述

但是x的范围比较大,在整个区域内进行限制有点困难,那么我们就找到一个区域叫做penalty,在penalty中进行限制。

机器学习与深度学习系列连载: 第四部分 对抗网络GAN (五) 对抗网络 Wasserstein GAN_第12张图片
penalty 是怎么得到的呢? P d a t a P_{data} Pdata P G P_{G} PG之间的连线的中点。

机器学习与深度学习系列连载: 第四部分 对抗网络GAN (五) 对抗网络 Wasserstein GAN_第13张图片

5. WGAN 整体算法

机器学习与深度学习系列连载: 第四部分 对抗网络GAN (五) 对抗网络 Wasserstein GAN_第14张图片

本专栏图片、公式很多来自台湾大学李宏毅老师的深度学习课程,在这里,感谢这些经典课程,向李老师致敬!

你可能感兴趣的:(深度学习,神经网络,GAN,对抗网络)