DeCo:如何把一张图的缺陷,干净地“搬”到另一张产品上?

快速导读:提出DeCo框架,通过解耦参考产品中的异常结构并重耦合至目标产品,实现零样本工业异常图像生成。

工业异常检测的核心瓶颈之一在于真实异常样本极度稀缺,这直接限制了监督模型的性能上限。近年来,零样本异常生成方法试图通过合成异常图像来缓解这一问题,但现有方案始终在两个关键环节上捉襟见肘:要么生成的异常图案不真实,要么在迁移异常时把源产品的纹理也一并“带”了过去。

DeCo(Decoupling and Recoupling)框架针对这一困境,提出了一种全新的解决思路。它不再试图让模型隐式地学会分离异常与产品,而是设计了一套显式的解耦与重耦合机制:先从参考图像中提取出纯粹的异常结构,再将其无缝地融合到任意目标产品上。这种“先拆后装”的策略,使得生成的异常图像在真实性和可控性上都取得了显著突破。

英文题目:DeCo: Zero-Shot Industrial Anomaly Generation through Decoupling and Recoupling

论文出处:arXiv 每日论文精选 · arXiv:2608.07904

原始论文:PDF / 论文页面

这篇论文解决什么问题?

在工业视觉领域,缺陷检测模型的训练高度依赖大量标注样本。然而,真实生产线上出现的缺陷往往是偶发且形态多变的,收集覆盖所有异常类型的样本集几乎不可能。零样本异常生成正是为了解决这一数据饥渴问题而提出的:它要求在完全不接触目标产品真实异常图像的前提下,合成出逼真的缺陷样本。

目前主流的零样本生成路线有两条。第一条是基于文本的方法,例如AnomalyAny和AnoStyler,它们通过文字描述来引导扩散模型生成异常。但问题在于,像“划痕”或“凹痕”这样固定的文本提示,很难精确描述工业缺陷的复杂几何形态和随机纹理,导致生成的图案常常显得虚假,这就是所谓的语义鸿沟。

第二条是基于参考的方法,例如AnomalyControl和AnoMagic。它们从一张已有的异常产品图中提取异常信息,再将其迁移到新的产品上。这条路线看似更直接,却面临一个更棘手的难题:如何把异常从它的“宿主”产品上干净地剥离?如果解耦不彻底,源产品的纹理、颜色甚至光照就会泄露到目标产品上,生成一张“四不像”的图像。

DeCo的作者敏锐地指出,上述两类方法的根本缺陷在于缺乏对异常信息的显式建模。文本方法把异常压缩进模糊的语义空间,参考方法则让异常与源产品特征纠缠在一起。因此,核心挑战就变成了两个子问题:第一,如何获取一份不包含任何源产品信息的“纯净”异常结构?第二,如何将这份结构可控地融合到一个全新的产品表面上?

核心创新

  • 提出DeCo零样本工业异常生成框架,显式解耦异常结构与源产品,并重耦合至目标产品。
  • 设计Dual-Routing Flow (DR-Flow),包含AP-Routing隔离异常文本token和QK-Routing提取异常结构,实现异常结构与异常token的显式绑定。
  • 引入Product-Invariant Flow (PI-Flow)作为并行约束,防止异常token继承源产品特征。
  • 提出Product Compatibility Correction (PCC),在推理时补偿异常结构与目标产品间的不兼容性。

方法概览

