Few-Shot Defect Segmentation Leveraging Abundant Normal Training Samples Through Normal Background Regularization and Crop-and-Paste Operation

07/18/2020
by   Dongyun Lin, et al.
0

In industrial product quality assessment, it is essential to determine whether a product is defect-free and further analyze the severity of anomality. To this end, accurate defect segmentation on images of products provides an important functionality. In industrial inspection tasks, it is common to capture abundant defect-free image samples but very limited anomalous ones. Therefore, it is critical to develop automatic and accurate defect segmentation systems using only a small number of annotated anomalous training images. This paper tackles the challenging few-shot defect segmentation task with sufficient normal (defect-free) training images but very few anomalous ones. We present two effective regularization techniques via incorporating abundant defect-free images into the training of a UNet-like encoder-decoder defect segmentation network. We first propose a Normal Background Regularization (NBR) loss which is jointly minimized with the segmentation loss, enhancing the encoder network to produce distinctive representations for normal regions. Secondly, we crop/paste defective regions to the randomly selected normal images for data augmentation and propose a weighted binary cross-entropy loss to enhance the training by emphasizing more realistic crop-and-pasted augmented images based on feature-level similarity comparison. Both techniques are implemented on an encoder-decoder segmentation network backboned by ResNet-34 for few-shot defect segmentation. Extensive experiments are conducted on the recently released MVTec Anomaly Detection dataset with high-resolution industrial images. Under both 1-shot and 5-shot defect segmentation settings, the proposed method significantly outperforms several benchmarking methods.

READ FULL TEXT

page 1

page 3

page 4

page 5

page 6

page 8

page 10

page 11

research
09/26/2022

Self-Supervised Guided Segmentation Framework for Unsupervised Anomaly Detection

Unsupervised anomaly detection is a challenging task in industrial appli...
research
01/06/2020

Semi-supervised Anomaly Detection using AutoEncoders

Anomaly detection refers to the task of finding unusual instances that s...
research
08/22/2023

Random Word Data Augmentation with CLIP for Zero-Shot Anomaly Detection

This paper presents a novel method that leverages a visual-language mode...
research
03/26/2023

WinCLIP: Zero-/Few-Shot Anomaly Classification and Segmentation

Visual anomaly classification and segmentation are vital for automating ...
research
07/28/2022

SPot-the-Difference Self-Supervised Pre-training for Anomaly Detection and Segmentation

Visual anomaly detection is commonly used in industrial quality inspecti...
research
10/22/2019

J Regularization Improves Imbalanced Multiclass Segmentation

We propose a new loss formulation to further advance the multiclass segm...
research
03/04/2023

Few-Shot Defect Image Generation via Defect-Aware Feature Manipulation

The performances of defect inspection have been severely hindered by ins...

Please sign up or login with your details

Forgot password? Click here to reset