Single-Step Reconstruction-Free Anomaly Detection and Segmentation via Diffusion Models
2508.04818v1
cs.CV, eess.IV, stat.ML, 62H35, 68T07, 62M40, 68T45, I.2.6; I.2.10; I.4.6; I.4.8; I.5.1; I.5.4
2025-08-09
Авторы:
Mehrdad Moradi, Marco Grasso, Bianca Maria Colosimo, Kamran Paynabar
Резюме на русском
Аномалийдиктовка и сегментация — важные задачи в области анализа изображений, требующие высокой точности и быстроты. Несмотря на успех генеративных моделей, в том числе diffusion models, традиционные подходы, основанные на восстановлении изображений, сталкиваются с тремя основными проблемами: высокой вычислительной сложностью, возможностью ошибочной интерпретации нормальных шаблонов и необходимостью дополнительной информации о нарушениях.
Мы предлагаем Reconstruction-Free Anomaly Detection with Attention-Based Diffusion Models in Real-Time (RADAR) — метод, который устраняет эти ограничения. RADAR не восстанавливает изображение, а напрямую производит anomaly maps, повышая тем самым точность и эффективность. Мы проверили RADAR на двух реальных датасетах — MVTec-AD и 3D-printed material. Наш подход показал лучшие результаты по всем ключевым метрикам в сравнении с современными diffusion-based и статистическими моделями. Этот результат подтверждает высокую эффективность RADAR в реальном времени.
Abstract
Generative models have demonstrated significant success in anomaly detection
and segmentation over the past decade. Recently, diffusion models have emerged
as a powerful alternative, outperforming previous approaches such as GANs and
VAEs. In typical diffusion-based anomaly detection, a model is trained on
normal data, and during inference, anomalous images are perturbed to a
predefined intermediate step in the forward diffusion process. The
corresponding normal image is then reconstructed through iterative reverse
sampling.
However, reconstruction-based approaches present three major challenges: (1)
the reconstruction process is computationally expensive due to multiple
sampling steps, making real-time applications impractical; (2) for complex or
subtle patterns, the reconstructed image may correspond to a different normal
pattern rather than the original input; and (3) Choosing an appropriate
intermediate noise level is challenging because it is application-dependent and
often assumes prior knowledge of anomalies, an assumption that does not hold in
unsupervised settings.
We introduce Reconstruction-free Anomaly Detection with Attention-based
diffusion models in Real-time (RADAR), which overcomes the limitations of
reconstruction-based anomaly detection. Unlike current SOTA methods that
reconstruct the input image, RADAR directly produces anomaly maps from the
diffusion model, improving both detection accuracy and computational
efficiency. We evaluate RADAR on real-world 3D-printed material and the
MVTec-AD dataset. Our approach surpasses state-of-the-art diffusion-based and
statistical machine learning models across all key metrics, including accuracy,
precision, recall, and F1 score. Specifically, RADAR improves F1 score by 7% on
MVTec-AD and 13% on the 3D-printed material dataset compared to the next best
model.
Code available at: https://github.com/mehrdadmoradi124/RADAR