中文

基于多通道小波滤波的噪声均匀化用于 GAN 高保真样本生成

机器学习 2020-05-15 v1 计算机视觉与模式识别 机器学习

摘要

在典型生成对抗网络(GANs)的生成器中,输入噪声并通过一系列卷积操作生成伪造样本。然而,当前的噪声生成模型仅依赖像素空间的信息,增加了逼近目标分布的难度。幸运的是,久经验证的小波变换能够从图像中分解出多频谱信息。本工作中,我们提出一种新颖的基于多通道小波滤波的 GAN 方法以应对该问题。当在生成器中嵌入小波反卷积层时,所得的 GAN 称为 WaveletGAN,其利用小波反卷积学习具有多通道的滤波,可通过平均操作高效均匀化生成的噪声,从而生成高保真样本。我们通过开放的 GAN 基准工具在 Fashion-MNIST、KMNIST 和 SVHN 数据集上进行了基准实验。结果表明,得益于在这些数据集上取得的最小 FID,WaveletGAN 在生成高保真样本方面具有优异性能。

关键词

引用

@article{arxiv.2005.06707,
  title  = {Noise Homogenization via Multi-Channel Wavelet Filtering for High-Fidelity Sample Generation in GANs},
  author = {Shaoning Zeng and Bob Zhang},
  journal= {arXiv preprint arXiv:2005.06707},
  year   = {2020}
}

备注

12 pages, 2 figures