arXiv:2508.04818cs.CVeess.IV2025-08ICML被引 1

无需重建图像,用扩散模型实时检测异常并生成定位图。

Single-Step Reconstruction-Free Anomaly Detection and Segmentation via Diffusion Models

  • 直接从扩散模型输出异常图,跳过耗时的重建步骤。
  • 在MVTec-AD数据集上F1分数提升7%,3D打印材料数据集提升13%。
  • 适合对速度和精度要求高的工业缺陷检测场景。

生成模型在过去十年中在异常检测与分割任务上表现优异。近期,扩散模型作为强大替代方案,超越了GAN和VAE等先前方法。典型扩散模型异常检测需在正常数据上训练,推理时将异常图像扰动至前向扩散过程中的预设中间步,再通过迭代反向采样重建对应正常图像。但基于重建的方法存在三大挑战:(1) 重建过程计算开销大,难以满足实时性需求;(2) 对复杂或细微模式,重建结果可能对应另一正常样本而非原始输入;(3) 中间噪声水平的选择困难,因依赖应用特性且常需先验知识,这在无监督设置下不成立。本文提出重建无关的实时异常检测方法RADAR,克服上述局限。不同于现有最优方法需重建输入图像,RADAR直接由扩散模型生成异常图,显著提升检测准确率与计算效率。我们在真实世界3D打印材料和MVTec-AD数据集上评估该方法,结果表明其在所有关键指标(准确率、精确率、召回率、F1)上均优于当前最先进的扩散模型及统计机器学习模型。具体而言,在MVTec-AD上F1分数较次优模型提升7%,在3D打印材料数据集上提升13%。代码已开源:https://github.com/mehrdadmoradi124/RADAR。

原文摘要 · Abstract (English)

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

异常检测扩散模型实时性缺陷分割

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。