arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.06495cs.CV

面向通用异常检测的基于少样本样本提示的上下文残差学习

Toward Generalist Anomaly Detection via In-context Residual Learning with Few-shot Sample Prompts

School of Computing and Information Systems, Singapore Management University(新加坡管理大学计算与信息系统学院)

Jiawen Zhu, Guansong Pang

AI总结:

本文提出InCTRL方法,利用少样本正常图像作为提示,通过上下文残差学习实现通用异常检测,在九个数据集上显著超越现有方法。

摘要:

本文探讨了通用异常检测(GAD)问题,旨在训练一个单一的检测模型,使其能够泛化到来自不同应用领域的多样化数据集,而无需对目标数据进行进一步训练。近期一些研究表明,像CLIP这样的大型预训练视觉-语言模型(VLM)在检测各种数据集的工业缺陷方面具有很强的泛化能力,但它们的方严重依赖关于缺陷的手工文本提示,这使得它们难以泛化到其他应用中的异常,例如医学图像异常或自然图像中的语义异常。在这项工作中,我们提出训练一个GAD模型,使用少样本正常图像作为样本提示,在多样数据集上即时进行异常检测。为此,我们引入了一种新颖的方法,学习一个用于GAD的上下文残差学习模型,称为InCTRL。它在辅助数据集上训练,基于查询图像与少样本正常样本提示之间残差的整体评估来区分异常和正常样本。无论数据集如何,根据异常的定义,异常样本的残差预计比正常样本更大,从而使InCTRL无需进一步训练即可跨不同领域泛化。我们在九个异常检测数据集上进行了全面实验,建立了一个GAD基准,涵盖工业缺陷异常、医学异常和语义异常的检测,包括一对一和多类设置,InCTRL在其中表现最佳,并显著优于最先进的竞争方法。代码可在https://github.com/mala-lab/InCTRL获取。

英文摘要:

This paper explores the problem of Generalist Anomaly Detection (GAD), aiming to train one single detection model that can generalize to detect anomalies in diverse datasets from different application domains without any further training on the target data. Some recent studies have shown that large pre-trained Visual-Language Models (VLMs) like CLIP have strong generalization capabilities on detecting industrial defects from various datasets, but their methods rely heavily on handcrafted text prompts about defects, making them difficult to generalize to anomalies in other applications, e.g., medical image anomalies or semantic anomalies in natural images. In this work, we propose to train a GAD model with few-shot normal images as sample prompts for AD on diverse datasets on the fly. To this end, we introduce a novel approach that learns an in-context residual learning model for GAD, termed InCTRL. It is trained on an auxiliary dataset to discriminate anomalies from normal samples based on a holistic evaluation of the residuals between query images and few-shot normal sample prompts. Regardless of the datasets, per definition of anomaly, larger residuals are expected for anomalies than normal samples, thereby enabling InCTRL to generalize across different domains without further training. Comprehensive experiments on nine AD datasets are performed to establish a GAD benchmark that encapsulate the detection of industrial defect anomalies, medical anomalies, and semantic anomalies in both one-vs-all and multi-class setting, on which InCTRL is the best performer and significantly outperforms state-of-the-art competing methods. Code is available at https://github.com/mala-lab/InCTRL.

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

↑