arXiv:2505.09264cs.CVcs.AI2025-05ECCV被引 33

仅用一张正常图像即可检测多种异常,提升精度与泛化能力。

Learning to Detect Multi-class Anomalies with Just One Normal Image Prompt

  • 通过单张正常图像提示重建正常特征,恢复异常特征。
  • 在MVTec、BTAD、VisA三个数据集上均超越现有方法。
  • 结合真实与合成数据的监督精修,显著改善像素级分割。

基于自注意力变换器的无监督重建网络在统一多类异常检测中已达到顶尖性能,但其主要处理目标特征,因与上下文高度一致,可能导致正常与异常特征都被完美重建,从而无法有效检测异常。此外,这些模型在低空间分辨率潜在空间中进行重建,常导致异常分割不准确。为提升重建模型的效率并增强统一异常检测的泛化能力,我们提出一种简单而有效的方法——仅需一张正常图像提示(OneNIP),即可实现对正常特征的重建和异常特征的恢复,首次实现仅凭一张正常图像完成异常重建与修复。此外,我们设计了一种监督精修模块,利用真实正常样本与合成异常样本回归重建误差,显著提升像素级异常分割效果。OneNIP在三个工业异常检测基准(MVTec、BTAD、VisA)上均优于现有方法。代码与预训练模型已开源:https://github.com/gaobb/OneNIP。

原文摘要 · Abstract (English)

Unsupervised reconstruction networks using self-attention transformers have achieved state-of-the-art performance for multi-class (unified) anomaly detection with a single model. However, these self-attention reconstruction models primarily operate on target features, which may result in perfect reconstruction for both normal and anomaly features due to high consistency with context, leading to failure in detecting anomalies. Additionally, these models often produce inaccurate anomaly segmentation due to performing reconstruction in a low spatial resolution latent space. To enable reconstruction models enjoying high efficiency while enhancing their generalization for unified anomaly detection, we propose a simple yet effective method that reconstructs normal features and restores anomaly features with just One Normal Image Prompt (OneNIP). In contrast to previous work, OneNIP allows for the first time to reconstruct or restore anomalies with just one normal image prompt, effectively boosting unified anomaly detection performance. Furthermore, we propose a supervised refiner that regresses reconstruction errors by using both real normal and synthesized anomalous images, which significantly improves pixel-level anomaly segmentation. OneNIP outperforms previous methods on three industry anomaly detection benchmarks: MVTec, BTAD, and VisA. The code and pre-trained models are available at https://github.com/gaobb/OneNIP.

异常检测自监督图像重建单样本学习

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