提出DeCo框架,核心思想是解耦与重耦合。在异常信息获取阶段,通过Dual-Routing Flow (DR-Flow)将纹理不变的异常结构绑定至异常token,并利用Product-Invariant Flow (PI-Flow)防止异常token继承源产品特征。在异常-产品融合阶段,采用混合LoRA注入将异常结构重耦合至目标产品,并引入Product Compatibility Correction (PCC)补偿异常结构与产品间的不兼容性。

  • 整体架构:DeCo基于预训练的Rectified Flow扩散模型,采用Dual-LoRA设计。其中Product LoRA(P-LoRA)在正常产品图像上预训练并冻结,负责维护产品的身份特征;Anomaly LoRA(A-LoRA)则在异常参考图像上训练,专门捕获异常结构。这种分离为后续的解耦操作提供了基础。
  • Dual-Routing Flow(DR-Flow):这是实现异常结构提取的核心模块,包含两条路由。AP-Routing在文本层面工作,它将输入提示中的异常描述词(如“scratch”)与产品描述词(如“bottle”)在token维度上分离开,确保异常token只与A-LoRA交互,而产品token只与P-LoRA交互。
  • QK-Routing:在特征层面,QK-Routing通过修改自注意力机制中的Query-Key映射,强制A-LoRA仅对图像中的结构信息(如边缘、轮廓)做出响应,而忽略颜色、纹理等外观特征。这两条路由协同工作,将纹理不变的异常结构显式地绑定到了异常token上。
  • Product-Invariant Flow(PI-Flow):仅靠DR-Flow还不够。在训练时,异常token仍然可能通过扩散模型的去噪路径,间接“记住”源产品的特征。PI-Flow作为一个并行约束分支被引入,它要求使用异常token生成的图像,在经过一个产品分类器时,无法被识别为任何已知产品类别。这迫使异常token彻底丢弃产品身份信息。
  • 混合LoRA注入:在推理生成阶段,DeCo将训练好的A-LoRA与目标产品的P-LoRA进行混合。通过调整A-LoRA的注入权重,可以控制异常结构的显著程度。这种重耦合方式使得同一份异常结构可以被灵活地应用到不同的产品上。
  • Product Compatibility Correction(PCC):一个关键的细节是,相同的异常结构在不同产品上可能表现出不同的视觉强度。例如,同样的划痕在光滑的金属表面上很显眼,在粗糙的木纹上却可能被淹没。PCC模块在推理时引入一个可调节的标量ω,对异常结构的强度进行补偿,确保其在目标产品上具有合适的可见度。

逐图理解论文

核心洞察与贡献

核心洞察与贡献
Fig. 2: Training pipeline of DeCo. It employs a Dual-LoRA design with a frozen Product LoRA (blue, pre-trained on the left) and a trainable Anomaly LoRA (green). DR-Flow (Middle) binds the anomaly structure to the abnormal token via AP- Routing and QK-Routing. PI-Flow (Right) acts as a constraint to prevent the ab- normal token from directly inheriting product features from the source reference.

该图展示了DeCo的训练流程。左侧是冻结的P-LoRA,中间是DR-Flow通过AP-Routing和QK-Routing将异常结构绑定到异常token,右侧是PI-Flow约束异常token不携带产品信息。三者协同实现了干净的异常解耦。

重耦合与兼容性校正

重耦合与兼容性校正
Fig. 4: Zero-shot anomaly generation using the same reference anomaly. (a) Across different products, the anomaly varies in visibility. (b) With PCC to address the in- compatibility, the anomaly can be generated well.

该图说明了PCC的必要性。同一参考异常在不同产品上生成的可见度差异很大,例如在光滑表面上很清晰,在纹理丰富的表面上几乎看不见。应用PCC后,异常在所有产品上都能以合适的强度呈现。

结论与意义

结论与意义
Table 2: Comparison of anomaly detection with zero-shot anomaly synthesis and generation methods and decoupling methods on MVTec AD.

这套显式解耦与重耦合的策略,让DeCo在生成质量和下游检测任务上都显著超越了以往的方法。更重要的是,它证明了在零样本设定下,通过精细的结构化设计,我们完全可以获得高质量、可控的异常生成能力。这为工业视觉的数据增广打开了一条新的技术路径。当然,该方法在推理时需要为每张图额外进行兼容性校正,增加了约4.5秒的计算开销,这是它目前的一个主要局限。

