Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

PANDA: Adapting Pretrained Features for Anomaly Detection and Segmentation

About

Anomaly detection methods require high-quality features. In recent years, the anomaly detection community has attempted to obtain better features using advances in deep self-supervised feature learning. Surprisingly, a very promising direction, using pretrained deep features, has been mostly overlooked. In this paper, we first empirically establish the perhaps expected, but unreported result, that combining pretrained features with simple anomaly detection and segmentation methods convincingly outperforms, much more complex, state-of-the-art methods. In order to obtain further performance gains in anomaly detection, we adapt pretrained features to the target distribution. Although transfer learning methods are well established in multi-class classification problems, the one-class classification (OCC) setting is not as well explored. It turns out that naive adaptation methods, which typically work well in supervised learning, often result in catastrophic collapse (feature deterioration) and reduce performance in OCC settings. A popular OCC method, DeepSVDD, advocates using specialized architectures, but this limits the adaptation performance gain. We propose two methods for combating collapse: i) a variant of early stopping that dynamically learns the stopping iteration ii) elastic regularization inspired by continual learning. Our method, PANDA, outperforms the state-of-the-art in the OCC, outlier exposure and anomaly segmentation settings by large margins.

Tal Reiss, Niv Cohen, Liron Bergman, Yedid Hoshen• 2020

Related benchmarks

TaskDatasetResultRank
Anomaly LocalizationMVTec AD
Pixel AUROC96.5
513
Anomaly DetectionCIFAR-10
AUC98.9
130
Anomaly DetectionWBC
ROCAUC0.874
104
Anomaly DetectionMVTec AD
Overall AUROC96.2
83
Anomaly DetectionMVTec
AUROC86.5
79
Anomaly DetectionCIFAR-100
AUROC97.3
72
Anomaly DetectionMVTec AD 1.0 (test)
Image AUROC85.5
67
Anomaly DetectionSDD
AUC0.988
57
Anomaly DetectionCheXpert (test)
AUROC0.686
49
Anomaly LocalizationMVTec AD 1.0 (test)--
47
Showing 10 of 62 rows

Other info

Code

Follow for update