Coarse-to-Fine Few-Shot Defect Recognition With Dynamic Weighting and Joint Metric | IEEE Journals & Magazine | IEEE Xplore

Coarse-to-Fine Few-Shot Defect Recognition With Dynamic Weighting and Joint Metric


Abstract:

Deep-learning-based methods have been widely used in defect recognition and achieved great success. However, deep-learning-based methods need a large-scale dataset. While...Show More

Abstract:

Deep-learning-based methods have been widely used in defect recognition and achieved great success. However, deep-learning-based methods need a large-scale dataset. While in real industrial scenarios, the training samples are always insufficient since the defect data acquisition is difficult and time-consuming. Therefore, in this article, the few-shot learning theory is introduced to address the challenge. We propose to achieve the few-shot defect recognition (FSDR) in a coarse-to-fine manner with dynamic weighting and joint metric. In the coarse-grained phase, following feature embedding, we propose an affine dynamic weighting (ADW) module to control the embedding output of all the channels according to the global context. By the dynamic weighting, the model can extract discriminative features better with learnable affine parameters. In the fine-grained phase, we propose a joint metric method which contains a Kullback–Leibler (K-L) divergence-based covariance metric module (KLCM) and cosine classifier. In this method, KLCM exploits the covariance matrix from the local descriptors to represent the distribution of a special defect class and then measures the similarity between the support and query defect images. A novel FSDR dataset which contains a variety of defects from four different surfaces is constructed to evaluate our method. The results show the state-of-the-art performance compared with the mainstream methods.
Article Sequence Number: 2514910
Date of Publication: 21 July 2022

ISSN Information:

Funding Agency:


Contact IEEE to Subscribe

References

References is not available for this document.