HyperAIHyperAI

Command Palette

Search for a command to run...

DRAEM——用于表面异常检测的判别性训练重构嵌入

Vitjan Zavrtanik Matej Kristan Danijel Skočaj

摘要

视觉表面异常检测旨在识别与正常外观显著偏离的图像局部区域。近年来的表面异常检测方法主要依赖生成模型,以精确重建正常区域,并在异常区域上表现失败。这类方法仅在无异常图像上进行训练,通常还需依赖人工设计的后处理步骤来实现异常定位,这限制了特征提取能力向最优检测性能的优化。除了重建式方法之外,本文将表面异常检测主要视为一个判别性问题,提出了一种判别性训练的重建异常嵌入模型(Discriminatively Trained Reconstruction Anomaly Embedding Model, DRAEM)。该方法在学习异常图像与其无异常重建版本的联合表示的同时,还同步学习正常样本与异常样本之间的决策边界。该方法无需额外复杂的网络输出后处理,即可实现直接的异常定位,并可通过简单且通用的异常模拟方式进行训练。在具有挑战性的MVTec异常检测数据集上,DRAEM显著超越了当前最先进的无监督方法;而在广泛使用的DAGM表面缺陷检测数据集上,其检测性能甚至接近全监督方法,且在定位精度方面实现显著超越。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供