3 个月前

基于归一化流密度估计的异常检测与合成缺陷分类

基于归一化流密度估计的异常检测与合成缺陷分类

摘要

我们提出了一种基于深度学习的新型异常检测(Anomaly Detection, AD)系统,该系统通过共享特征提取器,将像素级分类网络与条件归一化流(Conditional Normalizing Flow, CNF)网络相结合。首先,利用合成的异常数据训练像素级分类网络,以微调CNF网络的预训练特征提取器,从而学习到域内数据的判别性特征。随后,使用正常数据并结合已微调的特征提取器,对CNF网络进行训练,以估计正常数据的密度分布。在推理阶段,通过计算像素级分类网络与CNF网络异常得分的加权平均值来实现异常检测。由于所提出的系统不仅学习了域内数据的内在特性,还融合了分类网络与CNF网络的异常得分信息,因此在MvTecAD和BTAD数据集上的实验中,其性能显著优于现有方法。此外,由于分类与密度估计两个模块共享特征提取器,该系统并未显著增加计算开销。

基准测试

基准方法指标
anomaly-detection-on-btadAD-CLSCNFs
Detection AUROC: 95.93
Segmentation AP: 55.86
Segmentation AUPRO: 72.77
Segmentation AUROC: 97.13
anomaly-detection-on-mvtec-adAD-CLSCNFs
Detection AUROC: 98.85
Segmentation AP: 74.32
Segmentation AUPRO: 96.01
Segmentation AUROC: 98.74

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于归一化流密度估计的异常检测与合成缺陷分类 | 论文 | HyperAI超神经