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

用于异常检测的扩散模型中的噪声动态添加

Dynamic Addition of Noise in a Diffusion Model for Anomaly Detection

Otto von Guericke University Magdeburg(马格德堡奥托·冯·格里克大学);Endress + Hauser(恩德斯豪斯公司)

Justin Tebbe, Jawad Tayyub

AI总结:

针对扩散模型难定位不同尺度(尤其大尺度)异常的问题,提出从动态加噪步长、无噪缩放输入去噪、潜在空间投影三方面扩展隐式条件方法,加微调机制,在多数据集上实现优异性能,推动扩散式异常检测进展。

摘要:

扩散模型通过捕获标称数据分布并借助重建识别异常,已在异常检测领域展现出重要应用价值。尽管具备诸多优势,但这类模型难以对不同尺度的异常进行定位,尤其是像整个组件缺失这类较大的异常。针对这一问题,我们提出了一种可增强扩散模型性能的新型框架,该框架从三个重要方面对Meng等人(2022)提出的隐式条件方法进行了扩展。首先,我们引入了动态步长计算机制,可在初始异常预测的引导下,让前向过程中的加噪步数可变。其次,我们证明了仅对缩放后的输入(不添加任何噪声直接进行去噪,效果优于传统去噪过程。第三,我们将图像投影到潜在空间中,以剔除会干扰大型缺失组件重建的精细细节。此外,我们还提出了一种微调机制,帮助模型有效掌握目标域的细微差异。我们的方法在知名异常检测数据集VisA、BTAD和MVTec上经过了严格评估,取得了优异的性能。值得注意的是,我们的框架无论异常尺度如何都能有效定位,标志着基于扩散的异常检测领域的一项关键进展。

英文摘要:

Diffusion models have found valuable applications in anomaly detection by capturing the nominal data distribution and identifying anomalies via reconstruction. Despite their merits, they struggle to localize anomalies of varying scales, especially larger anomalies such as entire missing components. Addressing this, we present a novel framework that enhances the capability of diffusion models, by extending the previous introduced implicit conditioning approach Meng et al. (2022) in three significant ways. First, we incorporate a dynamic step size computation that allows for variable noising steps in the forward process guided by an initial anomaly prediction. Second, we demonstrate that denoising an only scaled input, without any added noise, outperforms conventional denoising process. Third, we project images in a latent space to abstract away from fine details that interfere with reconstruction of large missing components. Additionally, we propose a fine-tuning mechanism that facilitates the model to effectively grasp the nuances of the target domain. Our method undergoes rigorous evaluation on prominent anomaly detection datasets VisA, BTAD and MVTec yielding strong performance. Importantly, our framework effectively localizes anomalies regardless of their scale, marking a pivotal advancement in diffusion-based anomaly detection.

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

↑