HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Remembering Normality: Memory-guided Knowledge Distillation for Unsupervised Anomaly Detection

{Lizhuang Ma Guannan Jiang Annan Shu Chengjie Wang Yabiao Wang Jiangning Zhang Ran Yi Xu Chen Liang Liu Zhihao Gu}

Remembering Normality: Memory-guided Knowledge Distillation for Unsupervised Anomaly Detection

Abstract

Knowledge distillation (KD) has been widely explored in unsupervised anomaly detection (AD). The student is assumed to constantly produce representations of typical patterns within trained data, named "normality", and the representation discrepancy between the teacher and student model is identified as anomalies. However, it suffers from the "normality forgetting" issue. Trained on anomaly-free data, the student still well reconstructs anomalous representations for anomalies and is sensitive to fine patterns in normal data, which also appear in training. To mitigate this issue, we introduce a novel Memory-guided Knowledge-Distillation (MemKD) framework that adaptively modulates the normality of student features in detecting anomalies. Specifically, we first propose a normality recall memory (NR Memory) to strengthen the normality of student-generated features by recalling the stored normal information. In this sense, representations will not present anomalies and fine patterns will be well described. Subsequently, we employ a normality embedding learning strategy to promote information learning for the NR Memory. It constructs a normal exemplar set so that the NR Memory can memorize prior knowledge in anomaly-free data and later recall them from the query feature. Consequently, comprehensive experiments demonstrate that the proposed MemKD achieves promising results on five benchmarks, i.e., MVTec AD, VisA, MPDD, MVTec 3D-AD, and Eyecandies.

Benchmarks

BenchmarkMethodologyMetrics
anomaly-detection-on-mvtec-adRememberingNormality
Detection AUROC: 99.6
Segmentation AUROC: 98.2

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