arXiv:2603.18093cs.CV2026-03中稿 · CVPR被引 2

无需训练,用一张异常图生成逼真多异常样本,提升工业检测效果。

One-to-More: High-Fidelity Training-Free Anomaly Generation with Attention Control

  • 基于单张异常图的自注意力机制,控制扩散过程生成多异常样本。
  • 在多个数据集上优于现有方法,合成异常与真实分布更接近。
  • 适合缺乏异常数据的工业场景,尤其适用于快速部署检测系统。

工业异常检测面临正常图像丰富而异常样本稀缺的问题。尽管已有大量少样本异常生成方法用于扩充异常数据,但多数方法需耗时训练,且难以学习真实异常分布,限制了下游检测模型性能。为此,本文提出无需训练的少样本异常生成方法O2MAG,利用单张异常图像的自注意力机制,合成更真实的多异常样本,支持高效的下游异常检测。具体地,O2MAG通过自注意力嫁接操控三个并行扩散过程,并引入异常掩码以缓解前景-背景查询混淆,生成符合文本提示的异常。为弥合编码文本提示与真实异常语义间的差距,进一步引入异常引导优化,使生成过程更贴近目标异常分布。此外,为避免掩码区域内异常过弱,采用双注意力增强机制强化掩码区域的自注意力与跨注意力。大量实验验证了O2MAG的有效性,在多个下游异常检测任务中表现优于当前最优方法。

原文摘要 · Abstract (English)

Industrial anomaly detection (AD) is characterized by an abundance of normal images but a scarcity of anomalous ones. Although numerous few-shot anomaly synthesis methods have been proposed to augment anomalous data for downstream AD tasks, most existing approaches require time-consuming training and struggle to learn distributions that are faithful to real anomalies, thereby restricting the efficacy of AD models trained on such data. To address these limitations, we propose a training-free few-shot anomaly generation method, namely O2MAG, which leverages the self-attention in One reference anomalous image to synthesize More realistic anomalies, supporting effective downstream anomaly detection. Specifically, O2MAG manipulates three parallel diffusion processes via self-attention grafting and incorporates the anomaly mask to mitigate foreground-background query confusion, synthesizing text-guided anomalies that closely adhere to real anomalous distributions. To bridge the semantic gap between the encoded anomaly text prompts and the true anomaly semantics, Anomaly-Guided Optimization is further introduced to align the synthesis process with the target anomalous distribution, steering the generation toward realistic and text-consistent anomalies. Moreover, to mitigate faint anomaly synthesis inside anomaly masks, Dual-Attention Enhancement is adopted during generation to reinforce both self- and cross-attention on masked regions. Extensive experiments validate the effectiveness of O2MAG, demonstrating its superior performance over prior state-of-the-art methods on downstream AD tasks.

异常生成扩散模型少样本

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