arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2304.09024cs.CV

关注ATME:判别器均值熵需要被注意

Look ATME: The Discriminator Mean Entropy Needs Attention

German Aerospace Center (DLR)(德国航空航天中心)

Edgardo Solano-Carrillo, Angel Bueno Rodriguez, Borja Carrillo-Perez, Yannik Steiniger, Jannis Stoppe

AI总结:

本文提出ATME模型,通过让生成器感知判别器的熵状态来恢复信息对称性,使GAN稳定收敛至理论最优,并结合扩散模型的去噪机制,在图像到图像转换任务中以更低成本超越现有方法。

摘要:

生成对抗网络(GANs)已成功用于图像合成,但已知其在训练过程中面临不稳定问题。相比之下,概率扩散模型(DMs)稳定且能生成高质量图像,但代价是昂贵的采样过程。在本文中,我们引入一种简单方法,使GANs能够稳定收敛到其理论最优值,同时引入来自DMs的去噪机制。这些模型被组合成一个更简单的模型(ATME),该模型在推理时仅需一次前向传播,使得预测比DMs和流行的GANs更廉价且更准确。ATME打破了大多数GAN模型中存在的一种信息不对称,即判别器具有关于生成器在何处失败的空间知识。为了恢复信息对称性,生成器被赋予判别器熵状态的知识,并利用该知识使对抗博弈趋向均衡收敛。我们在多个图像到图像转换任务中展示了我们方法的强大能力,以更低的成本表现出优于最先进方法的性能。代码可在 https://github.com/DLR-MI/atme 获取。

英文摘要:

Generative adversarial networks (GANs) are successfully used for image synthesis but are known to face instability during training. In contrast, probabilistic diffusion models (DMs) are stable and generate high-quality images, at the cost of an expensive sampling procedure. In this paper, we introduce a simple method to allow GANs to stably converge to their theoretical optimum, while bringing in the denoising machinery from DMs. These models are combined into a simpler model (ATME) that only requires a forward pass during inference, making predictions cheaper and more accurate than DMs and popular GANs. ATME breaks an information asymmetry existing in most GAN models in which the discriminator has spatial knowledge of where the generator is failing. To restore the information symmetry, the generator is endowed with knowledge of the entropic state of the discriminator, which is leveraged to allow the adversarial game to converge towards equilibrium. We demonstrate the power of our method in several image-to-image translation tasks, showing superior performance than state-of-the-art methods at a lesser cost. Code is available at https://github.com/DLR-MI/atme

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