HyperAIHyperAI

Command Palette

Search for a command to run...

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

Seungmi Oh

摘要

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


用 AI 构建 AI

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

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

HyperAI Newsletters

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