3 个月前

面向高保真少样本图像合成的更快且更稳定的GAN训练

面向高保真少样本图像合成的更快且更稳定的GAN训练

摘要

在高保真图像上训练生成对抗网络(GAN)通常需要大规模GPU集群以及海量训练图像。本文研究了在计算成本最低的前提下,针对GAN的少样本图像生成任务。我们提出了一种轻量级GAN结构,在1024×1024分辨率下实现了优异的生成质量。值得注意的是,该模型仅需在单张RTX-2080 GPU上进行数小时的训练即可从零开始收敛,并且在训练样本少于100张的情况下仍能保持稳定一致的性能表现。本工作的核心由两项技术设计构成:一种跳跃连接的通道级激励模块,以及一个以特征编码器方式训练的自监督判别器。我们在涵盖多种图像领域的十三个数据集上进行了实验(数据集与代码已公开于:https://github.com/odegeasslbc/FastGAN-pytorch),结果表明,在数据量和计算资源均受限的情况下,本模型在性能上显著优于当前最先进的StyleGAN2。

基准测试

基准方法指标
image-generation-on-ade-indoorFastGAN
FID: 30.33
image-generation-on-pokemon-1024x1024FastGAN
FID: 56.46
image-generation-on-pokemon-256x256FastGAN
FID: 81.86

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
面向高保真少样本图像合成的更快且更稳定的GAN训练 | 论文 | HyperAI超神经