arXiv:2410.14379cs.CV2024-10CVPR被引 17

新方法让工业异常检测自动发现新类别,性能显著提升。

AnomalyNCD: Towards Novel Anomaly Class Discovery in Industrial Scenarios

  • 通过二值化聚焦异常区域,避免误检干扰
  • 利用掩码引导学习弱语义异常,提升识别精度
  • 支持区域与图像级分类,适合工业场景应用

近期多类异常分类受到关注,但现有方法因缺乏异常先验知识而表现受限,主要问题在于异常不显著和语义微弱。本文提出AnomalyNCD,一种兼容多种异常检测方法的多类异常分类网络。为解决异常不显著问题,设计主成分二值化(MEBin),生成以异常为中心的图像,确保异常被有效学习且不受错误检测影响。为应对语义微弱的异常,引入掩码引导表示学习,通过掩码聚焦孤立异常,并用修正的伪标签减少错误输入带来的混淆。为实现灵活的区域与图像级分类,提出区域合并策略,根据分类后的异常区域判定整图类别。在MVTec AD和MTD数据集上,该方法优于现有最先进模型:结合零样本异常检测方法时,在MVTec AD上分别提升10.8% F_1、8.8% NMI、9.5% ARI;在MTD上分别提升12.8% F_1、5.7% NMI、10.8% ARI。代码已开源。

原文摘要 · Abstract (English)

Recently, multi-class anomaly classification has garnered increasing attention. Previous methods directly cluster anomalies but often struggle due to the lack of anomaly-prior knowledge. Acquiring this knowledge faces two issues: the non-prominent and weak-semantics anomalies. In this paper, we propose AnomalyNCD, a multi-class anomaly classification network compatible with different anomaly detection methods. To address the non-prominence of anomalies, we design main element binarization (MEBin) to obtain anomaly-centered images, ensuring anomalies are learned while avoiding the impact of incorrect detections. Next, to learn anomalies with weak semantics, we design mask-guided representation learning, which focuses on isolated anomalies guided by masks and reduces confusion from erroneous inputs through corrected pseudo labels. Finally, to enable flexible classification at both region and image levels, we develop a region merging strategy that determines the overall image category based on the classified anomaly regions. Our method outperforms the state-of-the-art works on the MVTec AD and MTD datasets. Compared with the current methods, AnomalyNCD combined with zero-shot anomaly detection method achieves a 10.8% $F_1$ gain, 8.8% NMI gain, and 9.5% ARI gain on MVTec AD, and 12.8% $F_1$ gain, 5.7% NMI gain, and 10.8% ARI gain on MTD. Code is available at https://github.com/HUST-SLOW/AnomalyNCD.

异常检测工业质检多类分类零样本

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。