Diffusion-GAN: Training GANs with Diffusion
Zhendong Wang Affiliation: The University of Texas at Austin, Microsoft Azure AI{zhendong.wang, Huangjie Zheng Affiliation: The University of Texas at Austin, Microsoft Azure AI{zhendong.wang, Pengcheng He Affiliation: The University of Texas at Austin, Microsoft Azure AI{zhendong.wang, Weizhu Chen Affiliation: The University of Texas at Austin, Microsoft Azure AI{zhendong.wang, Mingyuan Zhou
Abstract
Generative adversarial networks (GANs) are challenging to train stably, and a promising remedy of injecting instance noise into the discriminator input has not been very effective in practice. In this paper, we propose Diffusion-GAN, a novel GAN framework that leverages a forward diffusion chain to generate Gaussian-mixture distributed instance noise. Diffusion-GAN consists of three components, including an adaptive diffusion process, a diffusion timestep-dependent discriminator, and a generator. Both the observed and generated data are diffused by the same adaptive diffusion process. At each diffusion timestep, there is a different noise-to-data ratio and the timestep-dependent discriminator learns to distinguish the diffused real data from the diffused generated data. The generator learns from the discriminator’s feedback by backpropagating through the forward diffusion chain, whose length is adaptively adjusted to balance the noise and data levels. We theoretically show that the discriminator’s timestep-dependent strategy gives consistent and helpful guidance to the generator, enabling it to match the true data distribution. We demonstrate the advantages of Diffusion-GAN over st
原文 arXiv:2206.02262;中英对照 + 大白话阅读 https://aha.fim.ai/paper/2206.02262v4