基于GAN的天文图像生成研究  

Study on Star-Galaxy Image Generation Method Based on GAN

作　　者：张光华 王福豹[1] 段渭军[1] ZHANG Guanghua;WANG Fubao;DUAN Weijun(School of Electronics and Information, Northwestern Polytechnical University, Xi′an 710072, China)

机构地区：[1]西北工业大学电子信息学院,陕西西安710072

出　　处：《西北工业大学学报》2019年第2期315-322,共8页Journal of Northwestern Polytechnical University

摘　　要：生成对抗网络(GAN)被广泛应用于图像生成。生成恒星和星系图像对预测未知恒星和星系有着重要的意义。首次将GAN用于生成天文图像,给出了天文图像生成的GAN模型结构;设计了GAN训练的策略;为了提高GAN的稳定性,提出了改进的神经元抛弃方法,通过网格搜索法对模型中的部分高级参数进行了优化,并采用了韦氏距离对损失函数进行了改进。以斯隆数字巡天数据库(SDSS)中的恒星以及星系图像作为训练图像,采用改进方法和原始GAN分别生成了2种不同分辨率的恒星和星系图像,并进行了对比,验证了改进方法的有效性。GAN technology has been widely used in image generation field. Generating images of stars and galaxy is of great significance for the prediction of unknown stars and galaxy. GAN has been used to generate star-galaxy images in this paper;the GAN model structure was built;the training strategy for GAN was designed;in order to stabilize the training procedure, we proposed a gird search method for the optimization of several hyper-parameters and an improved neuron discard method, EM-distance was used to modify the loss function in original GAN model. Taking the star-galaxy images in the Sloan digital sky survey (SDSS) as the training dataset, the improved method proposed in this paper and the original GAN were respectively used to generate two kinds of stars and galaxy images with different resolutions, and the comparison has been made to verify the effectiveness of the improved method.

关 键 词：生成对抗网络 恒星和星系图像 训练稳定 损失函数 

分 类 号：TP183[自动化与计算机技术—控制理论与控制工程]