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

UVCGAN v2:一种改进的循环一致生成对抗网络,用于非配对图像到图像转换

UVCGAN v2: An Improved Cycle-Consistent GAN for Unpaired Image-to-Image Translation

Brookhaven National Laboratory(布鲁克海文国家实验室)

Dmitrii Torbunov, Yi Huang, Huan-Hsin Tseng, Haiwang Yu, Jin Huang, Shinjae Yoo, Meifeng Lin, Brett Viren, Yihui Ren

AI总结:

本文提出UVCGAN v2,通过引入现代架构与训练改进,在多个非配对图像转换基准上显著超越GAN和扩散模型,并在CelebA性别转换任务中FID提升超40%。

摘要:

非配对图像到图像(I2I)转换技术旨在以完全无监督的方式寻找两个数据域之间的映射。虽然I2I问题的初始解决方案由生成对抗神经网络(GAN)提供,但扩散模型(DM)目前在I2I转换基准上以Frechet inception distance(FID)指标保持最先进水平。然而,DM存在局限性,例如在训练期间不使用源域的数据,或仅通过简单的逐像素误差来维持源图像与转换后图像之间的一致性。本工作改进了最近的UVCGAN模型,并为其配备了模型架构和训练流程方面的现代进展。由此产生的修订模型在各种基准上显著优于其他基于GAN和DM的先进竞争者。在CelebA的男性到女性转换任务中,与最先进的结果相比,该模型在FID分数上实现了超过40%的提升。本工作还证明了逐像素I2I转换保真度指标的无效性,并建议对其进行修订。代码和训练好的模型可在https://github.com/LS4GAN/uvcgan2获取。

英文摘要:

An unpaired image-to-image (I2I) translation technique seeks to find a mapping between two domains of data in a fully unsupervised manner. While initial solutions to the I2I problem were provided by generative adversarial neural networks (GANs), diffusion models (DMs) currently hold the state-of-the-art status on the I2I translation benchmarks in terms of Frechet inception distance (FID). Yet, DMs suffer from limitations, such as not using data from the source domain during the training or maintaining consistency of the source and translated images only via simple pixel-wise errors. This work improves a recent UVCGAN model and equips it with modern advancements in model architectures and training procedures. The resulting revised model significantly outperforms other advanced GAN- and DM-based competitors on a variety of benchmarks. In the case of Male-to-Female translation of CelebA, the model achieves more than 40% improvement in FID score compared to the state-of-the-art results. This work also demonstrates the ineffectiveness of the pixel-wise I2I translation faithfulness metrics and suggests their revision. The code and trained models are available at https://github.com/LS4GAN/uvcgan2

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

↑