Unsupervised method for detecting surface defects in steel based on joint optimization of pseudo-labeling and clustering

Abstract

Abstract Advances in the field of measurement science and technology have improved the detection of defects in industrial production. One of the key challenges in steel plate surface defect detection is the need to quickly detect a small number of defects in an overwhelmingly defect-free sample. Unlike supervised learning, which relies heavily on precise sample labeling, unsupervised learning leverages its inherent learning capabilities for detection. This paper introduces an innovative method for smart steel diagnosis, integrating joint optimization of feature extraction and clustering. The proposed approach merges mini-batch K-means clustering with a feature extraction network to acquire pseudo-label information for current images. It employs a multi-view transformation strategy, enabling classification through the optimized feedback from pseudo-labels. This method allows the network to self-optimize the distinction of image features through backpropagation. The method exhibits a mere 4% classification failure rate for steel surface images. This significant reduction in additional data processing requirements enhances the inspection system’s efficiency and accuracy. Furthermore, the versatility of this method extends beyond steel defect diagnosis. It holds potential for application in various engineering domains, particularly in scenarios characterized by data imbalance.