实验如何设计?

  • 生成质量评估:在MVTec AD数据集上,使用IS和IS(a)评估生成异常图像的整体真实性和局部真实性,使用IC-LPIPS评估生成样本的多样性。对比方法包括基于文本的AnomalyAny和基于参考的AnoMagic等。
  • 下游检测任务评估:将生成的异常图像作为训练数据,训练一个标准的异常检测模型,然后在MVTec AD的真实测试集上评估其检测性能。指标涵盖图像级AUROC(I-AUC)、像素级AP(P-AP)和PRO等。
  • 跨数据集泛化:为了验证方法的鲁棒性,将在MVTec AD上训练的生成模型直接应用于VisA和Real-IAD数据集,评估其在不熟悉的产品类别上的检测性能。
  • 消融实验:系统性地移除AP-Routing、QK-Routing、PI-Flow和PCC等组件,以验证每个模块对最终性能的贡献。同时分析了A-LoRA的秩、PI-Flow的损失权重以及PCC标量ω的影响。

关键结果与论文证据

  • 生成质量显著领先:在MVTec AD上,DeCo的IS达到2.06,IS(a)达到3.98,IC-LPIPS为0.33,全面优于AnomalyAny和AnoMagic等基线方法(Table 1, p.12)。这表明DeCo生成的异常图像不仅更逼真,而且多样性更高。
  • 检测性能达到最优:在下游检测任务中,DeCo取得了98.0%的图像级AUROC和68.0%的像素级AP,在像素级指标上大幅超越第二名方法(Table 2, p.12)。这说明高质量的生成数据直接转化为了更强的检测能力。
  • 跨数据集泛化能力强:在VisA数据集上,DeCo的图像级AUROC达到95.9%,PRO达到88.3%,性能衰减很小(Table 3, p.14)。在更具挑战性的Real-IAD数据集上,DeCo同样表现出了最佳的泛化效果(Table 12, p.21)。
  • PI-Flow至关重要:消融实验显示,移除PI-Flow会导致像素级AP从68.0%骤降至一个很低的水平(Table 4, p.14)。可视化结果也证实,没有PI-Flow时,生成的异常会携带源产品的纹理(Fig. 3, p.8)。
  • PCC有效解决兼容性问题:定性结果(Fig. 4, p.10)和定量分析(Table 10, p.20)共同证明,PCC模块能够根据目标产品特性调整异常强度,避免异常“消失”或“过强”的情况。
  • A-LoRA需要全秩配置:实验发现,将A-LoRA的秩设为64会导致性能明显下降,必须采用全秩配置才能充分捕获复杂的异常结构(Table 6, p.19)。这揭示了工业异常的高度复杂性对模型容量的需求。

阅读时需要注意

  • 推理效率:PCC模块在推理时引入了额外的优化步骤,导致每张图像的生成时间增加约4.5秒(Table 5, p.18),这对于实时性要求极高的产线应用是一个挑战。
  • 参数调节依赖经验:PCC标量ω需要根据不同的产品类别手动调整,过大或过小的ω值都会破坏生成质量(Table 10, p.20),这增加了实际部署的复杂度。
  • 零样本设定的严格限制:方法的训练和验证严格遵循零样本设定,即目标产品不能有任何真实异常图像参与训练。在实际应用中,如何确保这一数据隔离是部署者需要谨慎对待的问题。

关联工作

  • 与Dual-LoRA解耦方法的区别:UnzipLoRA和QR-LoRA等工作也探索了双LoRA架构,但它们依赖预训练模型中的强语义先验(如“物体”与“风格”的分离)。工业异常通常是抽象的几何结构,缺乏明确的语义,因此这些方法难以直接适用。DeCo通过DR-Flow和PI-Flow实现了不依赖语义的显式解耦。
  • 与B-LoRA的关系:B-LoRA通过识别注意力模块的不同功能角色来实现物体与风格的分离。DeCo借鉴了这种思路,但进一步将其发展为面向异常结构的专用解耦框架,并引入了产品不变性约束。

发表评论