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

更快的投影GAN:迈向更快的少样本图像生成

Faster Projected GAN: Towards Faster Few-Shot Image Generation

North China University of Technology(北方工业大学);Chinese PLA General Hospital(中国人民解放军总医院)

Chuang Wang, Zhengping Li, Yuwen Hao, Lijun Wang, Xiaoxue Li

AI总结:

针对GAN训练慢、资源消耗大的问题,提出Faster Projected GAN,通过深度可分离卷积改进生成器,在少样本数据集上实现20%提速和15%内存节省,FID损失不增。

摘要:

为了解决图像生成中GAN网络训练时间长、计算资源消耗大和参数量庞大的问题,本文基于Projected GAN提出了一种改进的GAN网络模型,命名为Faster Projected GAN。所提出的网络主要关注Projected GAN生成器的改进。通过引入深度可分离卷积(DSC),减少了Projected GAN的参数数量,加速了训练速度,并节省了内存。实验结果表明,在ffhq-1k、art-painting、Landscape等少样本图像数据集上,实现了20%的速度提升和15%的内存节省。同时,FID损失减少或无损失,模型参数量得到更好的控制。同时,在公共数据集较少的特殊场景(如地震场景)的小样本图像生成任务中,也实现了显著的训练速度提升。

英文摘要:

In order to solve the problems of long training time, large consumption of computing resources and huge parameter amount of GAN network in image generation, this paper proposes an improved GAN network model, which is named Faster Projected GAN, based on Projected GAN. The proposed network is mainly focuses on the improvement of generator of Projected GAN. By introducing depth separable convolution (DSC), the number of parameters of the Projected GAN is reduced, the training speed is accelerated, and memory is saved. Experimental results show that on ffhq-1k, art-painting, Landscape and other few-shot image datasets, a 20% speed increase and a 15% memory saving are achieved. At the same time, FID loss is less or no loss, and the amount of model parameters is better controlled. At the same time, significant training speed improvement has been achieved in the small sample image generation task of special scenes such as earthquake scenes with few public datasets.

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

↑