arXiv:2508.02903cs.CVeess.IV2025-08ICCV被引 4

用含异常数据训练扩散模型,实现无需纯净正常数据的异常分割。

RDDPM: Robust Denoising Diffusion Probabilistic Model for Unsupervised Anomaly Segmentation

  • 将扩散模型重构成鲁棒回归问题,从混杂数据中学习正常模式。
  • 在MVTec数据集上比现有方法高8.08% AUROC、10.37% AUPRC。
  • 适合无纯净正常数据的工业异常检测场景。

扩散模型在无监督异常分割中表现优异,通常需在纯正常数据上训练。然而,真实场景中难以获取纯净正常数据。本文提出鲁棒去噪扩散概率模型(RDDPM),仅使用包含正常与异常的混合未标注数据进行训练。通过将数据的最大似然估计重构为非线性回归问题,我们从回归视角重新诠释去噪扩散模型,并引入鲁棒回归构建鲁棒版本。该框架可灵活设计多种鲁棒扩散模型。实验表明,在仅使用污染数据的情况下,本方法显著优于当前最先进方法:在MVTec数据集上,AUROC提升达8.08%,AUPRC提升达10.37%。代码已开源。

原文摘要 · Abstract (English)

Recent advancements in diffusion models have demonstrated significant success in unsupervised anomaly segmentation. For anomaly segmentation, these models are first trained on normal data; then, an anomalous image is noised to an intermediate step, and the normal image is reconstructed through backward diffusion. Unlike traditional statistical methods, diffusion models do not rely on specific assumptions about the data or target anomalies, making them versatile for use across different domains. However, diffusion models typically assume access to normal data for training, limiting their applicability in realistic settings. In this paper, we propose novel robust denoising diffusion models for scenarios where only contaminated (i.e., a mix of normal and anomalous) unlabeled data is available. By casting maximum likelihood estimation of the data as a nonlinear regression problem, we reinterpret the denoising diffusion probabilistic model through a regression lens. Using robust regression, we derive a robust version of denoising diffusion probabilistic models. Our novel framework offers flexibility in constructing various robust diffusion models. Our experiments show that our approach outperforms current state of the art diffusion models, for unsupervised anomaly segmentation when only contaminated data is available. Our method outperforms existing diffusion-based approaches, achieving up to 8.08\% higher AUROC and 10.37\% higher AUPRC on MVTec datasets. The implementation code is available at: https://github.com/mehrdadmoradi124/RDDPM

异常分割扩散模型鲁棒学习

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