Abstract: In order to improve the accuracy of steel surface defect detection in small sample environment, a new method of steel surface defect detection based on the improved auxiliary classifier generative adversarial network (ACGAN) is proposed. Firstly, the residual block is used to optimize the network of ACGAN to improve the feature extraction ability of the model; Secondly, in order to improve the stability of model training, spectral norm normalization is added to the convolution layer of the network to prevent abnormal gradient changes of the model. Then, the loss function of discriminator is optimized based on positive-unlabeled classification to improve the quality of generated samples. At the same time, a gradient penalty is added to the loss function to constrain the gradient of the discriminator in order to alleviate the mode collapse of the Generative Adversarial Network. Finally, the sample expansion is realized through the adversarial optimization training of generator and discriminator. We conducted experiments on steel surface defect datasets to validate the proposed method can accurately and effectively detect steel surface defects in a small sample environment. Compared with the classical support vector machine, ResNet50 and some small sample classification models, the proposed method has higher detection accuracy.

Key words: steel surface defect detection, auxiliary classification generation adversarial network, small sample, gradient penalty