WGAN-GP:更容易训练的GAN

  WGAN-GP又称为具有梯度惩罚的WGAN,一般可以替代WGAN

1.WGAN:基于推土机距离原理的GAN

(1)原始GAN的问题及原因

  实际训练中,GAN存在训练困难、生成器和判别器的loss无法指示训练过程、生成样本缺乏多样性等问题。这与GAN的机制有关。
  GAN最终达到对抗的纳什均衡只是一个理想状态,而实际情况中得到的结果都是中间状态。大部分情况是,随着训练的次数越多判别器D的效果越好,一直可以将生成器G的输出和真实样本区分开。
  这是因为生成器G是从低维向高位空间映射,其生成的样本分布空间Pg难以充满整个真实样本的分布空间Pr。两个分布没有重叠的部分。
  这样会导致什么后果呢?如果D训练的很好,G的loss这无法训练到最下值,梯度消失,最后只会无限接近log2,相当于Pg和Pr的距离;而D训练的不好时,会导致生成器的梯度不准,loss到处跑。

(2)WGAN介绍

  WGAN的思想是将生成的模拟样本分布Pg和原始样本分布Pr组合起来,当成所有可能的联合分布集合。然后可以从中采样得到真实样本与模拟样本,并能够计算二者的距离,还可以算出距离的期望值。这样就可以通过训练,让网络在所有可能的联合分布中对这个期望值取下界的方向优化,也就是将两个分布的集合拉到一起。这样原来的判别器就不再是判别真伪的功能,而是计算两个分布集合距离的功能。所以称为评论器更加合适,sigmod会影响结果,需要去掉。
  为了实现计算Wassertein距离的功能,将这部分交给神经网络去拟合。
∣ f ( x 1 ) − f ( x 2 ) ∣ ≤ k ∣ x 1 − x 2   ∣ |f(x_1)-f(x_2)|\le k|x_1-x_2\ | f(x1)f(x2)kx1x2 
   f ( x ) f(x) f(x)可以理解为神经网络的计算,让判别器实现将 f ( x 1 ) f(x_1) f(x1) f ( x 2 ) f(x_2) f(x2)的距离变换成 x 1 − x 2 x_1-x_2 x1x2的绝对值×k。k代表f(x)的Lipschitz常数,这样两个分布集合的距离就可以表示成D(real)-D(G(x))的绝对值×k,这个k可以理解为梯度,即在神经网络分f(x)中x的梯度绝对值会小于k。
  将k忽略整理后可以得到二者分布的式子。
L = D ( r e a l ) − D ( G ( x ) ) L=D(real)-D(G(x)) L=D(real)D(G(x))
  现在要做的就是将L当做目标来计算loss,G将希望生成的结果Pg越来越接近Pr,所以希望通过训练让距离L最小化。因为G与第一项无关,所以G的loss可以简写为。
G ( l o s s ) = − D ( G ( x ) ) G(loss)=-D(G(x)) G(loss)=D(G(x))
  而D的任务是区分它们,所以希望二者距离变大,所以loss取反。
D ( l o s s ) = D ( G ( x ) ) − D ( r e a l ) D(loss)=D(G(x))-D(real) D(loss)=D(G(x))D(real)
  同样,通过D的loss值也看出G的生成质量,即loss越小代表距离越小,生成的质量越高。
  而对于前面的梯度限制,WGAN直接使用截断的方式,后来升级为WGAN-GP。

real_X = tf.placeholder(tf.float32, shape=[batch_size, mnist_dim])
random_X = tf.placeholder(tf.float32, shape=[batch_size, random_dim])
# 生成的模拟数据
random_Y = G(random_X)

D_loss = tf.reduce_mean(D(random_Y)) -tf.reduce_mean(D(real_X)) 
G_loss = -tf.reduce_mean(D(random_Y))

  截断方法指每次更新完判别器参数之后,检查判别器的所有参数的绝对值有没有超过阀值,比如0.01,如果有就把它截断回[-0.01,0.01]内。

