HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

PANDA: Adapting Pretrained Features for Anomaly Detection and Segmentation

Tal Reiss Niv Cohen Liron Bergman Yedid Hoshen

PANDA: Adapting Pretrained Features for Anomaly Detection and Segmentation

Abstract

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.

Code Repositories

talreiss/PANDA
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
anomaly-detection-on-cats-and-dogsSelf-Supervised DeepSVDD
ROC AUC: 50.5
anomaly-detection-on-cats-and-dogsSelf-Supervised One-class SVM, RBF kernel
ROC AUC: 51.7
anomaly-detection-on-cats-and-dogsPANDA
ROC AUC: 97.3
anomaly-detection-on-cats-and-dogsPANDA-OE
ROC AUC: 94.5
anomaly-detection-on-diorSelf-Supervised One-class SVM, RBF kernel
ROC AUC: 70.7
anomaly-detection-on-diorPANDA
ROC AUC: 94.3
anomaly-detection-on-diorPANDA-OE
ROC AUC: 95.9
anomaly-detection-on-diorSelf-Supervised DeepSVDD
ROC AUC: 70
anomaly-detection-on-fashion-mnistSelf-Supervised DeepSVDD
ROC AUC: 84.8
anomaly-detection-on-fashion-mnistSelf-Supervised One-class SVM, RBF kernel
ROC AUC: 92.8
anomaly-detection-on-fashion-mnistPANDA-OE
ROC AUC: 91.8
anomaly-detection-on-fashion-mnistPANDA
ROC AUC: 95.6
anomaly-detection-on-hyper-kvasir-datasetPANDA
AUC: 0.937
anomaly-detection-on-one-class-cifar-10PANDA
AUROC: 96.2
anomaly-detection-on-one-class-cifar-10Self-Supervised One-class SVM, RBF kernel
AUROC: 64.7
anomaly-detection-on-one-class-cifar-10Self-Supervised DeepSVDD
AUROC: 64.8
anomaly-detection-on-one-class-cifar-10PANDA-OE
AUROC: 98.9
anomaly-detection-on-one-class-cifar-100PANDA-OE
AUROC: 97.3
anomaly-detection-on-one-class-cifar-100Self-Supervised Multi-Head RotNet
AUROC: 80.1
anomaly-detection-on-one-class-cifar-100Self-Supervised One-class SVM, RBF kernel
AUROC: 62.6
anomaly-detection-on-one-class-cifar-100Self-Supervised DeepSVDD
AUROC: 67
anomaly-detection-on-one-class-cifar-100PANDA
AUROC: 94.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
PANDA: Adapting Pretrained Features for Anomaly Detection and Segmentation | Papers | HyperAI