HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Unlocking the Potential of Reverse Distillation for Anomaly Detection

Xinyue Liu; Jianyuan Wang; Biao Leng; Shuo Zhang

Unlocking the Potential of Reverse Distillation for Anomaly Detection

Abstract

Knowledge Distillation (KD) is a promising approach for unsupervised Anomaly Detection (AD). However, the student network's over-generalization often diminishes the crucial representation differences between teacher and student in anomalous regions, leading to detection failures. To addresses this problem, the widely accepted Reverse Distillation (RD) paradigm designs the asymmetry teacher and student, using an encoder as teacher and a decoder as student. Yet, the design of RD does not ensure that the teacher encoder effectively distinguishes between normal and abnormal features or that the student decoder generates anomaly-free features. Additionally, the absence of skip connections results in a loss of fine details during feature reconstruction. To address these issues, we propose RD with Expert, which introduces a novel Expert-Teacher-Student network for simultaneous distillation of both the teacher encoder and student decoder. The added expert network enhances the student's ability to generate normal features and optimizes the teacher's differentiation between normal and abnormal features, reducing missed detections. Additionally, Guided Information Injection is designed to filter and transfer features from teacher to student, improving detail reconstruction and minimizing false positives. Experiments on several benchmarks prove that our method outperforms existing unsupervised AD methods under RD paradigm, fully unlocking RD's potential.

Code Repositories

hito2448/urd
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
anomaly-detection-on-btadURD
Detection AUROC: 93.9
Segmentation AP: 65.2
Segmentation AUPRO: 78.5
Segmentation AUROC: 98.1
anomaly-detection-on-mvtec-adURD
Detection AUROC: 99.2
Segmentation AP: 72.4
Segmentation AUPRO: 96.3
Segmentation AUROC: 99.0
anomaly-detection-on-visaURD
Detection AUROC: 96.5
Segmentation AUPRO (until 30% FPR): 95.1
Segmentation AUROC: 99.1

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
Unlocking the Potential of Reverse Distillation for Anomaly Detection | Papers | HyperAI