A Unified Model for Multi-class Anomaly Detection
Despite the rapid advance of unsupervised anomaly detection, existing methods require to train separate models for different objects. In this work, we present UniAD that accomplishes anomaly detection for multiple classes with a unified framework. Under such a challenging setting, popular reconstruction networks may fall into an "identical shortcut", where both normal and anomalous samples can be well recovered, and hence fail to spot outliers. To tackle this obstacle, we make three improvements. First, we revisit the formulations of fully-connected layer, convolutional layer, as well as attention layer, and confirm the important role of query embedding (i.e., within attention layer) in preventing the network from learning the shortcut. We therefore come up with a layer-wise query decoder to help model the multi-class distribution. Second, we employ a neighbor masked attention module to further avoid the information leak from the input feature to the reconstructed output feature. Third, we propose a feature jittering strategy that urges the model to recover the correct message even with noisy inputs. We evaluate our algorithm on MVTec-AD and CIFAR-10 datasets, where we surpass the state-of-the-art alternatives by a sufficiently large margin. For example, when learning a unified model for 15 categories in MVTec-AD, we surpass the second competitor on the tasks of both anomaly detection (from 88.1% to 96.5%) and anomaly localization (from 89.5% to 96.8%). Code is available at https://github.com/zhiyuanyou/UniAD.
Code (1)
Tasks
Anomaly DetectionAnomaly LocalizationmodelMulti-class Anomaly DetectionUnsupervised Anomaly DetectionSimilar Papers 제목 키워드 기반
Absolute-Unified Multi-Class Anomaly Detection via Class-Agnostic Distribution Alignment
Conventional unsupervised anomaly detection (UAD) methods build separate models for each object category. Recent studies have proposed to train a unified model for multiple classes, namely model-unified UAD. However, suc…
Anomaly DetectionMulti-class Anomaly DetectionUnsupervised Anomaly DetectionToward Multi-class Anomaly Detection: Exploring Class-aware Unified Model against Inter-class Interference
In the context of high usability in single-class anomaly detection models, recent academic research has become concerned about the more complex multi-class anomaly detection. Although several papers have designed unified…
Anomaly DetectionDecoderMulti-class Anomaly DetectionText-Guided Multimodal Unified Industrial Anomaly Detection
Industrial anomaly detection based on RGB-3D multimodal data has emerged as a mainstream paradigm for intelligent quality inspection. However, existing unsupervised methods suffer from two critical limitations: ambiguous…
Anomaly DetectionCenter-aware Residual Anomaly Synthesis for Multi-class Industrial Anomaly Detection
Anomaly detection plays a vital role in the inspection of industrial images. Most existing methods require separate models for each category, resulting in multiplied deployment costs. This highlights the challenge of dev…
Anomaly DetectionMulti-class Anomaly DetectionOmniAL: A Unified CNN Framework for Unsupervised Anomaly Localization
Unsupervised anomaly localization and detection is crucial for industrial manufacturing processes due to the lack of anomalous samples. Recent unsupervised advances on industrial anomaly detection achieve high perfor…
Anomaly DetectionAnomaly LocalizationMulti-class Anomaly Detection