通过匹配成本过滤提升异常检测精度,有效抑制噪声并捕捉细微异常。
CostFilter-AD: Enhancing Anomaly Detection through Matching Cost Filtering
- 引入代价过滤机制,利用多层特征注意力抑制匹配噪声。
- 在MVTec-AD和VisA上实现更优的异常定位性能,显著提升检测准确率。
- 可作为通用插件适配各类无监督异常检测方法,适合工业质检场景。
无监督异常检测(UAD)旨在根据正常样本定位输入图像中的异常区域。现有方法通常基于图像级或特征级匹配来生成异常分数,但匹配过程常不准确却未被重视,导致检测效果不佳。为此,本文将经典匹配任务(如深度估计、光流计算)中的代价过滤思想引入UAD问题,提出CostFilter-AD。具体而言,先构建输入与正常样本间的匹配代价体积,包含两个空间维度和一个匹配维度,编码潜在匹配关系。随后设计一个由输入引导的代价体积过滤网络,作为多层特征上的注意力查询,有效抑制匹配噪声,同时保留边缘结构并捕捉微弱异常。该方法为通用后处理模块,可无缝集成于重建型或嵌入型方法中。在MVTec-AD与VisA基准上的大量实验验证了其在单类与多类异常检测任务中的普适优势。代码与模型将开源于https://github.com/ZHE-SAPI/CostFilter-AD。
原文摘要 · Abstract (English)
Unsupervised anomaly detection (UAD) seeks to localize the anomaly mask of an input image with respect to normal samples. Either by reconstructing normal counterparts (reconstruction-based) or by learning an image feature embedding space (embedding-based), existing approaches fundamentally rely on image-level or feature-level matching to derive anomaly scores. Often, such a matching process is inaccurate yet overlooked, leading to sub-optimal detection. To address this issue, we introduce the concept of cost filtering, borrowed from classical matching tasks, such as depth and flow estimation, into the UAD problem. We call this approach {\em CostFilter-AD}. Specifically, we first construct a matching cost volume between the input and normal samples, comprising two spatial dimensions and one matching dimension that encodes potential matches. To refine this, we propose a cost volume filtering network, guided by the input observation as an attention query across multiple feature layers, which effectively suppresses matching noise while preserving edge structures and capturing subtle anomalies. Designed as a generic post-processing plug-in, CostFilter-AD can be integrated with either reconstruction-based or embedding-based methods. Extensive experiments on MVTec-AD and VisA benchmarks validate the generic benefits of CostFilter-AD for both single- and multi-class UAD tasks. Code and models will be released at https://github.com/ZHE-SAPI/CostFilter-AD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。