HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Explainable Deep One-Class Classification

Philipp Liznerski Lukas Ruff Robert A. Vandermeulen Billy Joe Franks Marius Kloft Klaus-Robert Müller

Explainable Deep One-Class Classification

Abstract

Deep one-class classification variants for anomaly detection learn a mapping that concentrates nominal samples in feature space causing anomalies to be mapped away. Because this transformation is highly non-linear, finding interpretations poses a significant challenge. In this paper we present an explainable deep one-class classification method, Fully Convolutional Data Description (FCDD), where the mapped samples are themselves also an explanation heatmap. FCDD yields competitive detection performance and provides reasonable explanations on common anomaly detection benchmarks with CIFAR-10 and ImageNet. On MVTec-AD, a recent manufacturing dataset offering ground-truth anomaly maps, FCDD sets a new state of the art in the unsupervised setting. Our method can incorporate ground-truth anomaly maps during training and using even a few of these (~5) improves performance significantly. Finally, using FCDD's explanations we demonstrate the vulnerability of deep one-class classification models to spurious image features such as image watermarks.

Code Repositories

liznerski/fcdd
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
anomaly-detection-on-mvtec-adFCDD (unsupervised)
Segmentation AUROC: 88
anomaly-detection-on-mvtec-adFCDD (semi-supervised)
Segmentation AUROC: 94
anomaly-detection-on-one-class-cifar-10FCDD
AUROC: 92
anomaly-detection-on-one-class-imagenet-30FCDD
AUROC: 91

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