3 个月前

可度量多样性异常检测

可度量多样性异常检测

摘要

基于重建的异常检测模型通过抑制对异常的泛化能力来实现其目标,但由此导致正常样本中多样的模式也难以被有效重建。尽管已有部分研究尝试通过建模样本多样性来缓解该问题,但由于异常信息的不当传播,这些方法仍存在捷径学习(shortcut learning)的缺陷。为更好地应对这一权衡问题,本文提出一种名为“可度量多样性异常检测”(Diversity-Measurable Anomaly Detection, DMAD)的新框架,旨在增强重建过程中的多样性,同时避免对异常样本产生不必要的泛化。为此,我们设计了金字塔形形变模块(Pyramid Deformation Module, PDM),该模块通过从重建结果到原始输入之间的多尺度形变场估计,同时建模多样化的正常模式并量化异常的严重程度。结合信息压缩模块后,PDM 实质上实现了形变信息与原型嵌入的解耦,从而提升了最终异常评分的可靠性。在监控视频与工业图像上的实验结果表明,所提方法具有显著有效性。此外,DMAD 在面对污染数据以及具有异常特征的正常样本时,同样表现出稳健的性能。

代码仓库

FlappyPeggy/DMAD
官方
pytorch

基准测试

基准方法指标
anomaly-detection-on-chuk-avenueConvVQ
AUC: 84.3%
anomaly-detection-on-chuk-avenueDMAD
AUC: 92.8%
anomaly-detection-on-mvtec-adDMAD
Detection AUROC: 99.5
Segmentation AUROC: 98.2
anomaly-detection-on-shanghaitechDMAD
AUC: 78.8%
anomaly-detection-on-ucsd-ped2DMAD
AUC: 99.7%
anomaly-detection-on-ucsd-ped2ConvVQ
AUC: 90.2%

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
可度量多样性异常检测 | 论文 | HyperAI超神经