从测试图像内部提取正常特征,提升异常检测精度。
INP-Former++: Advancing Universal Anomaly Detection via Intrinsic Normal Prototypes and Residual Learning
- 从测试图中自动生成正常原型,避免外部参考对齐问题。
- 在多个数据集上达到顶尖性能,支持少样本与零样本检测。
- 结合残差学习与软挖掘策略,适合工业与医疗异常检测场景。
异常检测在工业质检和医学诊断中至关重要,但现有方法通常依赖训练集中正常样本与测试图像的对比,而外观与位置变化常导致对齐困难,限制检测精度。我们观察到多数异常表现为局部差异,即使在异常图像中仍保留有价值的正常信息。这些信息与异常来自同一图像,可能更匹配。因此,我们提出INP-Former,直接从测试图像中提取内在正常原型(INPs)。通过线性组合正常标记生成INPs,并引入INP一致性损失确保其忠实表征测试图像的正常特性。这些INPs引导解码器仅重建正常标记,重构误差作为异常评分。此外,设计软挖掘损失以优化难样本。INP-Former在MVTec-AD、VisA和Real-IAD上实现单类、多类及少样本任务的领先性能,具备通用性。显著的是,该方法还展现出一定零样本检测能力。进一步提出软化版本的一致性损失并引入残差学习,形成INP-Former++,显著提升单类、多类、半监督、少样本及零样本设置下的检测表现。
原文摘要 · Abstract (English)
Anomaly detection (AD) is essential for industrial inspection and medical diagnosis, yet existing methods typically rely on ``comparing'' test images to normal references from a training set. However, variations in appearance and positioning often complicate the alignment of these references with the test image, limiting detection accuracy. We observe that most anomalies manifest as local variations, meaning that even within anomalous images, valuable normal information remains. We argue that this information is useful and may be more aligned with the anomalies since both the anomalies and the normal information originate from the same image. Therefore, rather than relying on external normality from the training set, we propose INP-Former, a novel method that extracts Intrinsic Normal Prototypes (INPs) directly from the test image. Specifically, we introduce the INP Extractor, which linearly combines normal tokens to represent INPs. We further propose an INP Coherence Loss to ensure INPs can faithfully represent normality for the testing image. These INPs then guide the INP-guided Decoder to reconstruct only normal tokens, with reconstruction errors serving as anomaly scores. Additionally, we propose a Soft Mining Loss to prioritize hard-to-optimize samples during training. INP-Former achieves state-of-the-art performance in single-class, multi-class, and few-shot AD tasks across MVTec-AD, VisA, and Real-IAD, positioning it as a versatile and universal solution for AD. Remarkably, INP-Former also demonstrates some zero-shot AD capability. Furthermore, we propose a soft version of the INP Coherence Loss and enhance INP-Former by incorporating residual learning, leading to the development of INP-Former++. The proposed method significantly improves detection performance across single-class, multi-class, semi-supervised, few-shot, and zero-shot settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。