2.WGAN-GP:带梯度惩罚项的WGAN

  在判别器中,希望loss尽可能的大,才能来打真假样本的区别,这样会导致在判别器中通过loss计算梯度会沿着loss越来越大的方向变化,然而经过截断后,每个网络参数又被独立的限制了取值范围。这种结果使得所有参数走向极端,要么取最大值如0.01,要么取最下值(如-0.01),判别器没能充分利用自身的模型能力,经过它传给生成器的梯度也会跟着变差。

  WGAN-GP介绍

  WGAN-GP中的GP是梯度惩罚的意思,采用梯度惩罚项,来代替WGAN的截取。
N o r m = t f . g r a d i e n t s ( D ( X i n t e r ) , [ X i n t e r ] ) Norm=tf.gradients\left(D\left(X_inter\right),\left[X_in t e r\right]\right) Norm=tf.gradients(D(Xinter),[Xinter])
g r a d p e n = M S E ( N o r m − k ) grad_pen=MSE(Norm-k) gradpen=MSE(Normk)
  MSE为平方差公式,X_inter为整个联合分布的x取样,即梯度惩罚想grad_pne为求整个联合分布空间的x对应D的梯度与k的平方差。
  判别器尽可能的拉大真假样本的分数差距,希望梯度越大越好,变化幅度越大越好,所以判别器在充分训练之后,其梯度Norm就会k附近。因此可以将上面的loss改成使梯度接近k(k可以是任意的值),将k定义为1,再跟WGAN原来的判别器的loss加权合并,得到新的判别器。
L = D ( r e a l ) − D ( G ( x ) ) + λ × g r a d p e n L=D(real)-D(G(x))+\lambda\times grad_pen L=D(real)D(G(x))+λ×gradpen

eps = tf.random_uniform([batch_size, 1], minval=0., maxval=1.)  #随机数
X_inter = eps*real_X + (1. - eps)*random_Y      #联合分布x取样
grad = tf.gradients(D(X_inter), [X_inter])[0]   #对X_inter取梯度
grad_norm = tf.sqrt(tf.reduce_sum((grad)**2, axis=1))   
grad_pen = 10 * tf.reduce_mean(tf.nn.relu(grad_norm - 1.))  #取与k(1)的平方差

3.实例90:构建WGAN-GP生成MNIST数据集

实例描述

通过使用WGAN-GP网络学习MNIST数据特征,并生成以假乱真的MNIST数据集

1.引入头文件并加载MNIST数据


import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
import os
import numpy as np
from scipy import misc,ndimage
import tensorflow.contrib.slim as slim

mnist = input_data.read_data_sets("/data/", one_hot=True)

2.定义生成器与判别器

  网络的复杂部分都在loss方面,所以生成器G和判别器D就会相对简单点,各有3个全连接层。生成器最终输出与MNIST图片相同维度的数据最为模拟样本。判别器的输出就不在需要有激活函数,输出维度为1的数值表示其结果。

def G(x):
    reuse = len([t for t in tf.global_variables() if t.name.startswith('generator')]) > 0
    with tf.variable_scope('generator', reuse = reuse):
        x = slim.fully_connected(x, 32,activation_fn = tf.nn.relu)
        x = slim.fully_connected(x, 128,activation_fn = tf.nn.relu)
        x = slim.fully_connected(x, mnist_dim,activation_fn = tf.nn.sigmoid)
    return x


def D(X):
    reuse = len([t for t in tf.global_variables() if t.name.startswith('discriminator')]) > 0
    with tf.variable_scope('discriminator', reuse=reuse): 
        X = slim.fully_connected(X, 128,activation_fn = tf.nn.relu)
        X = slim.fully_connected(X, 32,activation_fn = tf.nn.relu)
        X = slim.fully_connected(X, 1,activation_fn = None)
    return X

