arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.08730cs.CV

DiffusionAD:用于异常检测的范数引导单步去噪扩散方法

DiffusionAD: Norm-guided One-step Denoising Diffusion for Anomaly Detection

Institute of Trustworthy Embodied AI, Fudan University(复旦大学可信具身人工智能研究院);School of Computer Science, Zhejiang University of Technology(浙江工业大学计算机学院);School of Communication & Information Engineering, Shanghai University(上海大学通信与信息工程学院)

Hui Zhang, Zheng Wang, Dan Zeng, Zuxuan Wu, Yu-Gang Jiang

AI总结:

针对生成式异常检测模型重建质量欠佳、扩散模型推理慢的问题,提出含重建与分割子网络的DiffusionAD,采用噪声到范数、单步去噪、范数引导范式,在4个基准上性能超SOTA且速度相当。

摘要:

异常检测凭借其出色的有效性和效率,已在实际工业制造中获得广泛应用。然而,以往基于生成式的模型受限于欠佳的重建质量,制约了其整体性能。我们提出了DiffusionAD,这是一种新型异常检测流水线,由重建子网络和分割子网络组成。一项核心改进在于我们利用扩散模型将重建过程重新构建为“噪声到范数”范式:异常区域在被高斯噪声干扰后会失去其独有特征,随后被重建为无异常区域。之后,分割子网络会基于输入图像与其无异常重建结果之间的相似性和差异,预测像素级异常分数。此外,鉴于扩散模型的迭代去噪特性会大幅降低推理速度,我们重新审视了去噪过程,并提出了快速单步去噪范式,该范式实现了数百倍的加速,同时保持了相当的重建质量。此外,考虑到异常表现形式的多样性,我们提出了范数引导范式,以整合多种噪声尺度的优势,提升重建的保真度。在四个标准且具挑战性的基准数据集上的综合评估表明,DiffusionAD优于当前的最优方法,且推理速度与之相当,证明了所提流水线的有效性和广泛适用性。代码已发布于https://github.com/HuiZhang0812/DiffusionAD

英文摘要:

Anomaly detection has garnered extensive applications in real industrial manufacturing due to its remarkable effectiveness and efficiency. However, previous generative-based models have been limited by suboptimal reconstruction quality, hampering their overall performance. We introduce DiffusionAD, a novel anomaly detection pipeline comprising a reconstruction sub-network and a segmentation sub-network. A fundamental enhancement lies in our reformulation of the reconstruction process using a diffusion model into a noise-to-norm paradigm. Here, the anomalous region loses its distinctive features after being disturbed by Gaussian noise and is subsequently reconstructed into an anomaly-free one. Afterward, the segmentation sub-network predicts pixel-level anomaly scores based on the similarities and discrepancies between the input image and its anomaly-free reconstruction. Additionally, given the substantial decrease in inference speed due to the iterative denoising nature of diffusion models, we revisit the denoising process and introduce a rapid one-step denoising paradigm. This paradigm achieves hundreds of times acceleration while preserving comparable reconstruction quality. Furthermore, considering the diversity in the manifestation of anomalies, we propose a norm-guided paradigm to integrate the benefits of multiple noise scales, enhancing the fidelity of reconstructions. Comprehensive evaluations on four standard and challenging benchmarks reveal that DiffusionAD outperforms current state-of-the-art approaches and achieves comparable inference speed, demonstrating the effectiveness and broad applicability of the proposed pipeline. Code is released at https://github.com/HuiZhang0812/DiffusionAD

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