HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Anomaly Detection via Reverse Distillation from One-Class Embedding

Hanqiu Deng Xingyu Li

Anomaly Detection via Reverse Distillation from One-Class Embedding

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.

Benchmarks

BenchmarkMethodologyMetrics
anomaly-classification-on-goodsadRD4AD
AUPR: 68.2
AUROC: 66.5
anomaly-detection-on-aebad-sReverseDistillation
Detection AUROC: 81.0
Segmentation AUPRO: 85.6
anomaly-detection-on-aebad-vReverseDistillation
Detection AUROC: 71.0
anomaly-detection-on-fashion-mnistReverse Distillation
ROC AUC: 95.0
anomaly-detection-on-mvtec-adReverse Distillation
Detection AUROC: 98.5
Segmentation AUPRO: 93.9
Segmentation AUROC: 97.8
anomaly-detection-on-mvtec-loco-adRD4AD
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-10Reverse Distillation
AUROC: 86.5
anomaly-detection-on-visaReverse 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.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Anomaly Detection via Reverse Distillation from One-Class Embedding | Papers | HyperAI