3.定义网络模型与loss

  生成的模拟数据为random_Y,与前面所描述的一致;生成器的loss为-D(random_Y);而判别器的loss为D(random_Y)-D(real_X)再加上一个联合分布样本梯度惩罚项grad_pen;惩罚项的采样X_inter由一部分Pg分布和一部分Pr分布组成。同时对D(X_inter)求梯度得到grad_pen.

real_X = tf.placeholder(tf.float32, shape=[batch_size, mnist_dim])
random_X = tf.placeholder(tf.float32, shape=[batch_size, random_dim])
# 生成的模拟数据
random_Y = G(random_X)

eps = tf.random_uniform([batch_size, 1], minval=0., maxval=1.)  #随机数
X_inter = eps*real_X + (1. - eps)*random_Y      #联合分布x取样
grad = tf.gradients(D(X_inter), [X_inter])[0]   #对X_inter取梯度
grad_norm = tf.sqrt(tf.reduce_sum((grad)**2, axis=1))   
grad_pen = 10 * tf.reduce_mean(tf.nn.relu(grad_norm - 1.))  #取与k(1)的平方差

D_loss = tf.reduce_mean(D(random_Y)) -tf.reduce_mean(D(real_X))  + grad_pen
G_loss = -tf.reduce_mean(D(random_Y))

4.定义优化器并开始训练

# 获得各个网络中各自的训练参数
t_vars = tf.trainable_variables()
d_vars = [var for var in t_vars if 'discriminator' in var.name]
g_vars = [var for var in t_vars if 'generator' in var.name]
print(len(t_vars),len(d_vars))

D_solver = tf.train.AdamOptimizer(1e-4, 0.5).minimize(D_loss, var_list=d_vars)
G_solver = tf.train.AdamOptimizer(1e-4, 0.5).minimize(G_loss, var_list=g_vars)

training_epochs =100

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())    
    if not os.path.exists('out/'):
        os.makedirs('out/')
        
    for epoch in range(training_epochs):

        total_batch = int(mnist.train.num_examples/batch_size)

        # 遍历全部数据集
        for e in range(total_batch):
            for i in range(5):
                real_batch_X,_ = mnist.train.next_batch(batch_size)
                random_batch_X = np.random.uniform(-1, 1, (batch_size, random_dim))
                _,D_loss_ = sess.run([D_solver,D_loss], feed_dict={real_X:real_batch_X, random_X:random_batch_X})
            
            random_batch_X = np.random.uniform(-1, 1, (batch_size, random_dim))
            _,G_loss_ = sess.run([G_solver,G_loss], feed_dict={random_X:random_batch_X})
        
        if epoch % 10 == 0:
            print ('epoch %s, D_loss: %s, G_loss: %s'%(epoch, D_loss_, G_loss_))
            n_rows = 6
            check_imgs = sess.run(random_Y, feed_dict={random_X:random_batch_X}).reshape((batch_size, width, height))[:n_rows*n_rows]
            imgs = np.ones((width*n_rows+5*n_rows+5, height*n_rows+5*n_rows+5))
            #print(np.shape(imgs))#(203, 203)
    
            for i in range(n_rows*n_rows):
                num1 = (i%n_rows)
                num2 = np.int32(i/n_rows)
                imgs[5+5*num1+width*num1:5+5*num1+width+width*num1, 5+5*num2+height*num2:5+5*num2+height+height*num2] = check_imgs[i]
    
            misc.imsave('out/%s.png'%(epoch/10), imgs)

    print("完成!")

  在会话优先让判别器学习次数多一些,让判别器每次训练5次,生成器优化一次,WGAN-GP不会因为判别器准确度太高而引起生成器梯度消失的问题,好的判别器只会让生成器有更好的模拟效果。

  • 19
    点赞
  • 72
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

无微の大白

你的鼓励将是我创作的最大动力

¥2 ¥4 ¥6 ¥10 ¥20
输入1-500的整数
余额支付 (余额:-- )
扫码支付
扫码支付:¥2
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值