arXiv:2503.19357cs.CV2025-03CVPR被引 21

提出新扩散模型,精准定位异常区域且不破坏正常部分

Correcting Deviations from Normality: A Reformulated Diffusion Model for Multi-Class Unsupervised Anomaly Detection

  • 将异常建模为潜在空间的噪声,仅对异常区域进行修正
  • 在多个数据集上像素级AUPRC提升11-14%优于现有方法
  • 适合需要精确定位异常的工业质检等场景

扩散模型在基于重构的无监督异常检测中展现出潜力,但在多类别场景下难以保持结构完整性并恢复异常区域的正常内容。传统扩散模型从纯噪声生成图像,难以选择性修改异常区域而保留正常区域,导致正常区域重建质量下降。本文提出一种重构版扩散模型——偏差纠正扩散(DeCo-Diff),将异常视为潜在空间中的噪声,仅对异常区域进行变换,保护正常区域。该选择性修复策略显著提升重建质量,实现有效且精确的异常检测与定位。大量实验表明,在多个知名异常检测数据集上,该方法在像素级AUPRC上相较最先进模型提升11-14%。代码已公开于https://github.com/farzad-bz/DeCo-Diff。

原文摘要 · Abstract (English)

Recent advances in diffusion models have spurred research into their application for Reconstruction-based unsupervised anomaly detection. However, these methods may struggle with maintaining structural integrity and recovering the anomaly-free content of abnormal regions, especially in multi-class scenarios. Furthermore, diffusion models are inherently designed to generate images from pure noise and struggle to selectively alter anomalous regions of an image while preserving normal ones. This leads to potential degradation of normal regions during reconstruction, hampering the effectiveness of anomaly detection. This paper introduces a reformulation of the standard diffusion model geared toward selective region alteration, allowing the accurate identification of anomalies. By modeling anomalies as noise in the latent space, our proposed Deviation correction diffusion (DeCo-Diff) model preserves the normal regions and encourages transformations exclusively on anomalous areas. This selective approach enhances the reconstruction quality, facilitating effective unsupervised detection and localization of anomaly regions. Comprehensive evaluations demonstrate the superiority of our method in accurately identifying and localizing anomalies in complex images, with pixel-level AUPRC improvements of 11-14% over state-of-the-art models on well known anomaly detection datasets. The code is available at https://github.com/farzad-bz/DeCo-Diff

异常检测扩散模型图像修复

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