3 个月前

重新审视用于异常检测的反向蒸馏

重新审视用于异常检测的反向蒸馏

摘要

异常检测在大规模工业制造中是一项重要应用。近年来,该任务的诸多方法虽展现出优异的检测精度,但普遍存在延迟较高的问题。以PatchCore或基于耦合超球面的特征适配(Coupled-hypersphere-based Feature Adaptation, CFA)为代表的基于内存的方法虽性能领先,但依赖外部记忆库,显著增加了执行时间。另一种采用反向蒸馏(Reversed Distillation, RD)的方法则能在保持低延迟的同时实现良好性能。本文重新审视这一思路,对其进行了改进,在具有挑战性的MVTec数据集上,同时实现了异常检测与定位任务的全新最先进性能。所提出的RD++方法在速度上较PatchCore提升六倍,较CFA提升两倍,且相较于RD仅引入可忽略的额外延迟。此外,我们在BTAD和Retinal OCT数据集上进行了实验,验证了该方法的泛化能力,并开展了关键的消融实验,深入分析了其模型配置的影响。代码将开源,地址为:https://github.com/tientrandinh/Revisiting-Reverse-Distillation。

基准测试

基准方法指标
anomaly-detection-on-btadReverse Distillation ++
Detection AUROC: 95.63
Segmentation AUROC: 97.43
anomaly-detection-on-inspladRD++ (ResNet-18)
Detection AUROC: 90.07
anomaly-detection-on-mvtec-adReverse Distillation ++
Detection AUROC: 99.44
Segmentation AUPRO: 94.99
Segmentation AUROC: 98.25
anomaly-detection-on-mvtec-ad-texturesReverse Distillation ++
Detection AUROC: 99.8

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
重新审视用于异常检测的反向蒸馏 | 论文 | HyperAI超神经