A Feature Shuffling and Restoration Strategy for Universal Unsupervised Anomaly Detection
About
Unsupervised anomaly detection is vital in industrial fields, with reconstruction-based methods favored for their simplicity and effectiveness. However, reconstruction methods often encounter an identical shortcut issue, where both normal and anomalous regions can be well reconstructed and fail to identify outliers. The severity of this problem increases with the complexity of the normal data distribution. Consequently, existing methods may exhibit excellent detection performance in a specific scenario, but their performance sharply declines when transferred to another scenario. This paper focuses on establishing a universal model applicable to anomaly detection tasks across different settings, termed as universal anomaly detection. In this work, we introduce a novel, straightforward yet efficient framework for universal anomaly detection: \uline{F}eature \uline{S}huffling and \uline{R}estoration (FSR), which can alleviate the identical shortcut issue across different settings. First and foremost, FSR employs multi-scale features with rich semantic information as reconstruction targets, rather than raw image pixels. Subsequently, these multi-scale features are partitioned into non-overlapping feature blocks, which are randomly shuffled and then restored to their original state using a restoration network. This simple paradigm encourages the model to focus more on global contextual information. Additionally, we introduce a novel concept, the shuffling rate, to regulate the complexity of the FSR task, thereby alleviating the identical shortcut across different settings. Furthermore, we provide theoretical explanations for the effectiveness of FSR framework from two perspectives: network structure and mutual information. Extensive experimental results validate the superiority and efficiency of the FSR framework across different settings.Code is available at https://github.com/luow23/FSR.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Anomaly Detection | MVTec-AD (test) | P-AUROC98.4 | 152 | |
| Anomaly Detection | MVTec-AD Unified setting | Image AUROC98.3 | 18 | |
| Anomaly Detection | MVTec AD few-shot 1 | AUROC (Image-level)93.5 | 18 | |
| Unified Anomaly Detection | BTAD | ROC AUC (Pixel)97.3 | 18 | |
| Anomaly Localization | MVTec AD few-shot 1 | Pixel-level AUROC97.8 | 12 | |
| Anomaly Detection | MVTec-AD Separate setting | Image AUROC99.2 | 10 | |
| Anomaly Detection | MVTec AD Few-shot-8 setting | Image AUROC93.5 | 10 | |
| Anomaly Detection | BTAD Separate setting | Image AUROC95.9 | 10 | |
| Anomaly Detection | BTAD Few-shot-8 setting | Image AUROC93.2 | 10 |