Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Registration based Few-Shot Anomaly Detection

About

This paper considers few-shot anomaly detection (FSAD), a practical yet under-studied setting for anomaly detection (AD), where only a limited number of normal images are provided for each category at training. So far, existing FSAD studies follow the one-model-per-category learning paradigm used for standard AD, and the inter-category commonality has not been explored. Inspired by how humans detect anomalies, i.e., comparing an image in question to normal images, we here leverage registration, an image alignment task that is inherently generalizable across categories, as the proxy task, to train a category-agnostic anomaly detection model. During testing, the anomalies are identified by comparing the registered features of the test image and its corresponding support (normal) images. As far as we know, this is the first FSAD method that trains a single generalizable model and requires no re-training or parameter fine-tuning for new categories. Experimental results have shown that the proposed method outperforms the state-of-the-art FSAD methods by 3%-8% in AUC on the MVTec and MPDD benchmarks.

Chaoqin Huang, Haoyan Guan, Aofan Jiang, Ya Zhang, Michael Spratling, Yan-Feng Wang• 2022

Related benchmarks

TaskDatasetResultRank
Anomaly LocalizationMVTec AD
Pixel AUROC94.5
369
Anomaly DetectionVisA
AUROC58.9
199
Anomaly DetectionCIFAR-10
AUC69.5
120
Anomaly DetectionMNIST
AUC60.2
87
Anomaly SegmentationMVTec-AD (test)
AUROC (Pixel)96.9
85
Image-level Anomaly DetectionMVTec-AD (test)
Overall AUROC91.2
68
Anomaly DetectionShanghaiTech--
68
Anomaly DetectionMVTec
AUROC95.9
65
Anomaly DetectionMNIST (test)
AUC54.8
65
Anomaly LocalizationMPDD (test)
Pixel AUROC0.951
60
Showing 10 of 53 rows

Other info

Code

Follow for update