arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.00430cs.CVcs.LGeess.IV

Training Generative Adversarial Networks in One Stage

Zhejiang University(浙江大学);National University of Singapore(新加坡国立大学);Alibaba Group(阿里巴巴集团);Zhejiang Lab(之江实验室);Stevens Institute of Technology(史蒂文斯理工学院)

Chengchao Shen, Youtan Yin, Xinchao Wang, Xubin Li, Jie Song, Mingli Song

英文摘要:

Generative Adversarial Networks (GANs) have demonstrated unprecedented success in various image generation tasks. The encouraging results, however, come at the price of a cumbersome training process, during which the generator and discriminator are alternately updated in two stages. In this paper, we investigate a general training scheme that enables training GANs efficiently in only one stage. Based on the adversarial losses of the generator and discriminator, we categorize GANs into two classes, Symmetric GANs and Asymmetric GANs, and introduce a novel gradient decomposition method to unify the two, allowing us to train both classes in one stage and hence alleviate the training effort. We also computationally analyze the efficiency of the proposed method, and empirically demonstrate that, the proposed method yields a solid $1.5\times$ acceleration across various datasets and network architectures. Furthermore, we show that the proposed method is readily applicable to other adversarial-training scenarios, such as data-free knowledge distillation. The code is available at https://github.com/zju-vipa/OSGAN.

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

↑