LAUSR.org creates dashboard-style pages of related content for over 1.5 million academic articles. Sign Up to like articles & get recommendations!

A New Cycle-consistent Adversarial Networks With Attention Mechanism for Surface Defect Classification With Small Samples

Photo by robertbye from unsplash

Surface defect detection is the essential process to ensure the quality of products. Surface defect classification (SDC) based on deep learning (DL) has shown its great potential. However, the well-trained… Click to show full abstract

Surface defect detection is the essential process to ensure the quality of products. Surface defect classification (SDC) based on deep learning (DL) has shown its great potential. However, the well-trained SDC model usually requires large training data, and the small intraclass differences between the defect and normal samples also degrades the performance of SDC model. To overcome these drawbacks, this article proposed a new cycle-consistent adversarial networks with attention mechanism (AttenCGAN). First, AttenCGAN is used for synthesizing defect samples to enlarge the samples volume. Second, the attention mechanism is adopted for the feature enhancement by finding the discriminative parts of the samples and enlarging the differences among the samples. AttenCGAN is tested on KolektorSDD and DAGM2007 datasets, and its accuracies are 98.53% and 99.57% with only a few samples. The experiment results show that AttenCGAN outperforms other published SDC methods based on DL and machine learning, which validates its potential.

Keywords: attention mechanism; surface defect; new cycle; defect classification

Journal Title: IEEE Transactions on Industrial Informatics
Year Published: 2022

Link to full text (if available)


Share on Social Media:                               Sign Up to like & get
recommendations!

Related content

More Information              News              Social Media              Video              Recommended



                Click one of the above tabs to view related content.