统一生成与检测,提升工业图像异常识别精度
HDM: Hybrid Diffusion Model for Unified Image Anomaly Detection
- 将生成与判别融合于统一扩散框架,实现协同优化
- 在多个工业数据集上,图像级与像素级检测的AUROC显著提升
- 适合需要高精度异常检测的工业质检和医疗影像场景
图像异常检测在工业质量检验和医学成像等应用中至关重要,直接影响产品品质与系统可靠性。然而,现有方法常难以应对复杂多样的异常模式,尤其因生成与判别任务分离,导致异常样本生成与区域检测无法有效协同。为此,我们提出一种新型混合扩散模型(HDM),将生成与判别整合为统一框架。模型包含三个核心模块:扩散异常生成模块(DAGM)、扩散判别模块(DDM)和概率优化模块(POM)。DAGM生成逼真且多样化的异常样本,增强代表性;DDM通过逆扩散过程捕捉生成样本与正常样本的差异,基于概率分布实现精准异常区域检测与定位;POM在生成与判别阶段优化概率分布,确保训练使用高质量样本。在多个工业图像数据集上的大量实验表明,本方法优于当前最优技术,在图像级与像素级异常检测性能上均显著提升,以AUROC为指标衡量。
原文摘要 · Abstract (English)
Image anomaly detection plays a vital role in applications such as industrial quality inspection and medical imaging, where it directly contributes to improving product quality and system reliability. However, existing methods often struggle with complex and diverse anomaly patterns. In particular, the separation between generation and discrimination tasks limits the effective coordination between anomaly sample generation and anomaly region detection. To address these challenges, we propose a novel hybrid diffusion model (HDM) that integrates generation and discrimination into a unified framework. The model consists of three key modules: the Diffusion Anomaly Generation Module (DAGM), the Diffusion Discriminative Module (DDM), and the Probability Optimization Module (POM). DAGM generates realistic and diverse anomaly samples, improving their representativeness. DDM then applies a reverse diffusion process to capture the differences between generated and normal samples, enabling precise anomaly region detection and localization based on probability distributions. POM refines the probability distributions during both the generation and discrimination phases, ensuring high-quality samples are used for training. Extensive experiments on multiple industrial image datasets demonstrate that our method outperforms state-of-the-art approaches, significantly improving both image-level and pixel-level anomaly detection performance, as measured by AUROC.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。