Contrastive self-supervised representation learning framework for metal surface defect detection

Mahe Zabin,Anika Nahian Binte Kabir,Muhammad Khubayeeb Kabir,Ho-Jin Choi,Jia Uddin
DOI: https://doi.org/10.1186/s40537-023-00827-z
2023-09-28
Journal Of Big Data
Abstract:Automated detection of defects on metal surfaces is crucial for ensuring quality control. However, the scarcity of labeled datasets for emerging target defects poses a significant obstacle. This study proposes a self-supervised representation-learning model that effectively addresses this limitation by leveraging both labeled and unlabeled data. The proposed model was developed based on a contrastive learning framework, supported by an augmentation pipeline and a lightweight convolutional encoder. The effectiveness of the proposed approach for representation learning was evaluated using an unlabeled pretraining dataset created from three benchmark datasets. Furthermore, the performance of the proposed model was validated using the NEU metal surface-defect dataset. The results revealed that the proposed method achieved a classification accuracy of 97.78%, even with fewer trainable parameters than the benchmark models. Overall, the proposed model effectively extracted meaningful representations from unlabeled image data and can be employed in downstream tasks for steel defect classification to improve quality control and reduce inspection costs.
What problem does this paper attempt to address?