3 个月前

DiffusionAD:基于范数引导的一步去噪扩散异常检测

DiffusionAD:基于范数引导的一步去噪扩散异常检测

摘要

异常检测因其卓越的效率与效果,在实际工业制造领域得到了广泛应用。然而,以往基于生成模型的方法受限于重建质量欠佳,从而制约了整体性能的提升。为此,本文提出 DiffusionAD,一种新型的异常检测框架,包含重建子网络与分割子网络两个核心模块。其关键创新在于将基于扩散模型的重建过程重新建模为“噪声到正常”(noise-to-norm)的范式:在高斯噪声干扰下,异常区域的显著特征被破坏,随后被重建为无异常状态的图像。在此基础上,分割子网络通过比较输入图像与其无异常重建结果之间的相似性与差异性,生成像素级的异常得分。此外,考虑到扩散模型固有的迭代去噪机制导致推理速度显著下降,本文重新审视去噪流程,提出一种快速的一步去噪范式,可在保持与现有方法相当重建质量的前提下,实现数百倍的加速。为进一步应对异常表现形式的多样性,本文还引入一种基于归一化引导(norm-guided)的多噪声尺度融合机制,有效提升重建结果的保真度。在四个标准且具有挑战性的基准数据集上的全面评估表明,DiffusionAD 在检测性能上超越当前最先进的方法,同时保持了相近的推理速度,充分验证了所提框架的有效性与广泛适用性。相关代码已开源,地址为:https://github.com/HuiZhang0812/DiffusionAD

代码仓库

基准测试

基准方法指标
anomaly-detection-on-mpddDiffusionAD
Detection AUROC: 96.2
Segmentation AUPRO: 95.3
Segmentation AUROC: 98.5
anomaly-detection-on-visaDiffusionAD
Detection AUROC: 98.8
Segmentation AUPRO: 96.0
Segmentation AUPRO (until 30% FPR): 96.0
Segmentation AUROC: 98.9
unsupervised-anomaly-detection-on-dagm2007DiffusionAD
Detection AUROC: 99.6

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
DiffusionAD:基于范数引导的一步去噪扩散异常检测 | 论文 | HyperAI超神经