3 个月前

EfficientAD:毫秒级延迟下的高精度视觉异常检测

EfficientAD:毫秒级延迟下的高精度视觉异常检测

摘要

在图像中检测异常是一项重要任务,尤其在实时计算机视觉应用中尤为关键。本文聚焦于计算效率,提出了一种轻量级特征提取器,可在现代GPU上以不到一毫秒的时间处理单张图像。随后,我们采用学生-教师(student-teacher)框架来检测异常特征:训练一个学生网络,使其能够预测正常(即无异常)训练图像的特征表示。在测试阶段,当学生网络无法准确预测某图像的特征时,即表明该图像存在异常。为此,我们设计了一种新型训练损失函数,有效抑制学生网络在正常图像之外对教师特征提取器的过度模仿,从而显著降低学生-教师模型的计算开销,同时提升异常特征的检测性能。此外,我们还针对具有挑战性的逻辑异常(logical anomalies)进行了研究,这类异常表现为正常局部特征的非法组合,例如物体排列顺序错误等。我们通过高效集成一个全局分析的自编码器(autoencoder),实现了对这类异常的有效检测。我们在三个工业级异常检测数据集集合中的32个数据集上对所提出的算法——EfficientAD进行了评估。实验结果表明,EfficientAD在异常检测与定位方面均树立了新的基准。在仅2毫秒延迟、每秒处理600张图像的高吞吐量下,该方法实现了对异常的快速响应。结合其极低的误报率,EfficientAD不仅为实际应用场景提供了经济高效的解决方案,也为未来的研究奠定了坚实基础。

代码仓库

SimonThomine/DistillationAD
pytorch
GitHub 中提及
ntkhoa95/EfficientAD
pytorch
GitHub 中提及
liutongkun/EfficientAD
pytorch
GitHub 中提及
arungansi/EfficientAD
pytorch
GitHub 中提及
WS-W/EfficientAD
pytorch
GitHub 中提及
crimama/EfficientAD
pytorch
GitHub 中提及
mjack3/EfficientAD
pytorch
GitHub 中提及
guome/EfficientAD
pytorch
GitHub 中提及
VealFang/EfficientAD_realdata
pytorch
GitHub 中提及
lemonxiaohei/EfficientAD
pytorch
GitHub 中提及
MohammedEltoum/EfficientAD
pytorch
GitHub 中提及
danielsoy/EfficientAD-1
pytorch
GitHub 中提及
ziyouzheilu/EfficientAD
pytorch
GitHub 中提及
Emyyr/EfficientAD
pytorch
GitHub 中提及
dellysursh/EfficientAD
pytorch
GitHub 中提及
LeapMind/PUAD
pytorch
GitHub 中提及
henry1985/EfficientAD
pytorch
GitHub 中提及
Zaza-97/EfficientAD
pytorch
GitHub 中提及
lorenzomammana/EfficientAD
pytorch
GitHub 中提及
nelson1425/EfficientAD
pytorch
GitHub 中提及
zhn6818/EfficientAD
pytorch
GitHub 中提及
FourthM/EfficientAD
pytorch
GitHub 中提及
cinout/EfficientAD
pytorch
GitHub 中提及
Eacient/EfficientAD
pytorch
GitHub 中提及

基准测试

基准方法指标
anomaly-detection-on-mvtec-adEfficientAD-M
Detection AUROC: 99.1
FPS: 269
Segmentation AUPRO: 93.5
anomaly-detection-on-mvtec-adEfficientAD-S
Detection AUROC: 98.7
FPS: 614
Segmentation AP: 65.9
Segmentation AUPRO: 93.1
anomaly-detection-on-mvtec-adEfficientAD (early stopping)
Detection AUROC: 99.8
FPS: 269
anomaly-detection-on-mvtec-loco-adEfficientAD-M
Avg. Detection AUROC: 90.7
Detection AUROC (only logical): 86.8
Detection AUROC (only structural): 94.7
Segmentation AU-sPRO (until FPR 5%): 79.8
anomaly-detection-on-mvtec-loco-adEfficientAD-S
Avg. Detection AUROC: 90.0
Detection AUROC (only logical): 85.8
Detection AUROC (only structural): 94.1
Segmentation AU-sPRO (until FPR 5%): 77.8
anomaly-detection-on-visaEfficientAD-S
Detection AUROC: 97.5
Segmentation AUPRO (until 30% FPR): 93.1
anomaly-detection-on-visaEfficientAD-M
Detection AUROC: 98.1
Segmentation AUPRO (until 30% FPR): 94.0

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
EfficientAD:毫秒级延迟下的高精度视觉异常检测 | 论文 | HyperAI超神经