Contrastive Knowledge Distillation for Anomaly Detection in Multi-Illumination/Focus Display Images
Organizations: KAIST · Hanyang University · Samsung Display
Abstract
In this paper, we tackle automatic anomaly detection in multi-illumination and multi-focus display images. The minute defects on the display surface are hard to spot out in RGB images and by a model trained with only normal data. To address this, we propose a novel contrastive learning scheme for knowledge distillation-based anomaly detection. In our framework, Multiresolution Knowledge Distillation (MKD) is adopted as a baseline, which operates by measuring feature similarities between the teacher and student networks. Based on MKD, we propose a novel contrastive learning method, namely Multiresolution Contrastive Distillation (MCD), which does not require positive/negative pairs with an anchor but operates by pulling/pushing the distance between the teacher and student features. Furthermore, we propose the blending module that transforms and aggregate multi-channel information to the three-channel input layer of MCD. Our proposed method significantly outperforms competitive state-of-the-art methods in both AUROC and accuracy metrics on the collected Multi-illumination and Multi-focus display image dataset for Anomaly Detection (MMdAD).
Figures & tables
| Method | Input Channels | AUROC |
| One-Class SVM [ 17 ] | [0, 2, 6] | 0.68 |
| RD4AD [ 1 ] | [0, 2, 6] | 0.77 |
| DRA [ 2 ] | [0, 2, 6] | 0.79 |
| MKD [ 3 ] | [0, 2, 6] | 0.91 |
| [0, 3, 11] | 0.70 | |
| [1, 3 ,4] | 0.78 |
| Evaluation Metric | MKD | BM | CL | Dataset Split 1 | Dataset Split 2 |
|---|---|---|---|---|---|
| AUROC | ✓ | 0.91 | 0.91 | ||
| ✓ | ✓ | 0.97 | 0.95 | ||
| ✓ | ✓ | ✓ | 0.99 | 0.99 | |
| Accuracy | ✓ | 0.85 | 0.84 | ||
| ✓ | ✓ | 0.93 | 0.92 | ||
| ✓ | ✓ | ✓ | 0.98 | 0.98 |
Industrial Anomaly Detection
Industrial anomaly detection finds defective products and abnormal equipment behavior from visual, sensor, audio, and other production data.
See all 262 papers