RADDA-Net: Residual attention-based dual discriminator adversarial network for surface defect detection

Abstract

Surface defect detection is a critical yet challenging task in the product production process, mainly because of the complex background texture on the material surface, the large intra-class defect differences, and the weak texture defect regions. To address this problem, a Residual Attention-based Dual Discriminator Adversarial Network (RADDA-Net) is developed to accurately detect various types of texture defect regions. Specifically, a residual network-based feature extraction module integrating multi-channel attention mechanism and multi-scale convolutional layers is designed to improve the detection ability of abnormal defects. Besides, to deeply excavate material surface texture information, the pixel attention-based up-sampling blocks are also incorporated into RADDA-Net to enhance the extraction of shallow features. To further improve the detection performance of RADDA-Net, a dual discriminant adversarial training strategy and a comprehensive measurement method are introduced to assist the developed network to identify defects from both macroscopic structure and microscopic edges. Finally, the six evaluation indexes of our proposed method outperformed other state-of-the-art methods on the four defect datasets, among which the highest detection precision, recall, and F-measure are 0.926, 0.916, 0.920, respectively. Therefore, the experimental results consistently verify the detection accuracy and effectiveness of RADDA-Net.