基于生成对抗网络的太阳能电池缺陷增强方法  

Solar cell defect enhancement method based on generative adversarial network

作　　者：刘坤 文熙 黄闽茗 杨欣欣 毛经坤 LIU Kun;WEN Xi;HUANG Min-ming;YANG Xin-xin;MAO Jing-kun(College of Artificial Intelligence,Hebei University of Technology,Tianjin 300131,China;College of Big Data and Information Engineering,Guizhou University,Guiyang 550025,China;Tianjin Xinsi Technology Company with Limited Liability,Tianjin 300450,China)

机构地区：[1]河北工业大学人工智能与数据科学学院,天津300131 [2]贵州大学大数据与信息工程学院,贵州贵阳550025 [3]天津芯思科技有限公司,天津300450

出　　处：《浙江大学学报（工学版）》2020年第4期684-693,共10页Journal of Zhejiang University：Engineering Science

基　　金：国家自然科学基金资助项目(61403119,61203275);河北省自然科学基金资助项目(F2018202078,F2019202305).

摘　　要：为了解决太阳能电池样本不均衡问题,提出负样本引导生成对抗网络的太阳能电池缺陷样本增强方法.通过在生成对抗模型中引入大量负样本和增加负样本引导损失,促进模型对正样本特征的表达,提升生成样本的多样性;设计自适应的权值约束方法,平衡生成器和判别器的表达能力,提升生成样本的质量.实验结果表明,在太阳能电池电致发光(EL)缺陷数据集上,提出方法的生成质量和检测精度优于深度卷积生成对抗网络(DCGAN)、梯度惩罚Wasserstein距离生成对抗网络(WGAN-GP)和一阶导数生成对抗网络(FOGAN);该方法的F测度较DCGAN、WGAN-GP和FOGAN分别最高提升了10%、8%和5%,具有较好的数据增强性能.在带钢表面缺陷数据集及DAGM 2007公共数据集上,提出方法的性能优于DCGAN、WGAN-GP和FOGAN,具有一定的泛化能力.A solar cells defect sample enhancement method which is negative sample-guided generative adversarial network was proposed in order to solve the problem of sample imbalance of solar cells.The representation ability of positive samples features was promoted and the diversity of generated samples was improved by introducing many negative samples and mixing negative sample guidance loss in the generative adversarial network.An adaptive weight constraint method was designed to balance the representation ability of generators and discriminators,and the quality of generated samples was improved.The experimental results show that the proposed method outperforms deep convolutional generative adversarial network(DCGAN),Wasserstein generative adversarial network-gradient penalty(WGAN-GP)and first-order generative adversarial network(FOGAN)in generation quality and detection accuracy on electroluminescence(EL)defect data sets of solar cells.F-measure of the method was 10%,8%and 5%higher than DCGAN,WGAN-GP and FOGAN respectively,which showed better data enhancement performance.The performance of the proposed method is better than DCGAN,WGAN-GP and FOGAN on strip steel surface defect dataset and DAGM 2007 public dataset,which shows certain generalization ability.

关 键 词：样本不均衡 数据增强 生成对抗网络(GAN) 太阳能电池 负样本引导 

分 类 号：TP391[自动化与计算机技术—计算机应用技术]