arXiv:2410.20047cs.CVcs.LG2024-10NeurIPS被引 40

提出ResAD框架,让异常检测模型无需微调即可跨类别通用。

ResAD: A Simple Framework for Class Generalizable Anomaly Detection

  • 通过学习残差特征分布,减少不同类别间正常特征差异。
  • 在未见过的类别上直接检测,准确率显著优于传统方法。
  • 适合需要快速部署到新类别的工业异常检测场景。

本文研究类通用异常检测问题,目标是训练一个统一的异常检测模型,能在不重新训练或微调的情况下,检测来自不同领域的多种类别的异常。由于不同类别间的正常特征表示差异显著,导致广泛使用的单类别异常检测模型在新类别上性能急剧下降。为此,本文提出简单有效的ResAD框架,直接应用于新类别异常检测。核心思想是学习残差特征分布而非初始特征分布,从而显著降低特征差异。即使在新类别中,正常残差特征分布也与学习到的分布保持一致。ResAD包含三个组件:(1) 特征转换器,将初始特征转换为残差特征;(2) 简单浅层特征约束器,将正常残差特征约束在空间超球体内,进一步减少特征变异并保持各类别特征尺度一致性;(3) 特征分布估计器,估计正常残差特征分布,异常被识别为分布外样本。尽管结构简单,ResAD在直接应用于新类别时仍取得优异检测效果。代码已开源。

原文摘要 · Abstract (English)

This paper explores the problem of class-generalizable anomaly detection, where the objective is to train one unified AD model that can generalize to detect anomalies in diverse classes from different domains without any retraining or fine-tuning on the target data. Because normal feature representations vary significantly across classes, this will cause the widely studied one-for-one AD models to be poorly classgeneralizable (i.e., performance drops dramatically when used for new classes). In this work, we propose a simple but effective framework (called ResAD) that can be directly applied to detect anomalies in new classes. Our main insight is to learn the residual feature distribution rather than the initial feature distribution. In this way, we can significantly reduce feature variations. Even in new classes, the distribution of normal residual features would not remarkably shift from the learned distribution. Therefore, the learned model can be directly adapted to new classes. ResAD consists of three components: (1) a Feature Converter that converts initial features into residual features; (2) a simple and shallow Feature Constraintor that constrains normal residual features into a spatial hypersphere for further reducing feature variations and maintaining consistency in feature scales among different classes; (3) a Feature Distribution Estimator that estimates the normal residual feature distribution, anomalies can be recognized as out-of-distribution. Despite the simplicity, ResAD can achieve remarkable anomaly detection results when directly used in new classes. The code is available at https://github.com/xcyao00/ResAD.

异常检测跨类别残差特征零样本

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