Reference-Based Autoencoder for Surface Defect Detection
- URL: http://arxiv.org/abs/2211.10060v1
- Date: Fri, 18 Nov 2022 07:13:55 GMT
- Title: Reference-Based Autoencoder for Surface Defect Detection
- Authors: Wei Luo, Haiming Yao, Wenyong Yu and Xue Wang
- Abstract summary: We propose a novel unsupervised reference-based autoencoder (RB-AE) to accurately inspect a variety of textured defects.
artificial defects and a novel pixel-level discrimination loss function are utilized for training to enable the model to obtain pixel-level discrimination ability.
- Score: 7.163582730053925
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Due to the extreme imbalance in the number of normal data and abnormal data,
visual anomaly detection is important for the development of industrial
automatic product quality inspection. Unsupervised methods based on
reconstruction and embedding have been widely studied for anomaly detection, of
which reconstruction-based methods are the most popular. However, establishing
a unified model for textured surface defect detection remains a challenge
because these surfaces can vary in homogeneous and non regularly ways.
Furthermore, existing reconstruction-based methods do not have a strong ability
to convert the defect feature to the normal feature. To address these
challenges, we propose a novel unsupervised reference-based autoencoder (RB-AE)
to accurately inspect a variety of textured defects. Unlike most
reconstruction-based methods, artificial defects and a novel pixel-level
discrimination loss function are utilized for training to enable the model to
obtain pixel-level discrimination ability. First, the RB-AE employs an encoding
module to extract multi-scale features of the textured surface. Subsequently, a
novel reference-based attention module (RBAM) is proposed to convert the defect
features to normal features to suppress the reconstruction of defects. In
addition, RBAM can also effectively suppress the defective feature residual
caused by skip-connection. Next, a decoding module utilizes the repaired
features to reconstruct the normal texture background. Finally, a novel
multiscale feature discrimination module (MSFDM) is employed to defect
detection and segmentation.
Related papers
- Feature Purified Transformer With Cross-level Feature Guiding Decoder For Multi-class OOD and Anomaly Deteciton [22.59400036717643]
We introduce the FUTUREG framework, which incorporates two innovative modules: the Feature Purification Module (FPM) and the CFG Decoder.
The FPM constrains the normality boundary within the latent space to effectively filter out anomalous features.
The CFG Decoder uses layer-wise encoder representations to guide the reconstruction of filtered features, preserving fine-grained details.
arXiv Detail & Related papers (2024-04-30T16:45:51Z) - DiAD: A Diffusion-based Framework for Multi-class Anomaly Detection [55.48770333927732]
We propose a Difusion-based Anomaly Detection (DiAD) framework for multi-class anomaly detection.
It consists of a pixel-space autoencoder, a latent-space Semantic-Guided (SG) network with a connection to the stable diffusion's denoising network, and a feature-space pre-trained feature extractor.
Experiments on MVTec-AD and VisA datasets demonstrate the effectiveness of our approach.
arXiv Detail & Related papers (2023-12-11T18:38:28Z) - Video Anomaly Detection via Spatio-Temporal Pseudo-Anomaly Generation : A Unified Approach [49.995833831087175]
This work proposes a novel method for generating generic Video-temporal PAs by inpainting a masked out region of an image.
In addition, we present a simple unified framework to detect real-world anomalies under the OCC setting.
Our method performs on par with other existing state-of-the-art PAs generation and reconstruction based methods under the OCC setting.
arXiv Detail & Related papers (2023-11-27T13:14:06Z) - Noise-to-Norm Reconstruction for Industrial Anomaly Detection and
Localization [5.101905755052051]
Anomaly detection has a wide range of applications and is especially important in industrial quality inspection.
Reconstruction-based methods use reconstruction errors to detect anomalies without considering positional differences between samples.
In this study, a reconstruction-based method using the noise-to-norm paradigm is proposed, which avoids the invariant reconstruction of anomalous regions.
arXiv Detail & Related papers (2023-07-06T08:06:48Z) - Two-stream Decoder Feature Normality Estimating Network for Industrial
Anomaly Detection [4.772323272202286]
We propose a two-stream decoder network (TSDN) to learn both normal and abnormal features.
We also propose a feature normality estimator (FNE) to eliminate abnormal features and prevent high-quality reconstruction of abnormal regions.
arXiv Detail & Related papers (2023-02-20T06:46:09Z) - Making Reconstruction-based Method Great Again for Video Anomaly
Detection [64.19326819088563]
Anomaly detection in videos is a significant yet challenging problem.
Existing reconstruction-based methods rely on old-fashioned convolutional autoencoders.
We propose a new autoencoder model for enhanced consecutive frame reconstruction.
arXiv Detail & Related papers (2023-01-28T01:57:57Z) - Self-Supervised Masked Convolutional Transformer Block for Anomaly
Detection [122.4894940892536]
We present a novel self-supervised masked convolutional transformer block (SSMCTB) that comprises the reconstruction-based functionality at a core architectural level.
In this work, we extend our previous self-supervised predictive convolutional attentive block (SSPCAB) with a 3D masked convolutional layer, a transformer for channel-wise attention, as well as a novel self-supervised objective based on Huber loss.
arXiv Detail & Related papers (2022-09-25T04:56:10Z) - Clear Memory-Augmented Auto-Encoder for Surface Defect Detection [10.829080460965478]
We propose a clear memory-augmented auto-encoder to repair abnormal foregrounds and preserve clear backgrounds.
A general artificial anomaly generation algorithm is proposed to simulate anomalies that are as realistic and feature-rich as possible.
At last, we propose a novel multi scale feature residual detection method for defect segmentation.
arXiv Detail & Related papers (2022-08-08T02:39:03Z) - A Feature Memory Rearrangement Network for Visual Inspection of Textured
Surface Defects Toward Edge Intelligent Manufacturing [4.33060257697635]
We propose an unsupervised feature memory rearrangement network (FMR-Net) to accurately detect various textural defects simultaneously.
We use artificial synthetic defects to enable the model to recognize anomalies, while traditional wisdom relies only on defect-free samples.
FMR-Net exhibits state-of-the-art inspection accuracy and shows great potential for use in edge-computing-enabled smart industries.
arXiv Detail & Related papers (2022-06-22T04:05:13Z) - Self-Supervised Predictive Convolutional Attentive Block for Anomaly
Detection [97.93062818228015]
We propose to integrate the reconstruction-based functionality into a novel self-supervised predictive architectural building block.
Our block is equipped with a loss that minimizes the reconstruction error with respect to the masked area in the receptive field.
We demonstrate the generality of our block by integrating it into several state-of-the-art frameworks for anomaly detection on image and video.
arXiv Detail & Related papers (2021-11-17T13:30:31Z) - Unsupervised Anomaly Detection with Adversarial Mirrored AutoEncoders [51.691585766702744]
We propose a variant of Adversarial Autoencoder which uses a mirrored Wasserstein loss in the discriminator to enforce better semantic-level reconstruction.
We put forward an alternative measure of anomaly score to replace the reconstruction-based metric.
Our method outperforms the current state-of-the-art methods for anomaly detection on several OOD detection benchmarks.
arXiv Detail & Related papers (2020-03-24T08:26:58Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.