论文标题
基于泛化的界限理论,用于改善多尺度GAN的自适应对抗训练方法
Adaptive adversarial training method for improving multi-scale GAN based on generalization bound theory
论文作者
论文摘要
近年来,已经提出了多尺度生成对抗网络(GAN)来建立基于单个样本的通用图像处理模型。限制样本量,多尺度gan的趋势很难融合到全球最佳距离,最终导致其能力的局限性。在本文中,我们率先将Pac-Bayes广义的理论引入了不同对抗训练方法下的特定模型的训练分析,该方法可以在指定的多尺度GAN结构的通用误差上获得非易变的上限。基于我们发现在不同的对抗攻击和不同训练状态下绑定的概括误差的急剧变化,我们提出了一种自适应训练方法,可以极大地提高多规模gan的图像操纵能力。最终的实验结果表明,本文中我们的自适应训练方法极大地有助于提高多尺度gan在多个图像操纵任务上产生的图像质量。特别是,对于图像超分辨率恢复任务,由建议方法训练的多尺度GAN模型可实现100%的自然图像质量评估器(NIQE)的降低,而根平方误差(RMSE)减少了60%,这比许多在大型数据集中训练的模型都更好。
In recent years, multi-scale generative adversarial networks (GANs) have been proposed to build generalized image processing models based on single sample. Constraining on the sample size, multi-scale GANs have much difficulty converging to the global optimum, which ultimately leads to limitations in their capabilities. In this paper, we pioneered the introduction of PAC-Bayes generalized bound theory into the training analysis of specific models under different adversarial training methods, which can obtain a non-vacuous upper bound on the generalization error for the specified multi-scale GAN structure. Based on the drastic changes we found of the generalization error bound under different adversarial attacks and different training states, we proposed an adaptive training method which can greatly improve the image manipulation ability of multi-scale GANs. The final experimental results show that our adaptive training method in this paper has greatly contributed to the improvement of the quality of the images generated by multi-scale GANs on several image manipulation tasks. In particular, for the image super-resolution restoration task, the multi-scale GAN model trained by the proposed method achieves a 100% reduction in natural image quality evaluator (NIQE) and a 60% reduction in root mean squared error (RMSE), which is better than many models trained on large-scale datasets.