arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.19611cs.CV
收藏

RAID: 基于检索的异常检测

RAID: Retrieval-Augmented Anomaly Detection

State Key Laboratory of Synthetical Automation for Process Industries, Northeastern University(过程工业综合自动化 state key laboratory,东北大学)

Mingxiu Cai, Zhe Zhang, Gaochang Wu, Tianyou Chai, Xiatian Zhu

AI总结:

RAID提出了一种基于检索的无监督异常检测框架,通过利用检索到的正常样本指导噪声抑制,实现鲁棒的异常检测和定位。

摘要:

无监督异常检测(UAD)旨在通过建立测试图像与正常模板之间的对应关系来识别异常区域。现有方法主要依赖于图像重建或模板检索,但面临一个根本性挑战:测试图像与正常模板之间的匹配不可避免地会引入噪声,因为存在类内变化、不完美的对应关系和有限的模板。观察到检索增强生成(RAG)在生成过程中直接利用检索到的样本,我们通过这一视角重新解读UAD,并引入RAID,一种基于检索的UAD框架,旨在实现噪声鲁棒的异常检测和定位。与标准RAG不同,我们专注于使用检索到的正常样本来指导异常图谱生成中的噪声抑制。RAID从分层向量数据库中检索类级、语义级和实例级表示,形成从粗到细的流程。匹配成本体积将输入与检索到的示例相关联,随后是引导的混合专家(MoE)网络,该网络利用检索到的样本自适应地抑制匹配噪声并生成细粒度的异常图谱。RAID在MVTec、VisA、MPDD和BTAD基准测试中实现了全-shot、少样本和多数据集设置下的最佳性能。

英文摘要:

Unsupervised Anomaly Detection (UAD) aims to identify abnormal regions by establishing correspondences between test images and normal templates. Existing methods primarily rely on image reconstruction or template retrieval but face a fundamental challenge: matching between test images and normal templates inevitably introduces noise due to intra-class variations, imperfect correspondences, and limited templates. Observing that Retrieval-Augmented Generation (RAG) leverages retrieved samples directly in the generation process, we reinterpret UAD through this lens and introduce \textbf{RAID}, a retrieval-augmented UAD framework designed for noise-resilient anomaly detection and localization. Unlike standard RAG that enriches context or knowledge, we focus on using retrieved normal samples to guide noise suppression in anomaly map generation. RAID retrieves class-, semantic-, and instance-level representations from a hierarchical vector database, forming a coarse-to-fine pipeline. A matching cost volume correlates the input with retrieved exemplars, followed by a guided Mixture-of-Experts (MoE) network that leverages the retrieved samples to adaptively suppress matching noise and produce fine-grained anomaly maps. RAID achieves state-of-the-art performance across full-shot, few-shot, and multi-dataset settings on MVTec, VisA, MPDD, and BTAD benchmarks. \href{https://github.com/Mingxiu-Cai/RAID}{https://github.com/Mingxiu-Cai/RAID}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