TailedCore: Few-Shot Sampling for Unsupervised Long-Tail Noisy Anomaly Detection
About
We aim to solve unsupervised anomaly detection in a practical challenging environment where the normal dataset is both contaminated with defective regions and its product class distribution is tailed but unknown. We observe that existing models suffer from tail-versus-noise trade-off where if a model is robust against pixel noise, then its performance deteriorates on tail class samples, and vice versa. To mitigate the issue, we handle the tail class and noise samples independently. To this end, we propose TailSampler, a novel class size predictor that estimates the class cardinality of samples based on a symmetric assumption on the class-wise distribution of embedding similarities. TailSampler can be utilized to sample the tail class samples exclusively, allowing to handle them separately. Based on these facets, we build a memory-based anomaly detection model TailedCore, whose memory both well captures tail class information and is noise-robust. We extensively validate the effectiveness of TailedCore on the unsupervised long-tail noisy anomaly detection setting, and show that TailedCore outperforms the state-of-the-art in most settings.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Anomaly Segmentation | VisA (test) | -- | 51 | |
| Anomaly Classification | MVTec 4 (test) | Average Image AUROC95.71 | 16 | |
| Anomaly Classification | MVTecAD Pareto distribution | Ct AUROC (%)96.55 | 14 | |
| Anomaly Classification | VisA Pareto distribution (test) | AUROC (Ct)87.55 | 7 | |
| Anomaly Classification | VisA step (K=4) distribution (test) | AUROC (Ct)85.16 | 7 | |
| Anomaly Segmentation | MVTecAD Pareto tail distribution shape 0.6 (test) | AUROC (Ct)96.08 | 7 | |
| Anomaly Segmentation | MVTecAD step tail distribution K=4 (test) | Pixel AU-ROC (Ct)95.56 | 7 | |
| Anomaly Segmentation | MVTecAD step tail distribution K=1 (test) | AUROC (Ct)94.19 | 7 | |
| Anomaly Classification | VisA step (K=1) distribution (test) | AUROC (Ct)82.97 | 7 | |
| Image Anomaly Detection | MVTec-LOCO Pareto distribution | Ct AUROC (all)63 | 3 |