3 个月前

基于弱监督残差Transformer的工业异常检测与定位

基于弱监督残差Transformer的工业异常检测与定位

摘要

近年来,工业异常检测(Anomaly Detection, AD)领域的进展表明,在训练过程中引入少量异常样本可显著提升检测精度。然而,这种性能提升通常伴随着高昂的标注成本,而精确的像素级标注在许多实际应用场景中难以实现。为此,本文提出一种新型框架——弱监督残差Transformer(Weakly-supervised RESidual Transformer, WeakREST),旨在实现高精度异常检测的同时,最大限度减少对人工标注的依赖。首先,我们将像素级异常定位任务重新建模为块级分类问题,从而降低标注复杂度。其次,提出一种基于残差的特征表示方法——位置快速异常残差(Positional Fast Anomaly Residuals, PosFAR),能够更有效地捕捉异常模式。为充分利用该特征表示,我们对Swin Transformer架构进行适配,以增强异常检测与定位能力。此外,本文设计了一种弱标注策略,仅需使用边界框(bounding boxes)和图像标签即可定义异常区域,构建半监督学习范式,显著降低对精细像素级标签的依赖。为进一步优化学习过程,我们提出一种新颖的ResMixMatch算法,能够有效处理弱标签与基于残差的特征表示之间的协同关系。在MVTec-AD基准数据集上的实验结果表明,本方法在无监督设置下实现了83.0%的平均精度(Average Precision, AP),超越此前最优结果82.7%。在有监督AD设置中,WeakREST达到87.6%的AP,优于先前最佳方法的86.0%。尤为突出的是,即便采用较弱的标注形式(如边界框),WeakREST仍显著优于依赖像素级标注的先进方法,在MVTec-AD上取得87.1%的AP,远超此前最优的86.0%。

基准测试

基准方法指标
anomaly-detection-on-btadWeakREST-Un
Detection AUROC: 94.4
Segmentation AP: 63.1
Segmentation AUPRO: 84.9
Segmentation AUROC: 98.7
anomaly-detection-on-mvtec-adWeakREST-Un
Detection AUROC: 99.6
FPS: 25.2
Segmentation AP: 83.0
Segmentation AUPRO: 97.6
Segmentation AUROC: 99.3
supervised-anomaly-detection-on-btadWeakREST-Block
Detection AUROC: 96.5
Segmentation AP: 84.6
Segmentation AUPRO: 90.8
Segmentation AUROC: 99.3
supervised-anomaly-detection-on-mvtec-adWeakREST-Block
Detection AUROC: 99.8
Segmentation AP: 87.6
Segmentation AUPRO: 98.4
Segmentation AUROC: 99.7
unsupervised-anomaly-detection-onWeakREST-Un
Segmentation AP: 76.9
Segmentation AUPRO: 98.5
Segmentation AUROC: 99.7

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于弱监督残差Transformer的工业异常检测与定位 | 论文 | HyperAI超神经