DRÆM – A discriminatively trained reconstruction embedding for surface anomaly detection

By Vitjan Zavrtanik, Matej Kristan, Danijel Skocaj. IEEE International Conference on Computer Vision, 2021

1,151 citations recorded by the source corpus.

Visual surface anomaly detection aims to detect local image regions that significantly deviate from normal appearance. Recent surface anomaly detection methods rely on generative models to accurately reconstruct the normal areas and to fail on anomalies. These methods are trained only on anomaly-free images, and often require hand-crafted post-processing steps to localize the anomalies, which prohibits optimizing the feature extraction for maximal detection capability. In addition to reconstructive approach, we cast surface anomaly detection primarily as a discriminative problem and propose a discriminatively trained reconstruction anomaly embedding model (DRÆM). The proposed method learns a joint representation of an anomalous image and its anomaly-free reconstruction, while simultaneously learning a decision boundary between normal and anomalous examples. The method enables direct anomaly localization without the need for additional complicated post-processing of the network output and can be trained using simple and general anomaly simulations. On the challenging MVTec anomaly detection dataset, DRÆM outperforms the current state-of-the-art unsupervised methods by a large margin and even de-livers detection performance close to the fully-supervised methods on the widely used DAGM surface-defect detection dataset, while substantially outperforming them in localization accuracy. Code at github.com/VitjanZ/DRAEM.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Industrial and Manufacturing Engineering

View Paper: https://www.semanticscholar.org/paper/95a26eafabf06b1fc5dec6c460a927cf5964e97e

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Ziwei Liu
  2. Fahad Shahbaz Khan
  3. Yixuan Li
  4. Kaiyang Zhou
  5. Jingkang Yang
  6. Wayne Zhang
  7. Bo Li
  8. Haoqi Wang

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Ming-Hsuan Yang
  3. Chen Change Loy
  4. Dacheng Tao
  5. Xiaogang Wang
  6. Ziwei Liu
  7. Trevor Darrell
  8. Fahad Shahbaz Khan

In Industrial and Manufacturing Engineering

  1. Bo Li
  2. Mubarak Shah
  3. Yaochu Jin
  4. Dan Hendrycks

Citations in the corpus