基于多任务对比约束与可控缺陷生成的异常检测算法
Anomaly detection algorithm based on multi-task contrastive constraints and controllable anomaly synthesis
通讯作者:
收稿日期: 2025-10-8
| 基金资助: |
|
Received: 2025-10-8
| Fund supported: | 国家自然科学基金资助项目(62361002). |
作者简介 About authors
刘宇轩(2001—),男,硕士生,从事图像、视频语义分析研究.orcid.org/0009-0005-8351-3454.E-mail:
针对无监督视觉异常检测中正常与异常特征边界区分模糊的问题,提出基于多任务对比约束的算法. 通过可控异常生成模块生成高拟真的异常图像,作为训练投影层的多任务监督信号;在特征紧凑聚合任务中,通过最小化正常样本特征嵌入之间的距离,指导投影层将正常特征紧密聚集;在异常敏感任务中,通过最小化对比度损失,提高投影层对异常敏感区域的关注. 在异常特征抑制任务中,通过最小化特征相似度损失,抑制特征传递过程中的异常特征;通过自适应融合重建模块融合投影层提取的多尺度特征,实现精确的异常检测和定位. 在多个公开数据集上开展算法的有效性验证实验. 结果表明,所提算法在MVTec数据集上的图像级接受者特性曲线下面积(AUROC)为99.3%.
关键词:
A multi-task, contrastive-constraint-based algorithm was proposed to address the challenge of ambiguous boundaries between normal and anomalous features in unsupervised visual anomaly detection. First, highly realistic anomalous images were generated by a controllable anomaly synthesis module and used as multi-task supervisory signals to train the projection layer. Second, in the feature compactness aggregation task, the distances between normal sample feature embeddings were minimized to guide the projection layer toward a compact representation of normal features. In the anomaly sensitivity task, contrastive loss was minimized to increase the projection layer’s focus on anomaly-sensitive regions. In the anomaly feature suppression task, feature similarity loss was minimized to suppress anomalous features during feature propagation. Finally, multi-scale features extracted by the projection layer were fused by an adaptive fusion reconstruction module to enable accurate anomaly detection and localization. Effectiveness validation experiments were performed on multiple public datasets for the proposed algorithm. Results showed that the proposed algorithm achieved an image-level area under the receiver operating characteristic curve (AUROC) of 99.3% on the MVTec dataset.
Keywords:
本文引用格式
刘宇轩, 苏恩祈, 井佩光, 孙宏博, 苏育挺.
LIU Yuxuan, SU Enqi, JING Peiguang, SUN Hongbo, SU Yuting.
视觉异常检测是异常检测中的关键任务,在质量控制、安全监控等场景中有广泛应用. 在实际应用场景中,异常样本稀缺且标注成本高,无监督方法因无需标签的优势而备受关注[1].
传统无监督方法缺乏有效监督信号,易将异常区域误重构为正常,影响检测可靠性. 现有无监督视觉异常检测方法大致可分为两类,一类通过构建伪异常样本引入自监督信号,另一类通过精细建模正常特征空间识别偏离正常分布的异常样本. 伪异常构建方法又可分为基于图像变换和基于异常合成的方法. 基于图像变换的方法通过定义预设的几何变换、遮挡处理等图像变换操作,构造具有类别区分度的伪标签作为监督信号,驱动模型学习正常样本的隐式特征分布. 例如,Ye等[2]提出属性擦除策略,迫使模型复原正常样本的色彩与方向属性,异常区域因无法复原而被识别;Salehi等[3]采用拼图解序任务,通过重构正常图像的语义连贯性实现异常感知. 基于异常合成的异常检测方法通过可控合成技术向正常样本注入人工异常以生成带标签的训练样本. 例如,CutPaste[4]通过剪切-粘贴图像补丁合成异常并训练单类分类器. Schlüter等[5]在CutPaste基础上引入泊松图像编辑技术以增强异常逼真度. Zavrtanik等[6]构建级联式修复-分割网络,首阶段利用Perlin噪声合成异常并进行修复,次阶段通过比对修复前后差异实现高精度分割. 基于特征建模的方法,通过特征表示、差异感知或特征拟合实现异常识别. Zhang等[7]提出稀疏特征表示框架,通过正交匹配追踪在正常特征库中重构测试样本,以高斯核距离度量异常;Cai等[8]设计差异感知框架,利用教师-学生网络生成外观无关的差异图,提升模型对合成策略的鲁棒性;Zhang等[9]基于图模型动态拟合多尺度正常特征,实现无需训练的异常定位. 尽管自监督方法在无监督异常检测中取得进展,但在生成真实、多样化的异常方面仍具挑战,尤其是难以生成复杂的结构异常和未知的新异常类别. 因缺乏异常样本与先验知识,现有方法依赖人工设计的数据增强[2-3]或外部数据[5-6]合成异常,导致合成与真实异常间存在显著分布差异,难以准确建模正常与异常的边界,限制了对细微异常的敏感性及模型泛化能力. 以扩散模型[10]为代表的系列工作能生成高拟真的图像,为高质量异常样本的可控生成提供了新思路. 如何利用扩散模型生成与自然分布紧密契合的多种类型的异常样本,成为解决异常样本合成问题的关键.
针对上述问题,本研究提出基于多任务对比约束与可控缺陷生成的异常检测算法. 1)提出可控异常生成模块(基于扩散过程的异常合成策略),仅利用正常图像生成具有不同异常强度的伪异常图像,以模仿实际异常的特征分布,且不依赖任何异常数据. 2)将正常样本特征与伪异常特征分别输入对应的投影层,通过多任务对比约束联合训练投影层. 在特征紧凑聚合任务中,通过最小化正常样本特征嵌入之间的距离,使经过投影层映射后的正常特征在嵌入空间中形成紧致分布. 在异常敏感任务和异常特征抑制任务中,在多任务训练中添加与正常训练样本同等数量的异常对比信号来最小化对比度损失和特征相似度损失,迫使网络在特征提取和传递过程中关注异常敏感区域并通过指导投影层从伪异常特征中恢复正常特征来强化区分异常的能力. 3)设计自适应融合重建模块,对投影层提取的多尺度特征进行融合,精准重建无异常的参考图像,实现异常的检测和定位.
1. 算法内容构成
本研究所提算法架构如图1所示. 对于输入的正常图像,通过可控异常生成模块生成语义一致但包含可控扰动的伪异常图像. 原始正常图像与合成异常图像分别送入相同的主干网络提取多尺度特征张量,这两类特征随后分别进入独立的投影层进行非线性映射,形成对应的嵌入表示. 为了解决单一监督信号无法全面刻画正常与异常在特征分布、局部响应及可恢复性等方面的本质差异,在训练阶段,系统通过多任务对比约束实现特征解耦:利用自监督最优传输损失拉近正常特征间的分布距离;通过异常抑制损失从伪异常特征中恢复正常特征来提高对异常特征的抑制能力;借助带边距的对比度损失增强对异常区域的敏感性. 仅使用正常路径的多尺度投影特征,经通道分组增强与自适应融合后,输入对称解码器重建各层级正常特征. 推理时,通过计算原始特征与重建特征的余弦差异生成多尺度异常图,经上采样、累加与高斯平滑后得到像素级异常分数图,以其最大值作为图像级异常得分.
图 1
图 1 基于多任务对比约束与可控缺陷生成的异常检测算法框架
Fig.1 Framework of anomaly detection algorithm based on multi-task contrastive constraints and controllable anomaly synthesis
1.1. 可控异常生成模块
本研究提出基于动态扩散的可控异常生成模块,实现可控强度的异常合成. 该模块通过扩散模型深度建模正常样本的数据分布特征,在保持异常强度可调节性的同时,生成与真实数据分布高度吻合的多样化异常样本. 具体而言,采用去噪扩散概率模型[11]的正向扩散过程逐步将噪声
式中:
式中:均值函数
方差项
式中:
为了确保生成的异常图像接近正常图像的分布,设置
1.2. 多任务对比约束模型
式中:
在每个提取的特征之后直接集成一个投影层,投影层由4层的卷积、二维实例归一化和激活函数组成,第
在本研究中,“多任务”不是传统多任务学习中对多个独立预测任务的联合优化,而是在无监督异常检测框架下设计的3个自监督代理任务. 这些任务共享同一特征主干,但采用独立的投影层处理正常与伪异常特征,分别从正常特征紧密聚集、异常特征抑制与局部异常敏感性这3个互补维度施加约束,共同构成多任务对比约束机制. 多任务损失包括3个组成部分:1)特征紧凑聚合任务的自监督最优传输损失
1.2.1. 自监督最优传输
损失投影层从各自对应的特征提取网络中接收正常特征,并有意地将它们投影到紧凑的特征空间中. 该任务旨在通过最小化正常样本特征嵌入之间的距离确保来自正常样本的投影特征表示彼此接近. 为此,采用去偏Sinkhorn距离作为最小化特征嵌入分布距离的度量准则. 去偏Sinkhorn距离是最优传输距离的变体,其距离测度允许计算2个特征空间分布之间的空间差异. 给定源域与目标域概率分布
式中:
为了消除传统Sinkhorn算法因熵正则化导致的边缘分布偏差,引入去偏Sinkhorn距离作为修正度量:
式中:跨域传输项
式中:
1.2.2. 异常相似度损失
式中:
1.2.3. 异常敏感损失
为了强化投影层对异常特征的敏感度,使用带有边距
式中:
1.3. 自适应融合重建模块
自适应融合重建模块通过自适应特征聚合机制融合
式中:
式中:
特征重建器采用特征重建损失不断优化重建能力,特征重建损失
1.4. 训练与异常推理过程
多任务优化框架通过协同训练机制实现特征解耦与异常表征学习,单个训练周期包含以下关键步骤:1)特征提取器
式中:
图 2
式中:
为了定位图像中的异常,将
式中:
2. 实验设置及结果分析
2.1. 实验设置
本研究提出的异常检测方法的数据集图像尺寸为256×256,不施加数据增强. 这些图像设置被广泛用作异常检测方法对比的标准配置. 使用Adam优化器,学习率设为0.005,自监督损失、对比损失和重构损失的权重因子
2.2. 视觉异常数据集
在视觉异常检测领域,基于MVTec[15]和BTAD[16]这2个主流工业数据集开展实验. 所提方法在处理以局部结构或纹理缺陷为主导的异常时表现出显著优势,契合这2个数据集中异常样本的共性规律,展现出良好的适用性与有效性. MVTec包含5 354张高分辨率图像(700 像素×700像素~1 024像素×1 024像素),覆盖15个类别(5类工业纹理与10类工业制品),其中训练集含3 629张正常样本,测试集由1 725张正常与异常样本组成. 该数据集在高分辨率细节、低对比度背景、异常形态的真实性以及多视角采集等方面凸显了工业场景下异常检测的挑战性. BTAD是用于实例分割、语义分割、目标检测和半监督学习任务的数据集,共包含2 540张RGB图像,覆盖3种不同产品,其中691个样本带有像素级异常掩膜. 这3类产品图像分辨率分别为1 600像素×1 600像素、600像素×600像素和800像素×600像素,对应的训练集规模为400、1 000和399张,所有图像在训练前统一缩放至512像素.
2.3. 评估指标
采用三级评估体系对算法性能进行量化分析. 该评估框架遵循现有研究公认的检测标准,从图像级异常判别、像素级定位精度和区域级重叠度这3个维度建立综合评价指标.
图像级异常检测性能通过AUROCI进行量化. 对于包含N个测试样本的数据集,给定模型输出的异常分数
式中:
AUROCP将异常定位结果视为像素二分类问题,通过对比预测热力图
式中:
针对视觉异常检测中常见的多尺度异常特性,引入AUPRO曲线积分方法,该指标赋予小尺寸异常更高权重,有效避免传统指标对大面积异常的评估偏好.
式中:
AUROCI反映模型对异常样本的全局判别能力,但对定位精度不敏感. AUROCP侧重像素级定位准确性,评估异常区域的空间识别精度. AUPRO通过区域加权机制,计算预测区域与真实掩膜的重叠度,重点考察小尺寸异常的检测鲁棒性,三者构成互补评估体系.
2.4. 对比实验与分析
在MVTec数据集上进行异常检测与异常定位,不同算法在各品类及整体平均值上的AUROCI与AUROCP定量结果分别如表1和表2所示. 可以看出,所提算法在整体性能上展现出明显优势:AUROCI平均值为99.3%,AUROCP平均值为98.3%,显著优于RD4AD、ATSN、PSVDD、RIAD等对比方法,验证了模型的有效性与泛化性. 在图像级异常判别中,在多个品类中的AUROCI达到100%,表明所提算法在这些场景下能完美区分正常与异常样本,优于对比方法PSVDD和RIAD. 即使在部分难度较高的品类中,所提算法仍超越多数对比方法,并在多数品类中AUROCI最大,且平均值领先,提升了异常检测的精度. 在像素级定位中,所提算法在多个品类中取得最大值,体现了该算法对纹理和复杂形状物体异常的精准定位能力. 即便在小尺寸、精细结构类别(如晶体管)中,该算法仍显著优于RIAD和NIGSF,体现了细粒度异常捕捉能力.
表 1 不同算法在MVTec数据集中的图像级异常检测结果
Tab.1
| 类别 | AUROCI/% | ||||||
| PSVDD | RIAD | PaDiM | RD4AD | ATSN | NIGSF | 本研究 | |
| 地毯 | 92.9 | 84.2 | 99.8 | 98.9 | 99.8 | 96.5 | 100 |
| 铁丝网 | 94.6 | 99.6 | 96.7 | 100 | 99.9 | 100 | 100 |
| 皮革 | 90.9 | 100 | 100 | 100 | 100 | 100 | 100 |
| 砖瓦 | 97.8 | 98.7 | 98.1 | 99.3 | 98.8 | 100 | 99.9 |
| 木头 | 96.5 | 93.0 | 99.2 | 99.2 | 99.6 | 99.7 | 99.5 |
| 瓶子 | 98.6 | 99.9 | 99.9 | 100 | 100 | 96.0 | 100 |
| 电缆 | 90.3 | 81.9 | 92.7 | 95.0 | 100 | 85.6 | 99.2 |
| 胶囊 | 76.7 | 88.4 | 91.3 | 96.3 | 96.8 | 93.9 | 97.5 |
| 榛子 | 92.0 | 83.3 | 92.0 | 99.9 | 100 | 99.8 | 100 |
| 金属 | 94.0 | 88.5 | 98.7 | 100 | 98.7 | 99.7 | 100 |
| 药丸 | 86.1 | 83.8 | 93.3 | 96.6 | 96.7 | 97.2 | 98.2 |
| 螺丝钉 | 81.3 | 84.5 | 85.8 | 97.0 | 96.8 | 86.6 | 98.6 |
| 牙刷 | 100 | 100 | 96.1 | 99.5 | 93.1 | 100 | 100 |
| 晶体管 | 91.5 | 90.9 | 97.4 | 96.7 | 98.3 | 95.3 | 98.8 |
| 拉链 | 97.9 | 98.1 | 90.3 | 98.5 | 98.2 | 99.6 | 98.4 |
| 平均值 | 92.1 | 91.7 | 95.5 | 98.5 | 98.4 | 96.7 | 99.3 |
表 2 不同算法在MVTec数据集中的像素级异常检测结果
Tab.2
| 类别 | AUROCP/% | ||||||
| PSVDD | RIAD | PaDiM | RD4AD | ATSN | NIGSF | 本研究 | |
| 地毯 | 92.6 | 96.3 | 99.1 | 98.9 | 99.1 | 97.4 | 99.2 |
| 铁丝网 | 96.2 | 98.8 | 97.3 | 99.3 | 99.0 | 99.2 | 99.3 |
| 皮革 | 97.4 | 99.4 | 99.2 | 99.4 | 99.4 | 98.9 | 99.5 |
| 砖瓦 | 91.4 | 89.1 | 94.1 | 95.6 | 95.0 | 99.7 | 96.6 |
| 木头 | 90.8 | 85.8 | 94.9 | 95.3 | 94.8 | 97.3 | 96.1 |
| 瓶子 | 98.1 | 98.4 | 98.3 | 98.7 | 98.8 | 96.3 | 98.8 |
| 电缆 | 96.8 | 84.2 | 96.7 | 97.4 | 98.0 | 93.3 | 98.5 |
| 胶囊 | 95.8 | 92.8 | 98.5 | 98.7 | 98.6 | 97.4 | 98.7 |
| 榛子 | 97.5 | 96.1 | 98.2 | 98.9 | 98.9 | 99.5 | 99.2 |
| 金属 | 98.0 | 92.5 | 97.2 | 97.3 | 96.0 | 99.5 | 98.1 |
| 药丸 | 95.1 | 95.7 | 95.7 | 98.2 | 98.0 | 99.5 | 98.4 |
| 螺丝钉 | 95.7 | 98.8 | 98.5 | 99.6 | 99.1 | 97.2 | 99.6 |
| 牙刷 | 98.1 | 98.9 | 98.8 | 99.1 | 98.9 | 99.6 | 99.1 |
| 晶体管 | 97.0 | 87.7 | 97.5 | 92.5 | 94.5 | 84.7 | 94.8 |
| 拉链 | 95.1 | 97.8 | 98.5 | 98.2 | 98.3 | 98.2 | 98.5 |
| 平均值 | 95.7 | 94.2 | 97.5 | 97.8 | 97.8 | 97.2 | 98.3 |
不同算法在MVTec数据集上基于AUPRO指标的异常定位结果如表3所示. AUPRO对小范围异常检测更具挑战性,可以看出,所提算法AUPRO平均值为95.0%,位居榜首. 所提算法在纹理类(如地毯、铁丝网、皮革)及规则物体、精密零件(如胶囊、螺丝钉)中均表现出色,优于PSVDD和RIAD. 这表明在小范围异常定位这一高难度任务中,所提算法能更精准捕捉异常区域,验证了算法对复杂异常检测场景的适应性与优越性.
表 3 不同算法在MVTec数据集中的区域级异常检测结果
Tab.3
| 类别 | AUPRO/% | ||||||
| PSVDD | RIAD | PaDiM | RD4AD | ATSN | NIGSF | 本研究 | |
| 地毯 | 86.5 | 61.4 | 96.2 | 97.0 | 97.3 | 70.5 | 97.7 |
| 铁丝网 | 84.0 | 36.4 | 94.6 | 97.6 | 97.2 | 66.9 | 97.7 |
| 皮革 | 75.7 | 49.1 | 97.8 | 99.1 | 99.1 | 73.2 | 99.2 |
| 砖瓦 | 92.5 | 52.6 | 86.0 | 90.6 | 88.8 | 97.2 | 92.5 |
| 木头 | 81.6 | 38.2 | 91.1 | 90.9 | 92.4 | 84.4 | 93.5 |
| 瓶子 | 80.5 | 76.4 | 94.8 | 96.6 | 96.4 | 81.5 | 96.9 |
| 电缆 | 84.0 | 24.4 | 88.8 | 91.0 | 92.7 | 62.0 | 94.3 |
| 胶囊 | 86.2 | 38.2 | 93.5 | 95.8 | 95.7 | 51.6 | 95.7 |
| 榛子 | 84.7 | 33.8 | 92.6 | 95.5 | 95.1 | 83.7 | 96.3 |
| 金属 | 84.0 | 64.3 | 85.6 | 92.3 | 88.4 | 96.8 | 92.3 |
| 药丸 | 82.5 | 51.6 | 92.7 | 96.4 | 96.1 | 93.2 | 97.0 |
| 螺丝钉 | 85.9 | 43.9 | 94.4 | 98.2 | 96.6 | 43.8 | 98.5 |
| 牙刷 | 83.9 | 50.6 | 93.1 | 94.5 | 93.2 | 94.2 | 94.5 |
| 晶体管 | 50.3 | 39.2 | 84.5 | 78.0 | 83.9 | 40.1 | 83.2 |
| 拉链 | 16.9 | 63.4 | 95.9 | 95.4 | 95.0 | 83.2 | 95.8 |
| 平均值 | 77.3 | 48.2 | 92.1 | 93.9 | 93.9 | 74.8 | 95.0 |
不同算法在BTAD数据集上的AUROCI与AUROCP如表4所示. BTAD数据集作为更具挑战性的测试场景,对方法的鲁棒性与精度提出更高要求. BTAD包含3类匿名工业产品:Class 01、Class 02和Class 03. Class 01和Class 03为具有环形结构的工业部件,Class 02为具有纵向纹理的平面工业产品. 所提算法平衡了检测与定位精度,在综合性能上实现了突破,AUROCI平均值为96.7%,AUROCP平均值为97.4%,均领先于对比方法,尤其在Class 03中展现出卓越性能. 整体结果表明,所提算法对复杂、多样化的异常场景具有更强的鲁棒性,进一步印证了研究方法的创新性与实用性.
表 4 不同算法在BTAD数据集中的图像级和像素级异常检测结果
Tab.4
| 算法 | AUROCI/% | AUROCP/% | |||||||
| Class 01 | Class 02 | Class 03 | 平均值 | Class 01 | Class 02 | Class 03 | 平均值 | ||
| VT-ADL | 99.0 | 94.0 | 77.0 | 90.0 | 97.6 | 71.0 | 82.6 | 83.7 | |
| P-SVDD | 91.6 | 93.6 | 91.0 | 92.1 | 95.7 | 72.1 | 82.1 | 83.3 | |
| FastFlow | 99.4 | 82.4 | 91.1 | 90.9 | 97.1 | 93.6 | 98.3 | 96.3 | |
| PatchCore | 95.5 | 94.7 | 99.3 | 96.5 | 90.9 | 79.3 | 99.8 | 90.0 | |
| CFA | 98.1 | 85.5 | 99.0 | 94.2 | 95.9 | 96.0 | 98.6 | 96.8 | |
| AWSSA-UNet | 93.3 | 91.9 | 99.0 | 94.7 | 100 | 82.0 | 99.8 | 93.9 | |
| 本研究 | 96.5 | 93.9 | 99.7 | 96.7 | 96.2 | 96.4 | 99.7 | 97.4 | |
2.5. 收敛性分析
对所提算法进行收敛性分析,重点关注MVTec数据集中的“地毯”类别,算法训练过程中的收敛曲线如图3所示. 结果表明,3项性能指标曲线均呈上升态势,初期上升斜率较大,此时模型快速学习数据特征,性能显著提升;后期趋于平缓,说明模型在像素级异常定位、图像级异常检测及小范围异常检测任务上的性能逐步饱和,能够稳定输出高精度结果,验证了算法对异常检测任务的有效建模能力. 训练的总损失曲线呈现持续下降趋势,前期下降速率快,后期趋于平稳并维持在较低水平. 这表明训练过程中模型参数不断优化,对数据特征的学习愈发精准,最终实现有效收敛,确保模型仍能保持优异的异常检测与定位性能,并且其余类别的收敛行为与此基本一致,整体趋势稳定. 综上,性能指标的提升与损失的下降相互佐证,表明了模型参数优化的合理性,也证实了算法在异常检测与定位任务中学习特征、提升精度的有效性.
图 3
2.6. 可视化分析
图 4
图 4 不同异常合成方法的图像对比
Fig.4 Image comparison of different anomaly synthesis methods
如图5所示为异常样本图像、模型生成的异常概率分布图谱和像素级真值标注组成的三级对比结构,直观展示了所提算法在MVTec数据集上的异常定位效果,对模型检测效果进行了验证. 该可视化图包含3个部分:首层选取如金属零部件和纤维制品的典型工业产品的异常样本,覆盖材料属性和形态特征的多样性;中间层呈现基于本研究的异常概率分布图谱,采用红-蓝渐变色阶表征区域异常程度;底层叠加实验数据集提供的像素级真值标注,通过二值化掩膜精确标定异常边界. 对比结果表明,本研究构建的异常检测模型能够准确定位孔洞、裂纹、形变等复杂异常形态,其热力分布图与标注真值具有高度空间一致性. 特别在跨品类适应性方面,模型对织物纹理异常与金属表面瑕疵均保持稳定检测能力,展现出优异的跨类别泛化性能. 综上,所提算法提升了对各数据类别异常区域的检测能力,体现出良好的泛化性与检测准确性.
图 5
图 5 所提算法在MVTec数据集上的异常定位可视化结果
Fig.5 Anomaly localization visualization results of proposed algorithm on MVTec dataset
2.7. 消融实验
对训练组件的有效性展开探究,不同目标训练组合的性能表现如表5所示. 为了验证可控异常生成模块的有效性,设置与CutPaste异常合成策略的对比实验. 结果显示,AUROCI、AUROCP及AUPRO分别提高了0.4、0.3与0.4个百分点. 这说明可控异常生成模块生成的异常特征在全局分布上更贴近真实异常,减少了CutPaste因几何形变导致的干扰,同时在边界过渡上更平滑,避免了CutPaste策略中硬性粘贴导致的边缘伪影,使算法能够更准确地学习异常区域的局部纹理特征. 为了验证多任务对比约束模块和自适应融合重建模块(adaptive fusion and reconstruction nodule,AFRM)的有效性,将仅采用基础特征重建损失
表 5 在MVTec数据集上不同训练目标组合的性能表现
Tab.5
| 目标训练组合 | AUROCI/% | AUROCP/% | AUPRO/% |
| 可控异常生成策略 | 99.3 | 98.3 | 95.0 |
| CutPaste 异常合成策略 | 98.9 | 98.0 | 94.6 |
| 基线 | 98.4 | 97.7 | 93.8 |
| 基线 | 99.0 | 98.1 | 94.5 |
| 基线 | 99.1 | 98.1 | 94.6 |
| 基线 | 99.3 | 98.2 | 94.8 |
| 基线 | 99.3 | 98.3 | 95.0 |
2.8. 复杂度实验
如表6所示为不同方法在推理时间ti、内存使用M及性能上的对比,反映模型复杂度与实际应用效率. 可以看出,所提算法的ti=0.22 s,显著低于对比算法,能够快速完成异常检测任务,满足实时性要求较高的应用场景需求. 在内存使用方面,所提算法仅需456 MB,远低于PaDiM与SPADE,降低了模型运行开销. 实验结果充分验证了所提算法在模型复杂度控制上的有效性,为其在视觉异常检测中的落地应用提供了坚实的性能支撑.
表 6 不同算法在MVTec数据集上的复杂度对比
Tab.6
| 算法 | ti/s | M/MB | AUROCI/% | AUROCP/% | AUPRO/% |
| P-SVDD | 0.48 | — | 92.1 | 95.7 | 77.3 |
| PaDiM | 0.95 | 95.5 | 97.5 | 92.1 | |
| SPADE | 1.40 | 85.5 | 96.5 | 91.7 | |
| 本研究 | 0.22 | 456 | 99.3 | 98.3 | 95.0 |
3. 结 语
本研究提出基于扩散模型构造训练样本的多任务对比约束视觉异常检测算法,利用正常样本合成具有可控异常强度的伪异常图像,在提取特征张量后引入各自对应的投影层,设计3项协同优化任务(特征紧凑聚合、异常敏感与异常特征抑制任务),有效实现了正常与异常特征在嵌入空间中的解耦. 在此基础上,结合自适应融合重建模块,实现了高精度的异常检测与定位. 实验结果表明,所提算法的性能优于现有的先进方法. 所提算法仍存在局限性:1)扩散模型生成的伪异常样本语义多样性与真实工业异常仍有差距,可能限制对复杂或结构性异常的泛化能力;2)尽管无需异常样本训练,模型性能仍依赖高质量的正常数据建模,当正常样本含噪声或类别混杂时,特征紧凑性易被破坏,影响检测精度. 未来工作将探索更高效的扩散引导机制与自适应对比约束策略,以增强模型在跨类别零样本场景下的泛化能力,并提升对未知异常的适应性与鲁棒性;将引入噪声鲁棒的特征学习机制,缓解正常样本污染对特征紧凑聚合任务的负面影响. 所提算法仅需正常样本进行训练,即可有效提升异常检测的判别能力与鲁棒性,其结合可控缺陷生成与多任务对比约束的设计思路,为后续面向复杂工业场景的异常检测研究奠定了基础.
参考文献
基于改进生成对抗网络的飞参数据异常检测方法
[J].DOI:10.3785/j.issn.1008-973X.2022.10.008 [本文引用: 1]
Flight parameter data anomaly detection method based on improved generative adversarial network
[J].DOI:10.3785/j.issn.1008-973X.2022.10.008 [本文引用: 1]
Attribute restoration framework for anomaly detection
[J].DOI:10.1109/TMM.2020.3046884 [本文引用: 2]
Low-shot unsupervised visual anomaly detection via sparse feature representation
[J].DOI:10.1109/TNNLS.2024.3420818 [本文引用: 1]
A discrepancy aware framework for robust anomaly detection
[J].DOI:10.1109/TII.2023.3318302 [本文引用: 1]
A graph model-based multiscale feature fitting method for unsupervised anomaly detection
[J].DOI:10.1016/j.patcog.2023.109373 [本文引用: 1]
Reconstruction by inpainting for visual anomaly detection
[J].DOI:10.1016/j.patcog.2020.107706 [本文引用: 1]
Normal image guided segmentation framework for unsupervised anomaly detection
[J].DOI:10.1109/TCSVT.2023.3327448 [本文引用: 1]
A hybrid prototype selection-based deep learning approach for anomaly detection in industrial machines
[J].DOI:10.1016/j.eswa.2022.117528 [本文引用: 1]
Dual-constraint autoencoder and adaptive weighted similarity spatial attention for unsupervised anomaly detection
[J].DOI:10.1109/TII.2024.3384583 [本文引用: 1]
/
| 〈 |
|
〉 |

