基于空间特征的生成对抗网络数据生成方法 |
| |
引用本文: | 孙磊,杨宇,毛秀青,汪小芹,李佳欣.基于空间特征的生成对抗网络数据生成方法[J].电子与信息学报,2023(6):1959-1969. |
| |
作者姓名: | 孙磊 杨宇 毛秀青 汪小芹 李佳欣 |
| |
作者单位: | 战略支援部队信息工程大学 |
| |
摘 要: | 传统的生成对抗网络(GAN)在特征图较大的情况下,忽略了原始特征的表示和结构信息,并且生成图像的像素之间缺乏远距离相关性,从而导致生成的图像质量较低。为了进一步提高生成图像的质量,该文提出一种基于空间特征的生成对抗网络数据生成方法(SF-GAN)。该方法首先将空间金字塔网络加入生成器和判别器,来更好地捕捉图像的边缘等重要的描述信息;然后将生成器和判别器进行特征加强,来建模像素之间的远距离相关性。使用CelebA,SVHN,CIFAR-10等小规模数据集进行实验,通过定性和盗梦空间得分(IS)、弗雷歇距离(FID)定量评估证明了所提方法相比梯度惩罚生成对抗网络(WGAN-GP)、自注意力生成对抗网络(SAGAN)能使生成的图像具有更高的质量。并且通过实验证明了该方法生成的数据能够进一步提升分类模型的训练效果。
|
关 键 词: | 生成对抗网络 空间金字塔网络 特征加强 特征图 |
|
|