HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Reconstruction by Inpainting for Visual Anomaly Detection

{Danijel Skočaj Matej Kristan Vitjan Zavrtanik}

Abstract

Visual anomaly detection addresses the problem of classification or localization of regions in an image that deviate from their normal appearance. A popular approach trains an auto-encoder on anomaly-free images and performs anomaly detection by calculating the difference between the input and the reconstructed image. This approach assumes that the auto-encoder will be unable to accurately reconstruct anomalous regions. But in practice neural networks generalize well even to anomalies and reconstruct them sufficiently well, thus reducing the detection capabilities. Accurate reconstruction is far less likely if the anomaly pixels were not visible to the auto-encoder. We thus cast anomaly detection as a self-supervised reconstruction-by-inpainting problem. Our approach (RIAD) randomly removes partial image regions and reconstructs the image from partial inpaintings, thus addressing the drawbacks of auto-enocoding methods. RIAD is extensively evaluated on several benchmarks and sets a new state-of-the art on a recent highly challenging anomaly detection benchmark.

Benchmarks

BenchmarkMethodologyMetrics
anomaly-detection-on-aebad-sRIAD
Detection AUROC: 40.0
Segmentation AUPRO: 58.2
anomaly-detection-on-aebad-vRIAD
Detection AUROC: 56.1
anomaly-detection-on-mvtec-adRIAD
Detection AUROC: 91.7
Segmentation AUROC: 94.2

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
Reconstruction by Inpainting for Visual Anomaly Detection | Papers | HyperAI