RDDPM: Robust Denoising Diffusion Probabilistic Model for Unsupervised Anomaly Segmentation
2508.02903v1
cs.CV, eess.IV, stat.ML, 68T07, I.4.9; I.2.10
2025-08-09
Авторы:
Mehrdad Moradi, Kamran Paynabar
Резюме на русском
Данная работа уделяет внимание проблеме непродолженного обнаружения аномалий в ситуациях, когда доступно только помеченное контролируемой нормальной и аномальной смеси данных. Авторы предлагают новую модель Robust Denoising Diffusion Probabilistic Model (RDDPM), призванную улучшить надежность диффузионных моделей в таких условиях. Она использует подход метода наименьших квадратов для реинтерпретации диффузионных моделей через регрессию. Результат — модель, устойчивая к выбросам и подходящая для контролируемой необученной аномалий сегментации. Эксперименты показали, что RDDPM превосходит современные диффузионные модели на датасете MVTec, достигая повышения метрик AUROC и AUPRC на 8,08% и 10,37% соответственно. Это демонстрирует мощь RDDPM в обнаружении аномалий при ограниченных ресурсах обучения.
Abstract
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