HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Anomaly Detection with Conditioned Denoising Diffusion Models

Arian Mousakhan Thomas Brox Jawad Tayyub

Anomaly Detection with Conditioned Denoising Diffusion Models

Abstract

Traditional reconstruction-based methods have struggled to achieve competitive performance in anomaly detection. In this paper, we introduce Denoising Diffusion Anomaly Detection (DDAD), a novel denoising process for image reconstruction conditioned on a target image. This ensures a coherent restoration that closely resembles the target image. Our anomaly detection framework employs the conditioning mechanism, where the target image is set as the input image to guide the denoising process, leading to a defectless reconstruction while maintaining nominal patterns. Anomalies are then localised via a pixel-wise and feature-wise comparison of the input and reconstructed image. Finally, to enhance the effectiveness of the feature-wise comparison, we introduce a domain adaptation method that utilises nearly identical generated examples from our conditioned denoising process to fine-tune the pretrained feature extractor. The veracity of DDAD is demonstrated on various datasets including MVTec and VisA benchmarks, achieving state-of-the-art results of (99.8 \%) and (98.9 \%) image-level AUROC respectively.

Code Repositories

arimousa/DDAD
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
anomaly-detection-on-mvtec-adDDAD
Detection AUROC: 99.8
Segmentation AUROC: 98.1
anomaly-detection-on-visaDDAD
Detection AUROC: 98.9
Segmentation AUPRO (until 30% FPR): 92.7
Segmentation AUROC: 97.6

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
Anomaly Detection with Conditioned Denoising Diffusion Models | Papers | HyperAI