arXiv:2412.03342cs.CV2024-12CVPR被引 50

无需训练即可跨领域检测异常,仅用少量正常样本实现高精度少样本异常识别。

UniVAD: A Training-free Unified Model for Few-shot Visual Anomaly Detection

  • 基于视觉基础模型与聚类的组件分割,实现跨域图像成分精准拆解。
  • 在9个跨领域数据集上达到领先性能,少样本下优于专用模型。
  • 适合工业质检、医疗影像等需快速部署的少样本异常检测场景。

视觉异常检测(VAD)旨在识别偏离正常模式的图像异常,涵盖工业、逻辑和医学等多个领域。由于各领域间存在显著差异,现有方法通常针对特定领域定制,难以跨域泛化;且多数方法采用“一类一模型”范式,需大量正常样本训练,导致泛化能力差,阻碍统一评估。为此,我们提出无需训练的统一少样本异常检测方法UniVAD,可跨工业、逻辑、医学等多领域检测异常,仅需测试时提供少量正常样本作为参考,无需在特定领域训练。UniVAD通过基于聚类与视觉基础模型的上下文组件聚类(C³)模块实现图像成分精准分割,并结合组件感知的补丁匹配(CAPM)与图增强组件建模(GECM)模块,在不同语义层级检测异常,最终融合输出结果。我们在九个覆盖多个领域的数据集上进行实验,结果表明,UniVAD在少样本异常检测任务中表现领先,优于各类域专用模型。代码已开源:https://github.com/FantasticGNU/UniVAD。

原文摘要 · Abstract (English)

Visual Anomaly Detection (VAD) aims to identify abnormal samples in images that deviate from normal patterns, covering multiple domains, including industrial, logical, and medical fields. Due to the domain gaps between these fields, existing VAD methods are typically tailored to each domain, with specialized detection techniques and model architectures that are difficult to generalize across different domains. Moreover, even within the same domain, current VAD approaches often follow a "one-category-one-model" paradigm, requiring large amounts of normal samples to train class-specific models, resulting in poor generalizability and hindering unified evaluation across domains. To address this issue, we propose a generalized few-shot VAD method, UniVAD, capable of detecting anomalies across various domains, such as industrial, logical, and medical anomalies, with a training-free unified model. UniVAD only needs few normal samples as references during testing to detect anomalies in previously unseen objects, without training on the specific domain. Specifically, UniVAD employs a Contextual Component Clustering ($C^3$) module based on clustering and vision foundation models to segment components within the image accurately, and leverages Component-Aware Patch Matching (CAPM) and Graph-Enhanced Component Modeling (GECM) modules to detect anomalies at different semantic levels, which are aggregated to produce the final detection result. We conduct experiments on nine datasets spanning industrial, logical, and medical fields, and the results demonstrate that UniVAD achieves state-of-the-art performance in few-shot anomaly detection tasks across multiple domains, outperforming domain-specific anomaly detection models. Code is available at https://github.com/FantasticGNU/UniVAD.

异常检测少样本跨领域统一模型

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