Command Palette
Search for a command to run...
Hanqiu Deng Xingyu Li

Abstract
Knowledge distillation (KD) achieves promising results on the challenging problem of unsupervised anomaly detection (AD).The representation discrepancy of anomalies in the teacher-student (T-S) model provides essential evidence for AD. However, using similar or identical architectures to build the teacher and student models in previous studies hinders the diversity of anomalous representations. To tackle this problem, we propose a novel T-S model consisting of a teacher encoder and a student decoder and introduce a simple yet effective "reverse distillation" paradigm accordingly. Instead of receiving raw images directly, the student network takes teacher model's one-class embedding as input and targets to restore the teacher's multiscale representations. Inherently, knowledge distillation in this study starts from abstract, high-level presentations to low-level features. In addition, we introduce a trainable one-class bottleneck embedding (OCBE) module in our T-S model. The obtained compact embedding effectively preserves essential information on normal patterns, but abandons anomaly perturbations. Extensive experimentation on AD and one-class novelty detection benchmarks shows that our method surpasses SOTA performance, demonstrating our proposed approach's effectiveness and generalizability.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| anomaly-classification-on-goodsad | RD4AD | AUPR: 68.2 AUROC: 66.5 |
| anomaly-detection-on-aebad-s | ReverseDistillation | Detection AUROC: 81.0 Segmentation AUPRO: 85.6 |
| anomaly-detection-on-aebad-v | ReverseDistillation | Detection AUROC: 71.0 |
| anomaly-detection-on-fashion-mnist | Reverse Distillation | ROC AUC: 95.0 |
| anomaly-detection-on-mvtec-ad | Reverse Distillation | Detection AUROC: 98.5 Segmentation AUPRO: 93.9 Segmentation AUROC: 97.8 |
| anomaly-detection-on-mvtec-loco-ad | RD4AD | Avg. Detection AUROC: 78.7 Detection AUROC (only logical): 69.4 Detection AUROC (only structural): 88.0 Segmentation AU-sPRO (until FPR 5%): 63.7 |
| anomaly-detection-on-one-class-cifar-10 | Reverse Distillation | AUROC: 86.5 |
| anomaly-detection-on-visa | Reverse Distillation | Segmentation AUPRO (until 30% FPR): 70.9 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.