HyperAIHyperAI

Command Palette

Search for a command to run...

改进的Wasserstein生成对抗网络训练方法

Ishaan Gulrajani Faruk Ahmed Martin Arjovsky Vincent Dumoulin Aaron Courville

摘要

生成对抗网络(GANs)是一种强大的生成模型,但其训练过程存在不稳定性问题。最近提出的Wasserstein GAN(WGAN)在实现GAN的稳定训练方面取得了进展,但在某些情况下仍然可能生成低质量样本或无法收敛。我们发现这些问题通常是由WGAN中使用的权重裁剪方法来强制执行判别器的Lipschitz约束所引起的,这可能导致不良行为。为此,我们提出了一种替代权重裁剪的方法:对判别器相对于其输入的梯度范数进行惩罚。我们的方法在性能上优于标准的WGAN,并且几乎无需超参数调整即可实现多种GAN架构的稳定训练,包括101层残差网络(ResNets)和离散数据上的语言模型。此外,我们在CIFAR-10和LSUN卧室数据集上也实现了高质量的图像生成。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
改进的Wasserstein生成对抗网络训练方法 | 论文 | HyperAI超神经