An Ultrasonic Signal Recognition Method for Roll Defects Based on C-GAN and CNN-Attention

Abstract

This paper proposes a roll defect recognition method based on C-GAN and CNN-Attention, addressing the challenges of limited data and low recognition accuracy in ultrasonic defect detection for rolls. Initially, an ultrasonic testing experimental system is employed to inspect artificially prepared roll defect samples, leading to the collection of actual defect data. Subsequently, a C-GAN data augmentation model is developed to learn the distribution patterns of various defects, generating high-quality new samples that align with the distribution of each defect type, thereby expanding the training dataset. Utilizing this augmented data, a convolutional neural network defect classification method that incorporates an attention mechanism is designed to further enhance prediction accuracy. By integrating an attention module to assign weights to each feature channel, improved feature representations are achieved, optimizing the learning mechanism of the CNN. The model attains a recognition accuracy of  95.83%, demonstrating the effectiveness of this method in roll defect recognition applications.