用图像自参考学习提升工业缺陷检测精度,无需依赖外部数据。
Self-Navigated Residual Mamba for Universal Industrial Anomaly Detection
- 测试图像内自我生成参考,动态对比异常区域
- 在多个基准上达到领先性能,各项指标均显著提升
- 适合工业质检场景,尤其对罕见缺陷检测有效
本文提出一种名为自导航残差Mamba(SNARM)的新框架,用于通用工业异常检测。与依赖正常数据预训练特征的传统方法不同,SNARM通过迭代比较测试图像块与自适应选择的图像内参考,动态优化检测能力。首先,通过对比测试图像块与训练特征库计算‘互残差’特征;残差范数小(表明正常性高)的块被用作自生成参考,进一步计算‘内残差’以增强判别信号。将互残差与内残差特征拼接后输入新型多头Mamba模块,其注意力由残差特性动态引导,聚焦异常区域。最终通过集成学习方式聚合自导航Mamba输出得到异常检测结果。在MVTec AD、MVTec 3D和VisA等多个基准上的大量实验表明,SNARM在所有指标(包括Image-AUROC、Pixel-AURC、PRO和AP)上均达到当前最优表现。
原文摘要 · Abstract (English)
In this paper, we propose Self-Navigated Residual Mamba (SNARM), a novel framework for universal industrial anomaly detection that leverages ``self-referential learning'' within test images to enhance anomaly discrimination. Unlike conventional methods that depend solely on pre-trained features from normal training data, SNARM dynamically refines anomaly detection by iteratively comparing test patches against adaptively selected in-image references. Specifically, we first compute the ``inter-residuals'' features by contrasting test image patches with the training feature bank. Patches exhibiting small-norm residuals (indicating high normality) are then utilized as self-generated reference patches to compute ``intra-residuals'', amplifying discriminative signals. These inter- and intra-residual features are concatenated and fed into a novel Mamba module with multiple heads, which are dynamically navigated by residual properties to focus on anomalous regions. Finally, AD results are obtained by aggregating the outputs of a self-navigated Mamba in an ensemble learning paradigm. Extensive experiments on MVTec AD, MVTec 3D, and VisA benchmarks demonstrate that SNARM achieves state-of-the-art (SOTA) performance, with notable improvements in all metrics, including Image-AUROC, Pixel-AURC, PRO, and AP.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。