通过分层扩散与显式掩码对齐,生成更精细的工业异常图像。
STAGE: Segmentation-oriented Industrial Anomaly Synthesis via Graded Diffusion with Explicit Mask Alignment
- 利用背景先验指导去噪,精准定位异常区域。
- 分层扩散框架保留细微异常,避免被忽略。
- 适合需要高精度异常合成的工业质检场景。
面向异常分割的工业异常合成(SIAS)在提升下游异常分割性能方面至关重要,能有效扩充异常数据。然而现有方法存在两大关键局限:合成异常常缺乏复杂纹理细节且与背景对不齐;难以生成细粒度、像素级异常。为此,本文提出一种基于分层扩散与显式掩码对齐的异常合成方法——STAGE。STAGE引入新颖的异常推理策略,将干净背景信息作为先验引导去噪分布,使模型更有效地区分并突出异常前景。同时,采用分层扩散框架,设置仅含异常的分支,在正向与逆向过程中显式记录局部异常,确保微小异常不被遗漏。此外,通过显式掩码对齐(EMA)策略,逐步对齐合成异常与背景,实现上下文一致且结构连贯的生成。在MVTec和BTAD数据集上的大量实验表明,STAGE在SIAS任务上达到当前最优性能,进而显著提升下游异常分割效果。
原文摘要 · Abstract (English)
Segmentation-oriented Industrial Anomaly Synthesis (SIAS) plays a pivotal role in enhancing the performance of downstream anomaly segmentation, as it provides an effective means of expanding abnormal data. However, existing SIAS methods face several critical limitations: (i) the synthesized anomalies often lack intricate texture details and fail to align precisely with the surrounding background, and (ii) they struggle to generate fine-grained, pixel-level anomalies. To address these challenges, we propose Segmentation-oriented Anomaly synthesis via Graded diffusion with Explicit mask alignment, termed STAGE. STAGE introduces a novel anomaly inference strategy that incorporates clean background information as a prior to guide the denoising distribution, enabling the model to more effectively distinguish and highlight abnormal foregrounds. Furthermore, it employs a graded diffusion framework with an anomaly-only branch to explicitly record local anomalies during both the forward and reverse processes, ensuring that subtle anomalies are not overlooked. Finally, STAGE incorporates the explicit mask alignment (EMA) strategy to progressively align the synthesized anomalies with the background, resulting in context-consistent and structurally coherent generations. Extensive experiments on the MVTec and BTAD datasets demonstrate that STAGE achieves state-of-the-art performance in SIAS, which in turn enhances downstream anomaly segmentation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。