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

基于类别无关配准学习的少样本异常检测

Few-Shot Anomaly Detection via Category-Agnostic Registration Learning

College of Information Communications, National University of Defense Technology(国防科技大学信息通信学院);Cooperative Medianet Innovation Center, Shanghai Jiao Tong University(上海交通大学协同媒体网络创新中心);National University of Singapore(新加坡国立大学);King’s College London(伦敦国王学院);Shanghai Artificial Intelligence Laboratory(上海人工智能实验室)

Chaoqin Huang, Haoyan Guan, Aofan Jiang, Ya Zhang, Michael Spratling, Xinchao Wang, Yanfeng Wang

AI总结:

针对现有异常检测需单类别单独建模效率低的问题,提出基于类别无关配准学习的少样本异常检测框架,以配准为自监督代理任务学习通用表征,无需微调即可泛化新类别,在两大基准上显著提升SOTA性能。

摘要:

现有大多数异常检测(AD)方法需要为每个类别配备专用模型。这种范式尽管效果不错,但计算成本高、效率低,无法满足实际应用的需求。受人类通过将查询图像与已知正常图像对比来检测异常的方式启发,本文提出了一种新颖的少样本异常检测(FSAD)框架。该框架利用来自不同类别的正常图像训练集,将旨在对齐同一类别正常图像的配准作为代理任务,进行自监督的类别无关表示学习。测试时,输入一张图像及其对应的支持集(由同一类别的少量正常图像组成),通过将测试图像的配准特征与对应支持图像的特征进行对比,识别出异常。这种设置使模型能够泛化到未见过的测试类别。据我们所知,这是首个无需针对新类别进行模型微调的FSAD方法:单个模型即可应用于所有类别。大量实验证明了所提方法的有效性。具体而言,它在MVTec和MPDD基准上,分别将当前FSAD的最优水平(SOTA)提升了11.3%和8.3%。源代码可在https://github.com/Haoyan-Guan/CAReg 获取。

英文摘要:

Most existing anomaly detection (AD) methods require a dedicated model for each category. Such a paradigm, despite its promising results, is computationally expensive and inefficient, thereby failing to meet the requirements for realworld applications. Inspired by how humans detect anomalies, by comparing a query image to known normal ones, this article proposes a novel few-shot AD (FSAD) framework. Using a training set of normal images from various categories, registration, aiming to align normal images of the same categories, is leveraged as the proxy task for self-supervised category-agnostic representation learning. At test time, an image and its corresponding support set, consisting of a few normal images from the same category, are supplied, and anomalies are identified by comparing the registered features of the test image to its corresponding support image features. Such a setup enables the model to generalize to novel test categories. It is, to our best knowledge, the first FSAD method that requires no model fine-tuning for novel categories: enabling a single model to be applied to all categories. Extensive experiments demonstrate the effectiveness of the proposed method. Particularly, it improves the current state-of-the-art (SOTA) for FSAD by 11.3% and 8.3% on the MVTec and MPDD benchmarks, respectively. The source code is available at https://github.com/Haoyan-Guan/CAReg.

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

↑