Surface Defect Detection of Aluminum Profiles Based on Multiscale and Self-Attention Mechanisms

Yichuan Shao,Shuo Fan,Qian Zhao,Le Zhang,Haijing Sun
DOI: https://doi.org/10.3390/s24092914
IF: 3.9
2024-05-02
Sensors
Abstract:To address the various challenges in aluminum surface defect detection, such as multiscale intricacies, sensitivity to lighting variations, occlusion, and noise, this study proposes the AluDef-ClassNet model. Firstly, a Gaussian difference pyramid is utilized to capture multiscale image features. Secondly, a self-attention mechanism is introduced to enhance feature representation. Additionally, an improved residual network structure incorporating dilated convolutions is adopted to increase the receptive field, thereby enhancing the network's ability to learn from extensive information. A small-scale dataset of high-quality aluminum surface defect images is acquired using a CCD camera. To better tackle the challenges in surface defect detection, advanced deep learning techniques and data augmentation strategies are employed. To address the difficulty of data labeling, a transfer learning approach based on fine-tuning is utilized, leveraging prior knowledge to enhance the efficiency and accuracy of model training. In dataset testing, our model achieved a classification accuracy of 98.01%, demonstrating significant advantages over other classification models.
engineering, electrical & electronic,instruments & instrumentation,chemistry, analytical
What problem does this paper attempt to address?
This paper primarily addresses various challenges in the detection of surface defects in aluminum profiles, such as multi-scale complexity, sensitivity to lighting changes, occlusion, and noise, by proposing a new model named AluDef-ClassNet. Specifically, the main objectives and contributions of the research are as follows: 1. **Proposing the AluDef-ClassNet Model**: To tackle the challenges in detecting surface defects in aluminum profiles, the authors designed a deep learning model named AluDef-ClassNet, which integrates a Gaussian difference pyramid, a self-attention mechanism, and an improved residual network structure (using dilated convolutions). 2. **Small-Scale Dataset Construction**: A high-quality small-scale dataset of aluminum profile surface defects was obtained using a CCD camera, including various types of surface defects such as non-conductive defects, scratches, transverse concave cracks, pinholes, and dents. 3. **Multi-Scale Feature Extraction**: The Gaussian difference pyramid technique is used to capture multi-scale features of the images, generating difference images at different scales to enhance the ability to extract and represent image details and features. 4. **Introducing the Self-Attention Mechanism**: By introducing the self-attention mechanism, the model can better understand the key information in the images, emphasizing important features by learning attention weights, thereby improving detection performance. 5. **Improved Residual Network Structure**: The improved residual network structure combined with dilated convolutions enhances the training efficiency of the neural network and addresses the problem of gradient vanishing, achieving the goal of efficiently classifying aluminum profile surface defect images on a small-scale imbalanced dataset. 6. **Experimental Validation**: Through experimental validation on the task of aluminum profile surface defect detection, the model demonstrated high accuracy and efficiency. Compared to traditional methods and other deep learning models, it showed significant improvement in classification accuracy, providing a more reliable method for industrial quality control. In summary, this research aims to solve the challenges encountered in the detection of surface defects in aluminum profiles using advanced deep learning techniques, proposing a comprehensive solution that effectively improves detection accuracy and generalization ability.