基于谱约束的生成对抗网络图像数据生成研究  

Image Data Generation Based on Generative Adversarial Network

作　　者：颜贝 张建林[1,2] YAN Bei;ZHANG Jianlin(The Institute of Optics and Electronics of the Chinese Academy of Sciences,Chengdu 610209,CHN;University of Chinese Academy of Sciences,Beijing 100049,CHN)

机构地区：[1]中国科学院光电技术研究所,成都610209 [2]中国科学院大学,北京100049

出　　处：《半导体光电》2019年第6期896-901,共6页Semiconductor Optoelectronics

基　　金：国家重点研发计划项目;国家“863”计划项目(18-H863-02-xx);重大专项基金项目(G158207)

摘　　要：数据匮乏是深度学习面临的一大难题。利用生成对抗网络(GAN)能够基于语义生成新的图像数据这一特性,提出一种基于谱约束的生成对抗网络图像数据生成方法,该方法针对卷积生成对抗网络模型易崩溃不收敛的问题,从每层神经网络的参数矩阵W的谱范数角度出发,引入谱范数归一化网络参数矩阵,将网络梯度限制在固定范围内,减缓判别网络收敛速度,从而提高GAN的训练稳定性。实验表明,通过该方法生成的数据相比原始GAN以及DCGAN、WGAN等生成的图像样本数据在图像识别网络中具有更高的准确率,能够对少量样本数据进行有效扩充。Insufficient data is a big problem for deep learning.In this paper,based on the feature that the generative adversarial networks(GAN)can semantically generate new data,a new GAN image data generation method based on spectral restriction was proposed.For the problems of collapse and divergence of deep convolution generation adversarial networks(DCGAN),this proposed method starts from the spectral norm of the parameter matrix W of each layer of the neural network,and introduces the spectral norm to normalize the network parameter matrix,thus the network gradient is limited to a fixed range,which slows down the discriminative network convergence speed and improves the training stability of GAN.Experimental results show that in the image recognition network,the proposed method presents higher accuracy than original GAN,DCGAN,WGAN and other methods,expanding a small amount of sample data efficiently.

关 键 词：深度学习 生成对抗网络 数据生成 卷积神经网络 

分 类 号：TP301.6[自动化与计算机技术—计算机系统结构]