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

基于CLIP的随机词数据增强用于零样本异常检测

Random Word Data Augmentation with CLIP for Zero-Shot Anomaly Detection

Hitachi America, Ltd.(美国日立公司)

Masato Tamura

AI总结:

本文提出利用CLIP文本编码器结合随机词数据增强生成多样化正常与异常文本嵌入,训练前馈神经网络实现无需训练图像和提示集成的类别无关零样本异常检测,达到最先进性能。

摘要:

本文提出了一种新方法,利用视觉语言模型CLIP作为零样本异常检测的数据源。由于异常检测具有潜在的工业应用价值,人们为开发异常检测器付出了巨大努力。考虑到获取各种异常样本用于训练的困难,大多数现有方法仅使用正常样本训练模型,并在推理阶段测量与正常样本分布的偏差,这需要为每个对象类别训练一个模型。这种低效训练需求的问题已通过设计一种基于CLIP的异常检测器得到解决,该检测器以滑动窗口方式对图像的每个部分应用提示引导分类。然而,该方法仍然需要针对已知对象类别进行繁琐的提示集成。为了克服上述问题,我们提出利用CLIP作为训练数据源。我们的方法使用CLIP中的文本编码器,通过包含正常和异常词汇的典型提示生成文本嵌入。除了这些词汇之外,我们还在提示中插入若干随机生成的词,使编码器能够生成多样化的正常和异常样本集。使用生成的嵌入作为训练数据,前馈神经网络学习从CLIP的嵌入中提取正常和异常特征,因此无需任何训练图像即可获得类别无关的异常检测器。实验结果表明,我们的方法在零样本设置下无需繁琐的提示集成即可实现最先进的性能。

英文摘要:

This paper presents a novel method that leverages a visual-language model, CLIP, as a data source for zero-shot anomaly detection. Tremendous efforts have been put towards developing anomaly detectors due to their potential industrial applications. Considering the difficulty in acquiring various anomalous samples for training, most existing methods train models with only normal samples and measure discrepancies from the distribution of normal samples during inference, which requires training a model for each object category. The problem of this inefficient training requirement has been tackled by designing a CLIP-based anomaly detector that applies prompt-guided classification to each part of an image in a sliding window manner. However, the method still suffers from the labor of careful prompt ensembling with known object categories. To overcome the issues above, we propose leveraging CLIP as a data source for training. Our method generates text embeddings with the text encoder in CLIP with typical prompts that include words of normal and anomaly. In addition to these words, we insert several randomly generated words into prompts, which enables the encoder to generate a diverse set of normal and anomalous samples. Using the generated embeddings as training data, a feed-forward neural network learns to extract features of normal and anomaly from CLIP's embeddings, and as a result, a category-agnostic anomaly detector can be obtained without any training images. Experimental results demonstrate that our method achieves state-of-the-art performance without laborious prompt ensembling in zero-shot setups.

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

↑